Update app.py

app.py

@@ -14,7 +14,10 @@ from markupsafe import Markup
 from gradio_client import Client,handle_file
 from functools import wraps
 import os
-import google.generativeai as genai
+
+from google import genai
+from google.genai import types
+
 import textwrap
 import PIL.Image
 from werkzeug.utils import secure_filename
@@ -58,8 +61,12 @@ safety_settings = [
 ]
 
 
-genai.configure(api_key=GOOGLE_API_KEY)
+#genai.configure(api_key=GOOGLE_API_KEY)
 
+client = genai.Client(
+    api_key=GOOGLE_API_KEY,
+    http_options={'api_version': 'v1alpha'},
+)
 
 MAX_CONTENT_LENGTH = 760 * 1024 * 1024  # 16MB max file size
 
@@ -399,8 +406,14 @@ La conclusion répond à la question critique posée en introduction, en s'appuy
         prompt = methodologie
 
         try:
-            model = genai.GenerativeModel(model_name="models/gemini-2.0-flash-exp", safety_settings=safety_settings)
-            response = model.generate_content(prompt, request_options={"timeout": 1100})
+            response = client.models.generate_content_stream(
+                    model="gemini-2.0-flash-thinking-exp-01-21",
+                    config={'thinking_config': {'include_thoughts': True}},
+                    contents=[prompt]
+            )
+            
+            #model = genai.GenerativeModel(model_name="models/gemini-2.0-flash-exp", safety_settings=safety_settings)
+            #response = model.generate_content(prompt, request_options={"timeout": 1100})
             
             dissertation = response.text
             return jsonify({"response": dissertation}), 200