Update app.py

app.py

@@ -4,7 +4,8 @@ from google.genai import types
 from PIL import Image
 import json
 import logging
-from typing import Optional, Generator, Any
+import re
+from typing import Optional, Generator, Any, Tuple
 import sys
 from pathlib import Path
 
@@ -19,6 +20,40 @@ logging.basicConfig(
 )
 logger = logging.getLogger(__name__)
 
+class LatexFormatter:
+    """Classe pour gérer le formatage LaTeX"""
+    
+    @staticmethod
+    def format_inline_latex(text: str) -> str:
+        """Formate le LaTeX inline en ajoutant les délimiteurs appropriés"""
+        # Remplace les délimiteurs $ simples par $$ pour inline math
+        text = re.sub(r'(?<!\\)\$([^$]+)(?<!\\)\$', r'$\1$', text)
+        return text
+    
+    @staticmethod
+    def format_block_latex(text: str) -> str:
+        """Formate les blocs LaTeX"""
+        # Ajoute des sauts de ligne avant et après les blocs LaTeX
+        text = re.sub(r'\$\$(.*?)\$\$', r'\n\n$$\1$$\n\n', text, flags=re.DOTALL)
+        return text
+    
+    @staticmethod
+    def enhance_latex_display(text: str) -> str:
+        """Améliore l'affichage LaTeX global"""
+        # Gère les cas spéciaux de LaTeX
+        text = text.replace('\\[', '$$').replace('\\]', '$$')  # Convertit \[ \] en $$
+        text = text.replace('\\(', '$').replace('\\)', '$')    # Convertit \( \) en $
+        
+        # Assure l'espacement correct autour des équations
+        text = re.sub(r'(\$\$.*?\$\$)', r'\n\n\1\n\n', text, flags=re.DOTALL)
+        
+        # Améliore l'alignement des environnements mathématiques
+        text = re.sub(r'(\\begin{[^}]*})(.*?)(\\end{[^}]*})', 
+                     r'\n\n$$\1\2\3$$\n\n', 
+                     text, flags=re.DOTALL)
+        
+        return text
+
 class GeminiClient:
     """Classe pour gérer les interactions avec l'API Gemini"""
     def __init__(self, api_key: str):
@@ -47,7 +82,7 @@ class GeminiClient:
                 config={'thinking_config': {'include_thoughts': True}},
                 contents=[
                     image,
-                    prompt
+                    prompt + " Si ta réponse contient des expressions mathématiques, utilise la notation LaTeX avec les délimiteurs appropriés ($ pour inline, $$ pour les blocs)."
                 ]
             )
             return response
@@ -55,14 +90,31 @@ class GeminiClient:
             logger.error(f"Erreur lors de l'analyse de l'image: {e}")
             raise
 
+def format_text_with_latex(text: str) -> str:
+    """Formate le texte en appliquant les améliorations LaTeX"""
+    formatter = LatexFormatter()
+    text = formatter.format_inline_latex(text)
+    text = formatter.format_block_latex(text)
+    text = formatter.enhance_latex_display(text)
+    return text
+
 def stream_response(container, response: Generator) -> None:
-    """Gère le streaming de la réponse avec un affichage progressif et une gestion d'erreurs robuste"""
+    """Gère le streaming de la réponse avec support LaTeX amélioré"""
     mode = 'starting'
     thinking_placeholder = None
     answer_placeholder = None
     thinking_text = ""
     answer_text = ""
     
+    # Configuration CSS pour améliorer l'affichage LaTeX
+    st.markdown("""
+        <style>
+        .katex { font-size: 1.1em !important; }
+        .katex-display { overflow: auto hidden; }
+        .markdown-text-container { max-width: 100%; }
+        </style>
+        """, unsafe_allow_html=True)
+    
     try:
         for chunk in response:
             logger.debug(f"Chunk reçu: {chunk}")
@@ -93,6 +145,9 @@ def stream_response(container, response: Generator) -> None:
                     text = getattr(part, 'text', '')
                     if not text:
                         continue
+                    
+                    # Formatage LaTeX du texte
+                    formatted_text = format_text_with_latex(text)
                         
                     if has_thought:
                         if mode != "thinking":
@@ -100,7 +155,7 @@ def stream_response(container, response: Generator) -> None:
                                 with container.expander("Voir le raisonnement", expanded=False):
                                     thinking_placeholder = st.empty()
                             mode = "thinking"
-                        thinking_text += text
+                        thinking_text += formatted_text
                         thinking_placeholder.markdown(thinking_text)
                     else:
                         if mode != "answering":
@@ -108,7 +163,7 @@ def stream_response(container, response: Generator) -> None:
                                 answer_placeholder = container.empty()
                                 container.subheader("Réponse")
                             mode = "answering"
-                        answer_text += text
+                        answer_text += formatted_text
                         answer_placeholder.markdown(answer_text)
                         
             except json.JSONDecodeError as e:
@@ -127,7 +182,7 @@ def stream_response(container, response: Generator) -> None:
     finally:
         if not answer_text and not thinking_text:
             container.warning("Aucune réponse n'a pu être générée. Veuillez réessayer.")
-            
+
 def validate_image(uploaded_file) -> Optional[Image.Image]:
     """Valide et ouvre une image téléchargée"""
     try:
@@ -147,6 +202,28 @@ def main():
     
     st.title("Mariam M-0")
     
+    # Configuration supplémentaire pour LaTeX
+    st.markdown("""
+        <style>
+        /* Styles globaux pour LaTeX */
+        .latex-container { 
+            margin: 1em 0;
+            padding: 0.5em;
+            background: #f9f9f9;
+            border-radius: 4px;
+        }
+        /* Amélioration de l'espacement des équations */
+        .katex-display { 
+            margin: 1.5em 0 !important; 
+            padding: 0.5em 0;
+        }
+        /* Amélioration du rendu sur mobile */
+        @media (max-width: 768px) {
+            .katex { font-size: 1em !important; }
+        }
+        </style>
+    """, unsafe_allow_html=True)
+    
     # Récupération de la clé API
     try:
         api_key = st.secrets["GEMINI_API_KEY"]
@@ -175,7 +252,7 @@ def main():
             st.image(image, caption="Image téléchargée", use_container_width=True)
             
             model_name = "gemini-2.0-flash-thinking-exp-01-21"
-            prompt = "Résous cet exercice. La réponse doit être bien présentée et espacée pour faciliter la lecture. Réponds en français."
+            prompt = "Résous cet exercice mathématique. La réponse doit être bien présentée et espacée pour faciliter la lecture. Réponds en français et utilise la notation LaTeX pour toutes les expressions mathématiques."
             
             if st.button("Analyser l'image", type="primary"):
                 response_container = st.container()