Update app.py

app.py

@@ -19,7 +19,6 @@ def clean_latex(latex):
     latex = latex.replace('\\\\', '\\')
     latex = re.sub(r'\\[ \t\n\r\f\v]*', '', latex)
     latex = re.sub(r'\\([+\-=])', r'\1', latex)
-
     replacements = {
         r'\\chi': 'x', r'chi': 'x',
         r'\\xi': 'x', r'xi': 'x',
@@ -33,14 +32,11 @@ def clean_latex(latex):
     }
     for wrong, correct in replacements.items():
         latex = re.sub(wrong, correct, latex)
-
     latex = re.sub(r'\\(cal|mathcal)\s*\{?\s*[Xx]\s*\}?', 'x', latex)
     latex = re.sub(r'\\(cal|mathcal)\s*\{?\s*[Yy]\s*\}?', 'y', latex)
     latex = re.sub(r'\\(cal|mathcal)\s*\{?\s*[Zz]\s*\}?', 'z', latex)
-
     latex = latex.replace('cal x', 'x').replace('cal X', 'x')
     latex = latex.replace('mathcal x', 'x').replace('mathcal X', 'x')
-
     latex = latex.replace('{', '').replace('}', '')
     latex = latex.strip().rstrip(',.')
     latex = re.sub(r'(?<![a-zA-Z0-9])e(?![a-zA-Z0-9])', 'E', latex)
@@ -48,23 +44,17 @@ def clean_latex(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)
     latex = re.sub(r'\(([^()]+?)\)\s*([a-zA-Z](\^\d+)?)', r'(\1)*\2', latex)
-
     latex = latex.replace(r'\cdot', '*')
-    latex = latex.replace('−', '-')  # Unicode minus
+    latex = latex.replace('−', '-')
     latex = re.sub(r'[^\w\s^=+*\-().]', '', latex)
-
     if '=' not in latex:
         latex += '=0'
-
     latex = latex.replace('pi', '3.1416')
     latex = latex.replace('e', '2.7183')
-
     return latex
 
 def request_llm_fallback(bad_latex, llm_url):
-    pre_cleaned = re.sub(
-        r'(\\!?pm|\\not=?|\\!|\\L|\\perp|\\bar|\\Sigma|\\boldmath|G|L(?=\^))', '', bad_latex
-    )
+    pre_cleaned = re.sub(r'(\\!?pm|\\not=?|\\!|\\L|\\perp|\\bar|\\Sigma|\\boldmath|G|L(?=\^))', '', bad_latex)
     try:
         response = requests.post(f"{llm_url}/clean", json={"prompt": pre_cleaned})
         if response.status_code == 200:
@@ -87,16 +77,12 @@ def solve_polynomial(image, llm_url):
     try:
         img = preprocess_handwritten_image(image)
         latex_result = model(img)
-
         if not latex_result or len(latex_result.strip()) < 2:
             return "❌ Could not extract valid LaTeX from image.", "", ""
-
         cleaned_latex = clean_latex(latex_result)
-
         try:
             expr = parse_latex(cleaned_latex)
             expr = expr.subs(sp.pi, sp.Float(3.1416)).subs(sp.E, sp.Float(2.7183))
-
             if not isinstance(expr, sp.Equality):
                 raise ValueError("Expression is not an equation.")
             lhs_minus_rhs = expr.lhs - expr.rhs
@@ -112,7 +98,6 @@ def solve_polynomial(image, llm_url):
             try:
                 expr = parse_latex(cleaned_latex)
                 expr = expr.subs(sp.pi, sp.Float(3.1416)).subs(sp.E, sp.Float(2.7183))
-
                 if not isinstance(expr, sp.Equality):
                     raise ValueError("Expression is not an equation.")
                 lhs_minus_rhs = expr.lhs - expr.rhs
@@ -123,25 +108,17 @@ def solve_polynomial(image, llm_url):
                     raise ValueError("Expression contains junk symbols.")
             except Exception:
                 expr = None
-
         if expr is None:
-            return (
-                f"❌ Could not parse expression even after fallback:\n\n```latex\n{cleaned_latex}\n```",
-                cleaned_latex,
-                ""
-            )
-
+            return f"❌ Could not parse expression even after fallback:\n\n```latex\n{cleaned_latex}\n```", cleaned_latex, ""
         output = (
             f"## 📄 Extracted LaTeX\n\n```latex\n{latex_result}\n```\n"
             f"\n---\n## 🧹 Cleaned LaTeX Used\n\n```latex\n{cleaned_latex}\n```\n"
             f"\n---\n## 🧠 Parsed Expression\n\n$$ {sp.latex(expr)} $$\n\n---\n"
         )
-
         if isinstance(expr, sp.Equality):
             lhs = expr.lhs - expr.rhs
             output += f"## ✏️ Step 1: Standard Form\n$$ {sp.latex(lhs)} = 0 $$\n---\n"
             output += f"## 🧩 Step 2: Factor\n$$ {sp.latex(sp.factor(lhs))} = 0 $$\n---\n"
