Update app.py

app.py

@@ -20,40 +20,60 @@ logging.basicConfig(
 )
 logger = logging.getLogger(__name__)
 
+
 class LatexFormatter:
-    """Classe pour gérer le formatage LaTeX"""
+    """Classe améliorée pour 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)
+    def cleanup_latex_fractions(text: str) -> str:
+        """Nettoie et formate correctement les fractions LaTeX"""
+        # Remplace \frac{x}{y} par \frac{x}{y}
+        text = re.sub(r'\\frac\{([^}]*)\}\{([^}]*)\}', r'$$\\frac{\1}{\2}$$', text)
+        # Gère les cas spéciaux comme \left et \right
+        text = re.sub(r'\\left\(', r'\\left(', text)
+        text = re.sub(r'\\right\)', r'\\right)', 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
+    def format_inline_math(text: str) -> str:
+        """Formate les expressions mathématiques en ligne"""
+        # Ajoute des espaces autour des expressions 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
         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 $
+        """Améliore globalement l'affichage LaTeX"""
+        # Prétraitement
+        text = text.replace('\\[', '$$').replace('\\]', '$$')
+        text = text.replace('\\(', '$').replace('\\)', '$')
+        
+        # Nettoyage des fractions
+        text = LatexFormatter.cleanup_latex_fractions(text)
         
-        # Assure l'espacement correct autour des équations
-        text = re.sub(r'(\$\$.*?\$\$)', r'\n\n\1\n\n', text, flags=re.DOTALL)
+        # Formatage du mode inline
+        text = LatexFormatter.format_inline_math(text)
         
-        # 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)
+        # Formatage du mode display
+        text = LatexFormatter.format_display_math(text)
+        
+        # Assure que chaque ligne d'équation est bien espacée
+        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):
@@ -99,22 +119,41 @@ def format_text_with_latex(text: str) -> str:
     return text
 
 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 = ""
+    """Fonction de streaming améliorée avec meilleur support LaTeX"""
     
-    # Configuration CSS pour améliorer l'affichage LaTeX
+    # Configuration CSS améliorée pour le rendu LaTeX
     st.markdown("""
         <style>
-        .katex { font-size: 1.1em !important; }
-        .katex-display { overflow: auto hidden; }
-        .markdown-text-container { max-width: 100%; }
+        /* Amélioration du rendu LaTeX */
+        .katex { 
+            font-size: 1.2em !important; 
+            padding: 0.2em 0;
+        }
+        .katex-display { 
+            margin: 1.5em 0 !important;
+            overflow: auto hidden;
+            background: rgba(248, 249, 250, 0.05);
+            padding: 0.5em;
+            border-radius: 4px;
+        }
+        /* Amélioration de l'espacement des fractions */
+        .katex .frac-line {
+            border-bottom-width: 0.08em;
+        }
+        .katex .mfrac .frac-line {
+            margin: 0.1em 0;
+        }
+        /* Meilleur rendu sur mobile */
+        @media (max-width: 768px) {
+            .katex { font-size: 1.1em !important; }
+            .katex-display { 
+                padding: 0.3em;
+                margin: 1em 0 !important;
+            }
+        }
         </style>
-        """, unsafe_allow_html=True)
-    
+    """, unsafe_allow_html=True)
+        
     try:
         for chunk in response:
             logger.debug(f"Chunk reçu: {chunk}")