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)