Update app.py

app.py

@@ -4,15 +4,45 @@ from google.genai import types
 from PIL import Image
 import json
 
+def stream_response(container, response):
+    """Gère le streaming de la réponse avec un affichage progressif"""
+    mode = 'starting'
+    thinking_placeholder = None
+    answer_placeholder = None
+    thinking_text = ""
+    answer_text = ""
+    
+    for chunk in response:
+        for part in chunk.candidates[0].content.parts:
+            if part.thought:
+                if mode != "thinking":
+                    if thinking_placeholder is None:
+                        with container.expander("Voir le raisonnement", expanded=False):
+                            thinking_placeholder = st.empty()
+                    mode = "thinking"
+                thinking_text += part.text
+                thinking_placeholder.markdown(thinking_text)
+            else:
+                if mode != "answering":
+                    if answer_placeholder is None:
+                        answer_placeholder = container.empty()
+                        container.subheader("Réponse")
+                    mode = "answering"
+                answer_text += part.text
+                answer_placeholder.markdown(answer_text)
+
 def main():
     st.title("Analyseur d'Images Géométriques avec Gemini")
     
     # Utilisation des secrets pour la clé API
     try:
-        api_key = st.secrets["GEMINI_API_KEY"]
+        api_key = st.secrets["GOOGLE_API_KEY"]
     except Exception as e:
-        st.error("Erreur: .")
-        
+        st.error("Erreur: La clé API Google n'est pas configurée dans les secrets.")
+        st.info("Créez un fichier .streamlit/secrets.toml avec le contenu suivant:")
+        st.code("""
+        GOOGLE_API_KEY = "votre_clé_api"
+        """)
         return
     
     # Initialisation du client
@@ -38,10 +68,13 @@ def main():
             model_name = "gemini-2.0-flash-thinking-exp-01-21"
             
             if st.button("Analyser l'image"):
+                # Création d'un conteneur pour la réponse en streaming
+                response_container = st.container()
+                
                 with st.spinner("Analyse en cours..."):
                     try:
-                        # Génération de la réponse
-                        response = client.models.generate_content(
+                        # Génération de la réponse en streaming
+                        response = client.models.generate_content_stream(
                             model=model_name,
                             config={'thinking_config': {'include_thoughts': True}},
                             contents=[
@@ -50,14 +83,8 @@ def main():
                             ]
                         )
                         
-                        # Affichage des résultats
-                        for part in response.candidates[0].content.parts:
-                            if part.thought:
-                                with st.expander("Voir le raisonnement"):
-                                    st.markdown(part.text)
-                            else:
-                                st.subheader("Réponse")
-                                st.markdown(part.text)
+                        # Streaming de la réponse
+                        stream_response(response_container, response)
                                 
                     except Exception as e:
                         st.error(f"Erreur lors de l'analyse: {e}")