Update app.py

app.py

@@ -15,31 +15,67 @@ model = LatexOCR()
 
 # Clean LaTeX output
 def clean_latex(latex):
-    # 🔧 NEW FIX: Remove bad spacing escapes and redundant slashes
-    latex = latex.replace('\\ ', '')             # Remove space escapes
-    latex = latex.replace('\\\\', '\\')          # Reduce double slashes
-    latex = re.sub(r'\\[ \t\n\r\f\v]*', '', latex)  # Remove \ followed by whitespace or control chars
+    # 🔧 Remove bad spacing escapes and redundant slashes
+    latex = latex.replace('\\ ', '')
+    latex = latex.replace('\\\\', '\\')
+    latex = re.sub(r'\\[ \t\n\r\f\v]*', '', latex)
 
     # ✅ Fix invalid LaTeX escapes like \+ \- \=
     latex = re.sub(r'\\([+\-=])', r'\1', latex)
 
-    # 🔁 Replace common misrecognized symbols
     # 🔁 Replace common misrecognized symbols (with and without backslash)
     replacements = {
-        r'\\chi': 'x',  # With escape
-        r'chi': 'x',    # After \chi becomes plain
-        r'\\xi': 'x',
-        r'xi': 'x',
-        r'\\alpha': 'a',
-        r'alpha': 'a',
-        r'\\beta': 'b',
-        r'beta': 'b',
-        r'\\gamma': 'y',
-        r'gamma': 'y'
+        r'\\chi': 'x', r'chi': 'x',
+        r'\\xi': 'x', r'xi': 'x',
+        r'\\alpha': 'a', r'alpha': 'a',
+        r'\\beta': 'b', r'beta': 'b',
+        r'\\gamma': 'y', r'gamma': 'y'
     }
     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)
+
+    # ✅ 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')
+
+    # 🧹 Remove curly braces and punctuation
+    latex = latex.replace('{', '').replace('}', '')
+    latex = latex.strip().rstrip(',.')  # remove trailing punctuation
+
+    # ✅ Replace constants
+    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
+    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
+    latex = re.sub(r'[^\w\s^=+*\-().]', '', latex)
+
+    # ✅ Ensure it's an equation
+    if '=' not in latex:
+        latex += '=0'
+
+    return latex
+
 
     # Replace mathcal variables
     latex = re.sub(r'\\(cal|mathcal)\s*X', 'x', latex)