Update app.py

app.py

@@ -15,31 +15,31 @@ model = LatexOCR()
 # Clean LaTeX output
 def clean_latex(latex):
     # Replace \mathcal{X} or \cal X with 'x'
-    latex = re.sub(r'\\(cal|mathcal)\\s*X', 'x', latex)
+    latex = re.sub(r'\\(cal|mathcal)\s*X', 'x', latex)
 
     # Remove curly braces
     latex = latex.replace('{', '').replace('}', '')
     latex = latex.strip().rstrip(',.')
-
+    
     # Replace coefficients like 5\pi with (5*3.1416)
-    latex = re.sub(r'(\d+)\\s*\\pi', r'(\1*3.1416)', latex)
+    latex = re.sub(r'(\d+)\s*\\pi', r'(\1*3.1416)', latex)
     latex = latex.replace(r'\pi', '3.1416')
 
     # Replace coefficients like 5e with (5*2.7183)
-    latex = re.sub(r'(\d+)\\s*e', r'(\1*2.7183)', latex)
+    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)
 
     # Insert * between number and variable (e.g., 45x → 45*x)
     latex = re.sub(r'(\d)([a-zA-Z])', r'\1*\2', latex)
 
     # Replace number followed by i with number*I
-    latex = re.sub(r'(\d+)\\s*i', r'\1*I', latex)
+    latex = re.sub(r'(\d+)\s*i', r'\1*I', latex)
 
     # Replace standalone i with I
     latex = re.sub(r'(?<![a-zA-Z0-9])i(?![a-zA-Z0-9])', 'I', latex)
 
     # Wrap complex coefficients with variables: (a+bI)x^n → (a+b*I)*x^n
-    latex = re.sub(r'\\(([^()]+?)\\)\\s*([xX](\\^\d+)?)', r'(\1)*\2', latex)
+    latex = re.sub(r'\(([^()]+?)\)\s*([xX](\^\d+)?)', r'(\1)*\2', latex)
 
     # Fix common OCR mistakes
     latex = re.sub(r'\bL(\d)', r'1\1', latex)
@@ -80,14 +80,11 @@ def solve_polynomial(image):
             expr = fallback_parse(cleaned_latex)
 
         if not expr:
-            return f"❌ Could not parse expression from cleaned LaTeX:\n```
-{cleaned_latex}\n```"
+            return f"❌ Could not parse expression from cleaned LaTeX:\n```\n{cleaned_latex}\n```"
 
-        output = f"## 📄 Extracted LaTeX\n```
-{latex_result}\n```\n"
+        output = f"## 📄 Extracted LaTeX\n```\n{latex_result}\n```\n"
         output += "---\n"
-        output += f"## 🧹 Cleaned LaTeX Used\n```
-{cleaned_latex}\n```\n"
+        output += f"## 🧹 Cleaned LaTeX Used\n```\n{cleaned_latex}\n```\n"
         output += "---\n"
         output += f"## 🧠 Parsed Expression\n\n$$ {sp.latex(expr)} $$\n"
         output += "---\n"
@@ -111,7 +108,7 @@ def solve_polynomial(image):
                 output += "$$\n\\begin{aligned}\n"
                 for i, sol in enumerate(roots, 1):
                     for var, val in sol.items():
-                        output += f"\\text{{Root {i}}}:\\quad {var} &= {sp.latex(val)} \\\\n"
+                        output += f"\\text{{Root {i}}}:\\quad {var} &= {sp.latex(val)} \\\\\n"
                 output += "\\end{aligned}\n$$\n"
 
         else:
@@ -136,4 +133,3 @@ demo = gr.Interface(
 
 if __name__ == "__main__":
     demo.launch()
-