Update app.py

app.py

@@ -15,10 +15,10 @@ model = LatexOCR()
 
 # Clean LaTeX output
 def clean_latex(latex):
-    # Fix accidental LaTeX escapes like \+ or \=
+    # ✅ Fix invalid LaTeX escapes like \+ \- \=
     latex = re.sub(r'\\([+\-=])', r'\1', latex)
 
-    # Fix common OCR misreads like \chi → x
+    # 🔁 Replace common misrecognized symbols
     replacements = {
         r'\chi': 'x',
         r'\xi': 'x',
@@ -29,23 +29,37 @@ def clean_latex(latex):
     for wrong, correct in replacements.items():
         latex = latex.replace(wrong, correct)
 
+    # 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(',.')
+    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('−', '-')
+    latex = latex.replace('−', '-')  # Replace Unicode minus
+
+    # Ensure it's an equation
     if '=' not in latex:
         latex += '=0'
+
     return latex