Update app.py

app.py

@@ -5,7 +5,7 @@ from PIL import Image
 import json
 import logging
 import re
-from typing import Optional, Generator, Any, Tuple
+from typing import Optional, Generator, Any, Dict
 import sys
 from pathlib import Path
 
@@ -20,16 +20,15 @@ logging.basicConfig(
 )
 logger = logging.getLogger(__name__)
 
-
 class LatexFormatter:
     """Classe améliorée pour le formatage LaTeX"""
     
     @staticmethod
     def cleanup_latex_fractions(text: str) -> str:
         """Nettoie et formate correctement les fractions LaTeX"""
-        # Remplace \frac{x}{y} par \frac{x}{y}
+        # Améliore le rendu des fractions
         text = re.sub(r'\\frac\{([^}]*)\}\{([^}]*)\}', r'$$\\frac{\1}{\2}$$', text)
-        # Gère les cas spéciaux comme \left et \right
+        # Gère les délimiteurs left/right
         text = re.sub(r'\\left\(', r'\\left(', text)
         text = re.sub(r'\\right\)', r'\\right)', text)
         return text
@@ -37,17 +36,30 @@ class LatexFormatter:
     @staticmethod
     def format_inline_math(text: str) -> str:
         """Formate les expressions mathématiques en ligne"""
-        # Ajoute des espaces autour des expressions inline
+        # Améliore l'espacement autour des expressions inline
         text = re.sub(r'(?<!\\)\$([^$]+)(?<!\\)\$', r' $\1$ ', text)
+        # Gère les cas spéciaux où il y a déjà des $$ mais en inline
+        text = re.sub(r'(?<!\\)\$\$([^$]+)(?<!\\)\$\$', r' $\1$ ', text)
         return text
 
     @staticmethod
     def format_display_math(text: str) -> str:
         """Formate les expressions mathématiques en mode display"""
-        # Ajoute des sauts de ligne autour des expressions display
+        # Ajoute des sauts de ligne et de l'espacement
         text = re.sub(r'\$\$(.*?)\$\$', r'\n\n$$\1$$\n\n', text, flags=re.DOTALL)
         return text
 
+    @staticmethod
+    def format_special_environments(text: str) -> str:
+        """Formate les environnements mathématiques spéciaux"""
+        # Gère les environnements align, gather, etc.
+        envs = ['align', 'gather', 'equation', 'array', 'matrix']
+        for env in envs:
+            pattern = f'\\\\begin{{{env}}}(.*?)\\\\end{{{env}}}'
+            text = re.sub(pattern, f'\n\n$$\\\\begin{{{env}}}\\1\\\\end{{{env}}}$$\n\n', 
+                         text, flags=re.DOTALL)
+        return text
+
     @staticmethod
     def enhance_latex_display(text: str) -> str:
         """Améliore globalement l'affichage LaTeX"""
@@ -55,25 +67,17 @@ class LatexFormatter:
         text = text.replace('\\[', '$$').replace('\\]', '$$')
         text = text.replace('\\(', '$').replace('\\)', '$')
         
-        # Nettoyage des fractions
+        # Application des différentes améliorations
         text = LatexFormatter.cleanup_latex_fractions(text)
-        
-        # Formatage du mode inline
+        text = LatexFormatter.format_special_environments(text)
         text = LatexFormatter.format_inline_math(text)
-        
-        # Formatage du mode display
         text = LatexFormatter.format_display_math(text)
         
-        # Assure que chaque ligne d'équation est bien espacée
+        # Assure un bon espacement des équations
         text = re.sub(r'(\n\s*\$\$[^$]+\$\$)', r'\n\n\1\n\n', text)
         
         return text
 
-
-    
-
-
-
 class GeminiClient:
     """Classe pour gérer les interactions avec l'API Gemini"""
     def __init__(self, api_key: str):
@@ -102,7 +106,7 @@ class GeminiClient:
                 config={'thinking_config': {'include_thoughts': True}},
                 contents=[
                     image,
-                    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)."
+                    prompt + " Utilise la notation LaTeX appropriée avec $...$ pour les expressions en ligne et $$...$$ pour les équations importantes. Pour les fractions, utilise \\frac{num}{den}."
                 ]
             )
             return response
