diff options
Diffstat (limited to 'backend/ai-vision.py')
-rw-r--r-- | backend/ai-vision.py | 48 |
1 files changed, 0 insertions, 48 deletions
diff --git a/backend/ai-vision.py b/backend/ai-vision.py deleted file mode 100644 index 866bcb8..0000000 --- a/backend/ai-vision.py +++ /dev/null @@ -1,48 +0,0 @@ -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) |