Update app.py

app.py

@@ -108,29 +108,31 @@ def create_math_system_message():
 Always be precise, educational, and encourage mathematical thinking."""
 
 def render_latex(text):
-    """Enhanced LaTeX rendering - fixes raw code output"""
+    """Enhanced LaTeX rendering with better error handling and formatting"""
     if not text or len(text) < 5:
         return text
     
     try:
-        # Fix common LaTeX patterns from Qwen
-        text = re.sub(r'(?<!\\)\$([^\$]+)\$(?!\$)', r'$\1$', text)
-        text = re.sub(r'\$\$([^\$]+)\$$', r'$$\1$$', text)
-        text = re.sub(r'\\\[([^\\]+)\\\]', r'$$\1$$', text)
-        text = re.sub(r'\\\(([^\\]+)\\\)', r'$\1$', text)
+        # Ensure proper LaTeX delimiters
+        text = re.sub(r'(?<!\\)\$([^\$]+)\$(?!\$)', r'$\1$', text)  # Inline math
+        text = re.sub(r'\$\$([^\$]+)\$\$', r'$$\1$$', text)        # Display math
+        text = re.sub(r'\\\[([^\\]+)\\\]', r'$$\1$$', text)       # Display math alternative
+        text = re.sub(r'\\\(([^\\]+)\\\)', r'$\1$', text)         # Inline math alternative
         
-        # Fix escaped LaTeX commands
-        text = re.sub(r'\\(lim|frac|sqrt|int|sum|prod|partial|nabla|infty|to|le|ge|neq|approx|cdot|times|div|deg|prime|log|ln|sin|cos|tan|cot|sec|csc|arcsin|arccos|arctan|sinh|cosh)', r'\1', text)
+        # Fix common LaTeX commands
+        text = re.sub(r'\\(lim|frac|sqrt|int|sum|prod|partial|nabla|infty|to|le|ge|neq|approx|cdot|times|div|deg|prime|log|ln|sin|cos|tan|cot|sec|csc|arcsin|arccos|arctan|sinh|cosh)', r'\\\1', text)
         
-        # Ensure boxed answers render
+        # Ensure boxed answers and fractions render correctly
         text = re.sub(r'\\boxed\{([^}]+)\}', r'$$\boxed{\1}$$', text)
         text = re.sub(r'\\frac\{([^}]+)\}\{([^}]+)\}', r'$\frac{\1}{\2}$', text)
         
-        # Clean up extra spaces
-        text = re.sub(r'\s*([\$\\])\s*', r'\1', text)
+        # Clean extra spaces and escape issues
+        text = re.sub(r'\s+([\$\\])\s+', r'\1', text)
+        text = text.replace(r'frac', r'\frac')  # Catch any unescaped frac
         
     except Exception as e:
         print(f"⚠️ LaTeX formatting error: {e}")
+        return text  # Return original text if formatting fails
     
     return text