@@ -110,21 +114,11 @@ 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:
-    """Fonction de streaming améliorée avec meilleur support LaTeX"""
-    
-    # Configuration CSS améliorée pour le rendu LaTeX
+def setup_latex_display():
+    """Configure l'affichage LaTeX dans Streamlit"""
     st.markdown("""
         <style>
-        /* Amélioration du rendu LaTeX */
+        /* Styles de base pour LaTeX */
         .katex { 
             font-size: 1.2em !important; 
             padding: 0.2em 0;
@@ -136,24 +130,61 @@ def stream_response(container, response: Generator) -> None:
             padding: 0.5em;
             border-radius: 4px;
         }
-        /* Amélioration de l'espacement des fractions */
+        /* Amélioration des fractions */
         .katex .frac-line {
             border-bottom-width: 0.08em;
         }
         .katex .mfrac .frac-line {
             margin: 0.1em 0;
         }
-        /* Meilleur rendu sur mobile */
+        /* Style des matrices */
+        .katex .mord.matrix {
+            margin: 0.2em 0;
+        }
+        /* Amélioration des indices et exposants */
+        .katex .msupsub {
+            font-size: 0.9em;
+        }
+        /* Meilleure lisibilité des symboles */
+        .katex .mathdefault {
+            color: inherit;
+        }
+        /* Adaptation mobile */
         @media (max-width: 768px) {
-            .katex { font-size: 1.1em !important; }
+            .katex { 
+                font-size: 1.1em !important; 
+            }
             .katex-display { 
                 padding: 0.3em;
                 margin: 1em 0 !important;
+                font-size: 0.9em !important;
             }
         }
+        /* Conteneur de réponse */
+        .response-container {
+            margin: 1em 0;
+            padding: 1em;
+            border-radius: 8px;
+            background: rgba(255, 255, 255, 0.05);
+        }
+        /* Mise en évidence des équations importantes */
+        .important-equation {
+            border-left: 3px solid #4CAF50;
+            padding-left: 1em;
+        }
         </style>
     """, unsafe_allow_html=True)
-        
+
+def stream_response(container, response: Generator) -> None:
+    """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 = ""
+    
+    setup_latex_display()
+    
     try:
         for chunk in response:
             logger.debug(f"Chunk reçu: {chunk}")
@@ -186,8 +217,8 @@ def stream_response(container, response: Generator) -> None:
                         continue
                     
                     # Formatage LaTeX du texte
-                    formatted_text = format_text_with_latex(text)
-                        
+                    formatted_text = LatexFormatter.enhance_latex_display(text)
+                    
                     if has_thought:
                         if mode != "thinking":
                             if thinking_placeholder is None:
@@ -216,7 +247,7 @@ def stream_response(container, response: Generator) -> None:
     except Exception as e:
         logger.error(f"Erreur fatale dans le streaming de la réponse: {e}")
         if not answer_text and not thinking_text:
-            container.error("Une erreur est survenue lors de l'analyse de l'image. Veuillez réessayer.")
+            container.error("Une erreur est survenue lors de l'analyse. Veuillez réessayer.")
         raise
     finally:
         if not answer_text and not thinking_text:
@@ -236,32 +267,12 @@ def main():
     st.set_page_config(
         page_title="Mariam M-0",
         page_icon="🔍",
-        layout="wide"
+        layout="wide",
+        initial_sidebar_state="collapsed"
     )
     
     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)
+    setup_latex_display()
     
     # Récupération de la clé API
     try:
@@ -280,7 +291,7 @@ def main():
     
     # Interface utilisateur
     uploaded_file = st.file_uploader(
-        "Choisissez une image géométrique",
+        "Choisissez une image contenant des mathématiques",
         type=['png', 'jpg', 'jpeg'],
         help="Formats supportés: PNG, JPG, JPEG"
     )
@@ -293,6 +304,7 @@ def main():
             model_name = "gemini-2.0-flash-thinking-exp-01-21"
             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()