Update app.py

app.py

@@ -3,6 +3,11 @@ from google import genai
 from google.genai import types
 from PIL import Image
 import json
+import logging
+
+# Configuration du logging
+logging.basicConfig(level=logging.DEBUG)
+logger = logging.getLogger(__name__)
 
 def stream_response(container, response):
     """Gère le streaming de la réponse avec un affichage progressif"""
@@ -12,24 +17,36 @@ def stream_response(container, response):
     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)
+    try:
+        for chunk in response:
+            logger.debug(f"Chunk reçu: {chunk}")
+            
+            try:
+                for part in chunk.candidates[0].content.parts:
+                    if hasattr(part, 'thought') and 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)
+            except Exception as e:
+                logger.error(f"Erreur lors du traitement d'un chunk: {e}")
+                logger.debug(f"Contenu du chunk problématique: {chunk}")
+                raise
+                
+    except Exception as e:
+        logger.error(f"Erreur dans le streaming de la réponse: {e}")
+        raise
 
 def main():
     st.title("Analyseur d'Images Géométriques avec Gemini")
@@ -38,8 +55,8 @@ def main():
     try:
         api_key = st.secrets["GEMINI_API_KEY"]
     except Exception as e:
-        st.error("Erreur: dans les secrets.")
-        
+        logger.error(f"Erreur dans la récupération des secrets: {e}")
+        st.error("Erreur: Impossible d'accéder aux secrets de l'application.")
         return
     
     # Initialisation du client
@@ -49,7 +66,8 @@ def main():
             http_options={'api_version':'v1alpha'}
         )
     except Exception as e:
-        st.error(f"Erreur lors de l'initialisation du client: {e}")
+        logger.error(f"Erreur d'initialisation du client: {e}")
+        st.error(f"Erreur lors de l'initialisation du client Gemini: {e}")
         return
     
     # Upload de l'image
@@ -70,6 +88,16 @@ def main():
                 
                 with st.spinner("Analyse en cours..."):
                     try:
+                        # Préparation de la requête
+                        request = {
+                            "contents": [
+                                {"image": image},
+                                {"text": "What's the area of the overlapping region?"}
+                            ]
+                        }
+                        
+                        logger.debug(f"Envoi de la requête au modèle: {request}")
+                        
                         # Génération de la réponse en streaming
                         response = client.models.generate_content_stream(
                             model=model_name,
@@ -84,10 +112,12 @@ def main():
                         stream_response(response_container, response)
                                 
                     except Exception as e:
-                        st.error(f"Erreur lors de l'analyse: {e}")
+                        logger.error(f"Erreur lors de l'analyse: {e}", exc_info=True)
+                        st.error(f"Erreur lors de l'analyse de l'image: {str(e)}")
                         
         except Exception as e:
-            st.error(f"Erreur lors du traitement de l'image: {e}")
+            logger.error(f"Erreur lors du traitement de l'image: {e}", exc_info=True)
+            st.error(f"Erreur lors du traitement de l'image: {str(e)}")
 
 if __name__ == "__main__":
     main()