Update app.py

app.py

@@ -9,6 +9,7 @@ import re
 def preprocess_handwritten_image(pil_img):
     return pil_img.convert('RGB')  # Minimal processing, just convert to RGB
 
+
 # Load Pix2Tex model (once)
 model = LatexOCR()
 
@@ -41,28 +42,12 @@ def clean_latex(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)
 
-    # Fix common OCR mistakes
-    latex = re.sub(r'\bL(\d)', r'1\1', latex)
-    latex = latex.replace('\\pm', '').replace('!', '')
-    latex = re.sub(r'[^\w\s\^\+\-\*/=xX\(\)\.\*]', '', latex)
-
     # Append '=0' if not already present
     if '=' not in latex:
         latex += '=0'
 
     return latex
 
-# Fallback parser if LaTeX parsing fails
-def fallback_parse(cleaned_latex):
-    try:
-        if '=' in cleaned_latex:
-            cleaned_latex = cleaned_latex.split('=')[0]
-        cleaned_latex = cleaned_latex.replace('^', '**')
-        x = sp.symbols('x')
-        expr = eval(cleaned_latex, {"x": x, "I": sp.I, "pi": sp.pi})
-        return sp.Eq(expr, 0)
-    except Exception:
-        return None
 
 # Main function
 def solve_polynomial(image):
@@ -71,20 +56,18 @@ def solve_polynomial(image):
         latex_result = model(img)
         cleaned_latex = clean_latex(latex_result)
 
-        try:
-            expr = parse_latex(cleaned_latex)
-        except Exception:
-            expr = None
-
-        if not expr:
-            expr = fallback_parse(cleaned_latex)
+        expr = parse_latex(cleaned_latex)
+        output = f"## 📄 Extracted LaTeX\n
 
-        if not expr:
-            return f"❌ Could not parse expression from cleaned LaTeX:\n```\n{cleaned_latex}\n```"
+\n{latex_result}\n
 
-        output = f"## 📄 Extracted LaTeX\n```\n{latex_result}\n```\n"
+\n"
         output += "---\n"
-        output += f"## 🧹 Cleaned LaTeX Used\n```\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"
@@ -121,6 +104,7 @@ def solve_polynomial(image):
     except Exception as e:
         return f"❌ **Error**: {str(e)}"
 
+
 # Gradio UI
 demo = gr.Interface(
     fn=solve_polynomial,