Update app.py

app.py

@@ -1,5 +1,6 @@
 import streamlit as st
-import google.generativeai as genai
+from google import genai
+from google.genai import types
 import os
 from PIL import Image
 import io
@@ -8,10 +9,10 @@ import json
 
 GOOGLE_API_KEY =  os.environ.get("GEMINI_API_KEY")
 
-genai.configure(api_key=GOOGLE_API_KEY)
-client = genai.GenerativeModel("gemini-2.0-flash-thinking-exp-01-21")
-client.http_options = {'api_version': 'v1alpha'}
-
+client = genai.Client(
+    api_key=GOOGLE_API_KEY,
+    http_options={'api_version': 'v1alpha'},
+)
 
 st.title("Résolution d'exercice avec Gemini")
 
@@ -31,16 +32,13 @@ if uploaded_file is not None:
 
         mode = 'starting'
         try:
-            responses = client.generate_content(
-                model="gemini-2.0-flash-thinking-exp-01-21",
-                generation_config=genai.types.GenerationConfig(
-                    include_thoughts=True
-                ),
-                contents=[
-                    {'inline_data': {'mime_type': 'image/png', 'data': img_str}},
-                    "Résous cet exercice. ça doit être bien présenté et espacé afin d'être facile à lire."
-                ],
-                stream=True # Activer le streaming
+            response = client.models.generate_content_stream(
+                    model="gemini-2.0-flash-thinking-exp-01-21",
+                    config={'thinking_config': {'include_thoughts': True}},
+                    contents=[
+                        {'inline_data': {'mime_type': 'image/png', 'data': img_str}},
+                        "Resous cette exercice. ça doit être bien présentable et espacé afin d'être facile à lire."
+                    ]
             )
 
             full_response = "" # Pour accumuler toute la réponse pour un affichage final propre