Update app.py

app.py

@@ -1,18 +1,14 @@
 from flask import Flask, render_template, request, jsonify
+import google.generativeai as genai
 import os
 from PIL import Image
 import tempfile
 
-
-from google import genai
-from google.genai import types
-
-
 app = Flask(__name__)
 
 # Configuration Gemini
 token = os.environ.get("TOKEN")
-client = genai.Client(api_key=token)
+genai.configure(api_key=token)
 
 
 safety_settings = [
@@ -22,14 +18,13 @@ safety_settings = [
     {"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "BLOCK_NONE"},
 ]
 
-mm = """resous cet exercice. tu répondras en détaillant au maximum ton procédé de calcul.En français stp et sachant qu'il s'agit d'un devoir niveau terminal s je veux donc une présentation type de cela.  Réponds donc en me donnant directement le devoir et rien d'autre. """
+mm = """ resous cet exercice. tu répondras en détaillant au maximum ton procédé de calcul.En français stp et sachant qu'il s'agit d'un devoir niveau terminal s je veux donc une présentation type de cela.  Réponds donc en me donnant directement le devoir et rien d'autre. write you answer in french with rendering Latex"""
 
-model_name = "gemini-2.0-flash-thinking-exp" 
 
-#model = genai.GenerativeModel(
-#    model_name="gemini-2.0-flash-thinking-exp",
-#    safety_settings=safety_settings
-#)
+model = genai.GenerativeModel(
+    model_name="gemini-exp-1206",
+    safety_settings=safety_settings
+)
 
 @app.route('/')
 def home():
@@ -49,20 +44,10 @@ def generate():
             image = Image.open(temp_file.name)
         
             # Générer le contenu
-            #response = model.generate_content([mm, image])
-            
-            response = client.models.generate_content(
-    model=model_name,
-    contents=[
-        image,
-       mm
-    ]
-)
-            #result = response.text
-            
-            result = response.candidates[0].content.parts[1].text
+            response = model.generate_content([mm, image])
+            result = response.text
+            #result = response.candidates[0].content.parts[1].text
             
-            print(result)
             return jsonify({"result": result})
         except Exception as e:
              return jsonify({'error': str(e)}), 500