-
             x_sym = None
             for sym in lhs.free_symbols:
                 if str(sym) == "x":
@@ -149,9 +126,7 @@ def solve_polynomial(image, llm_url):
                     break
             if x_sym is None:
                 raise ValueError("❌ Could not find variable 'x' in the equation.")
-
             roots = sp.solve(sp.Eq(lhs, 0), x_sym, dict=True)
-
             output += "## ✅ Step 3: Solve Roots\n"
             if roots:
                 output += "$$\n\\begin{aligned}\n"
@@ -163,9 +138,7 @@ def solve_polynomial(image, llm_url):
         else:
             simplified = sp.simplify(expr)
             output += f"## ➕ Simplified Expression\n$$ {sp.latex(simplified)} $$"
-
         return output, cleaned_latex, ""
-
     except Exception as e:
         return f"❌ **Error**: {str(e)}", "", ""
 
@@ -173,49 +146,46 @@ def wrapped_solver(img, url):
     result, cleaned, _ = solve_polynomial(img, url)
     return result, cleaned
 
+def solve_from_coeffs(degree, coeff_str):
+    try:
+        coeffs = [float(c.strip()) for c in coeff_str.split(",")]
+        if len(coeffs) != int(degree) + 1:
+            return f"❌ You entered {len(coeffs)} coefficients, but degree {degree} needs {int(degree)+1}.", ""
+        x = sp.Symbol("x")
+        poly_expr = sum(coeffs[i] * x**(int(degree) - i) for i in range(len(coeffs)))
+        latex = sp.latex(sp.Eq(poly_expr, 0))
+        roots = sp.solve(poly_expr, x)
+        result = "## 🧮 Symbolic Roots:\n"
+        for i, r in enumerate(roots, 1):
+            result += f"- Root {i}: $${sp.latex(sp.N(r, 6))}$$\n"
+        return result, latex
+    except Exception as e:
+        return f"❌ Error: {str(e)}", ""
+
 with gr.Blocks() as demo:
     with gr.Tab("🖼️ Parse from Image"):
         llm_url = gr.Textbox(label="🔗 Enter LLM Microservice URL (from Colab)", placeholder="https://xxxx.ngrok-free.app")
         image_input = gr.Image(type="pil", label="📷 Upload Image of Polynomial")
         hidden_latex = gr.Textbox(visible=False)
-
         output_box = gr.Markdown(label="📋 Step-by-step Solution")
         submit_btn = gr.Button("🔍 Solve")
         submit_btn.click(fn=wrapped_solver, inputs=[image_input, llm_url], outputs=[output_box, hidden_latex])
-
         explain_box = gr.Markdown(label="🗣️ Human-style Explanation")
         explain_btn = gr.Button("🧠 Explain Human-Solution")
         explain_btn.click(fn=request_llm_explanation, inputs=[hidden_latex, llm_url], outputs=explain_box)
-
         demo.title = "🧠 Polynomial Solver from Image"
         demo.description = "Upload a polynomial image (typed or handwritten). View symbolic steps and human-style explanation."
 
-with demo:
     with gr.Tab("🧮 Solve by Coefficients"):
         degree_input = gr.Number(label="Enter Degree of Polynomial (e.g. 3)")
         coeffs_input = gr.Textbox(label="Enter Coefficients (comma-separated)", placeholder="e.g. 1, 2, 0, -4")
         roots_output = gr.Markdown(label="✅ Roots")
-
-        def solve_from_coeffs(degree, coeff_str):
-            try:
-                coeffs = [float(c.strip()) for c in coeff_str.split(",")]
-                if len(coeffs) != int(degree) + 1:
-                    return f"❌ You entered {len(coeffs)} coefficients, but degree {degree} needs {int(degree)+1}."
-
-                x = sp.Symbol("x")
-                poly_expr = sum(coeffs[i] * x**(int(degree) - i) for i in range(len(coeffs)))
-                roots = sp.solve(poly_expr, x)
-
-                result = "### Roots:\n"
-                for i, r in enumerate(roots, 1):
-                    result += f"- Root {i}: $${sp.latex(sp.N(r, 6))}$$\n"
-                return result
-            except Exception as e:
-                return f"❌ Error: {str(e)}"
-
+        coeff_hidden_latex = gr.Textbox(visible=False)
         coeff_btn = gr.Button("📌 Solve")
-        coeff_btn.click(fn=solve_from_coeffs, inputs=[degree_input, coeffs_input], outputs=roots_output)
+        coeff_btn.click(fn=solve_from_coeffs, inputs=[degree_input, coeffs_input], outputs=[roots_output, coeff_hidden_latex])
+        coeff_explain_box = gr.Markdown(label="🗣️ Human-style Explanation")
+        coeff_explain_btn = gr.Button("🧠 Explain Human-Solution")
+        coeff_explain_btn.click(fn=request_llm_explanation, inputs=[coeff_hidden_latex, llm_url], outputs=coeff_explain_box)
 
 if __name__ == "__main__":
     demo.launch()
-