Update app.py

app.py

@@ -56,18 +56,30 @@ def solve_polynomial(image):
         latex_result = model(img)
         cleaned_latex = clean_latex(latex_result)
 
-        expr = parse_latex(cleaned_latex)
-        output = f"## 📄 Extracted LaTeX\n
-
-\n{latex_result}\n
-
-\n"
+        try:
+            expr = parse_latex(cleaned_latex)
+        except Exception:
+            expr = None
+
+        if not expr:
+            expr = fallback_parse(cleaned_latex)
+
+        if not expr:
+            return (
+                "❌ Could not parse expression from cleaned LaTeX:\n"
+                "```latex\n"
+                f"{cleaned_latex}\n"
+                "```"
+            )
+
+        output = (
+            f"## 📄 Extracted LaTeX\n"
+            "```\n"
+            f"{latex_result}\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"