From 2780072a278fa5128cf961284e5c04312d5e6119 Mon Sep 17 00:00:00 2001 From: altaf-creator Date: Mon, 30 Jun 2025 20:55:53 +0800 Subject: initial commit --- backend/ai-vision.py | 48 ++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 48 insertions(+) create mode 100644 backend/ai-vision.py (limited to 'backend/ai-vision.py') diff --git a/backend/ai-vision.py b/backend/ai-vision.py new file mode 100644 index 0000000..866bcb8 --- /dev/null +++ b/backend/ai-vision.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, 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 after the '>' prefix. Do not add numbers 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) -- cgit v1.2.3