Update app.py

app.py

@@ -8,7 +8,6 @@ import re
 def preprocess_handwritten_image(pil_img):
     return pil_img.convert('RGB')
 
-# Load Pix2Tex model (fine-tuned)
 model = LatexOCR(weights='trained_model')
 
 def clean_latex(latex):
@@ -33,34 +32,28 @@ 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"
-        output += "---\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"
+
+        output = f"## 📄 Extracted LaTeX\n```\n{latex_result}\n```\n---\n"
+        output += f"## 🧹 Cleaned LaTeX Used\n```\n{cleaned_latex}\n```\n---\n"
+        output += f"## 🧠 Parsed Expression\n\n$$ {sp.latex(expr)} $$\n---\n"
 
         if isinstance(expr, sp.Equality):
             lhs = expr.lhs - expr.rhs
-            output += "## ✏️ Step 1: Standard Form of the Polynomial\n"
-            output += f"$$ {sp.latex(lhs)} = 0 $$\n"
-            output += "---\n"
-            output += "## 🧩 Step 2: Factor the Polynomial\n"
-            factored = sp.factor(lhs)
-            output += f"$$ {sp.latex(factored)} = 0 $$\n"
-            output += "---\n"
-            output += "## ✅ Step 3: Solve for Roots\n"
+            output += "## ✏️ Step 1: Standard Form\n"
+            output += f"$$ {sp.latex(lhs)} = 0 $$\n---\n"
+            output += "## 🧩 Step 2: Factorized\n"
+            output += f"$$ {sp.latex(sp.factor(lhs))} = 0 $$\n---\n"
+            output += "## ✅ Step 3: Roots\n"
             roots = sp.solve(sp.Eq(lhs, 0), dict=True)
             if roots:
                 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 += "\\end{aligned}\n$$\n"
+                        output += f"\\text{{Root {i}}}: {var} &= {sp.latex(val)}\\\\\n"
+                output += "\\end{aligned}\n$$"
         else:
-            simplified = sp.simplify(expr)
-            output += "## ➕ Simplified Expression\n"
-            output += f"$$ {sp.latex(simplified)} $$"
+            output += "## ➕ Simplified\n"
+            output += f"$$ {sp.latex(sp.simplify(expr))} $$"
 
         return output
     except Exception as e:
@@ -68,10 +61,10 @@ def solve_polynomial(image):
 
 demo = gr.Interface(
     fn=solve_polynomial,
-    inputs=gr.Image(type="pil", label="📷 Upload Image of Polynomial"),
-    outputs=gr.Markdown(label="📋 Step-by-step Solution"),
-    title="🧠 Polynomial Solver from Image",
-    description="Upload an image of a polynomial (typed or handwritten). The app will extract, solve, and explain it step-by-step.",
+    inputs=gr.Image(type="pil", label="📷 Upload Image"),
+    outputs=gr.Markdown(label="📋 Solution"),
+    title="🧠 Handwritten Polynomial Solver",
+    description="Extract handwritten polynomials and solve step-by-step.",
     allow_flagging="never"
 )