Update app.py

app.py

@@ -23,51 +23,49 @@ def clean_latex(latex):
     # ✅ Fix invalid LaTeX escapes like \+ \- \=
     latex = re.sub(r'\\([+\-=])', r'\1', latex)
 
-    # 🔁 Replace common misrecognized symbols (with and without backslash)
+    # 🔁 Replace common misrecognized symbols
     replacements = {
         r'\\chi': 'x', r'chi': 'x',
         r'\\xi': 'x', r'xi': 'x',
-        r'\\alpha': 'a', r'alpha': 'a',
+        r'\\alpha': 'x', r'alpha': 'x',
         r'\\beta': 'b', r'beta': 'b',
-        r'\\gamma': 'y', r'gamma': 'y'
+        r'\\gamma': 'y', r'gamma': 'y',
+        r'\\vartheta': '3', r'vartheta': '3',
+        r'\\mathcalW': 'x', r'mathcalW': 'x',
     }
     for wrong, correct in replacements.items():
         latex = re.sub(wrong, correct, latex)
 
     # ✅ Fix mathcal and cal variables in all formats
-    latex = re.sub(r'\\(cal|mathcal)\s*\{?\s*X\s*\}?', 'x', latex)
-    latex = re.sub(r'\\(cal|mathcal)\s*\{?\s*Y\s*\}?', 'y', latex)
-    latex = re.sub(r'\\(cal|mathcal)\s*\{?\s*Z\s*\}?', 'z', latex)
+    latex = re.sub(r'\\(cal|mathcal)\s*\{?\s*[Xx]\s*\}?', 'x', latex)
+    latex = re.sub(r'\\(cal|mathcal)\s*\{?\s*[Yy]\s*\}?', 'y', latex)
+    latex = re.sub(r'\\(cal|mathcal)\s*\{?\s*[Zz]\s*\}?', 'z', latex)
 
-    # ✅ Also clean up OCR leftovers like `cal X`, `mathcal x`
-    latex = latex.replace('cal x', 'x')
-    latex = latex.replace('cal X', 'x')
-    latex = latex.replace('mathcal x', 'x')
-    latex = latex.replace('mathcal X', 'x')
+    # ✅ Also clean up OCR leftovers like "cal X", "mathcal x"
+    latex = latex.replace('cal x', 'x').replace('cal X', 'x')
+    latex = latex.replace('mathcal x', 'x').replace('mathcal X', 'x')
 
     # 🧹 Remove curly braces and punctuation
     latex = latex.replace('{', '').replace('}', '')
     latex = latex.strip().rstrip(',.')  # remove trailing punctuation
 
-    # ✅ Replace constants
+    # ✅ Replace constants like pi and e
     latex = re.sub(r'(\d+)\s*\\pi', r'(\1*3.1416)', latex)
     latex = latex.replace(r'\pi', '3.1416')
     latex = re.sub(r'(\d+)\s*e', r'(\1*2.7183)', latex)
     latex = re.sub(r'(?<![a-zA-Z0-9])e(?![a-zA-Z0-9])', '2.7183', latex)
 
-    # ✅ Add multiplication where missing
+    # ✅ Add multiplication
     latex = re.sub(r'(\d)([a-zA-Z])', r'\1*\2', latex)
     latex = re.sub(r'(\d+)\s*i', r'\1*I', latex)
     latex = re.sub(r'(?<![a-zA-Z0-9])i(?![a-zA-Z0-9])', 'I', latex)
-
-    # ✅ (x+1)x → (x+1)*x
     latex = re.sub(r'\(([^()]+?)\)\s*([a-zA-Z](\^\d+)?)', r'(\1)*\2', latex)
 
     # ✅ Fix symbols
     latex = latex.replace(r'\cdot', '*')
     latex = latex.replace('−', '-')  # Unicode minus
 
-    # ✅ Final cleanup: remove any stray garbage symbols
+    # ✅ Final cleanup
     latex = re.sub(r'[^\w\s^=+*\-().]', '', latex)
 
     # ✅ Ensure it's an equation
@@ -77,43 +75,6 @@ def clean_latex(latex):
     return latex
 
 
-    # Replace mathcal variables
-    latex = re.sub(r'\\(cal|mathcal)\s*X', 'x', latex)
-    latex = re.sub(r'\\(cal|mathcal)\s*Y', 'y', latex)
-    latex = re.sub(r'\\(cal|mathcal)\s*Z', 'z', latex)
-
-    # Remove curly braces
-    latex = latex.replace('{', '').replace('}', '')
-    latex = latex.strip().rstrip(',.')  # remove trailing punctuation
-
-    # Replace constants like pi and e
-    latex = re.sub(r'(\d+)\s*\\pi', r'(\1*3.1416)', latex)
-    latex = latex.replace(r'\pi', '3.1416')
-    latex = re.sub(r'(\d+)\s*e', r'(\1*2.7183)', latex)
-    latex = re.sub(r'(?<![a-zA-Z0-9])e(?![a-zA-Z0-9])', '2.7183', latex)
-
-    # Add explicit multiplication between numbers and letters
-    latex = re.sub(r'(\d)([a-zA-Z])', r'\1*\2', latex)
-    latex = re.sub(r'(\d+)\s*i', r'\1*I', latex)
-    latex = re.sub(r'(?<![a-zA-Z0-9])i(?![a-zA-Z0-9])', 'I', latex)
-
-    # Add multiplication for expressions like (x+1)x
-    latex = re.sub(r'\(([^()]+?)\)\s*([a-zA-Z](\^\d+)?)', r'(\1)*\2', latex)
-
-    # Fix other symbols
-    latex = latex.replace(r'\cdot', '*')
-    latex = latex.replace('−', '-')  # Replace Unicode minus
-
-    # 🔧 NEW FIX: Final cleanup - remove stray characters if needed
-    latex = re.sub(r'[^\w\s^=+*\-().]', '', latex)
-
-    # Ensure it's an equation
-    if '=' not in latex:
-        latex += '=0'
-
-    return latex
-
-
 
 
 # Fallback to external LLM microservice