Update app.py

app.py

@@ -109,31 +109,21 @@ def create_math_system_message():
 Always be precise, educational, and encourage mathematical thinking."""
 
 def render_latex(text):
-    """Enhanced LaTeX rendering with better error handling and formatting"""
-    if not text or len(text) < 5:
+    """Minimal LaTeX cleanup - let the model do the work"""
+    if not text:
         return text
     
     try:
-        # 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
+        # Convert LaTeX bracket notation to dollar signs
+        text = re.sub(r'\\\[(.*?)\\\]', r'$$\1$$', text, flags=re.DOTALL)
+        text = re.sub(r'\\\((.*?)\\\)', r'$\1$', text, flags=re.DOTALL)
         
-        # 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 and fractions render correctly
-        text = re.sub(r'\\boxed\{([^}]+)\}', r'$$\boxed{\1}$$', text)
-        text = re.sub(r'\\frac\{([^}]+)\}\{([^}]+)\}', r'$\frac{\1}{\2}$', 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
+        # Fix boxed answers ONLY if not already in math mode
+        if '\\boxed' in text and not re.search(r'\$.*\\boxed.*\$', text):
+            text = re.sub(r'\\boxed\{([^}]+)\}', r'$\\boxed{\1}$', text)
         
     except Exception as e:
-        print(f"⚠️ LaTeX formatting error: {e}")
-        return text  # Return original text if formatting fails
+        print(f"⚠️ LaTeX error: {e}")
     
     return text