1 files changed, 48 insertions, 0 deletions
diff --git a/backend/main.py b/backend/main.py
new file mode 100644
index 0000000..128485f
--- /dev/null
+++ b/backend/main.py
@@ -0,0 +1,48 @@
+from google import genai
+from google.genai import types
+import flask
+import flask_cors
+
+app = flask.Flask(__name__)
+flask_cors.CORS(app)
+
+UPLOAD_FOLDER = "uploads"
+app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
+
+client = genai.Client(api_key="AIzaSyA-mA7v4xlbEAz80uKf3Dywpbeh9ZfpLMs")
+
+
+@app.route("/yeah", methods=["POST"])
+def yeah_flask():
+    uploaded_files = flask.request.files.getlist("files")
+
+    file_names = []
+
+    contents = []
+
+    for file in uploaded_files:
+        file_data = file.read()
+        print(file.filename)
+        file_names.append(file.filename)
+        contents.append(
+                types.Part.from_bytes(
+                    data=file_data,
+                    mime_type=file.mimetype
+                )
+        )
+
+    contents.append("Please summarise the following content. Make it concise, but detailed (this is a priority), organised into a mind map, structured with clear hierarchy, where each item is first prefixed and just prefixed with a > times depth, where depth starts at 1. For example, a sub item would be \">> subitem\". Do not start or end your message with an approval message, start immediately with the content requested. Absolutely do not add any blank lines, and do not add spaces OR numbering after the '>' prefix.")
+
+    response = client.models.generate_content(
+        model="gemini-2.0-flash",
+        contents=contents
+    )
+
+    print(response)
+
+    print("hi")
+    return flask.jsonify({"status": "success", "map": response.text})
+
+
+if __name__ == '__main__':
+    app.run(debug=True, host='127.0.0.1', port=9000)