Update app.py

app.py

@@ -36,7 +36,6 @@ def clean_latex(latex):
 
 # Fallback to external LLM microservice
 def request_llm_fallback(bad_latex, llm_url):
-    # Clean known junk before sending to LLM
     pre_cleaned = re.sub(
         r'(\\!?pm|\\not=?|\\!|\\L|\\perp|\\bar|\\Sigma|\\boldmath|G|L(?=\^))', '', bad_latex
     )
@@ -48,6 +47,17 @@ def request_llm_fallback(bad_latex, llm_url):
         print(f"LLM fallback failed: {e}")
     return pre_cleaned
 
+# NEW: Request human-style explanation
+def request_llm_explanation(latex, llm_url):
+    try:
+        response = requests.post(f"{llm_url}/explain", json={"prompt": latex})
+        if response.status_code == 200:
+            return response.json().get("explanation", "❌ No explanation returned.")
+        else:
+            return f"❌ LLM explanation failed: {response.status_code}"
+    except Exception as e:
+        return f"❌ LLM explanation error: {e}"
+
 # Main function
 def solve_polynomial(image, llm_url):
     try:
@@ -55,7 +65,7 @@ def solve_polynomial(image, llm_url):
         latex_result = model(img)
 
         if not latex_result or len(latex_result.strip()) < 2:
-            return "❌ Could not extract valid LaTeX from image."
+            return "❌ Could not extract valid LaTeX from image.", "", ""
 
         cleaned_latex = clean_latex(latex_result)
 
@@ -69,7 +79,6 @@ def solve_polynomial(image, llm_url):
             if not lhs_minus_rhs.is_polynomial():
                 raise ValueError("Expression is not a polynomial.")
 
-            # Fallback if junk symbols found
             junk_symbols = {"pm", "cdot", "mathrm", "boldmath", "bar", "L"}
             symbols = {str(s) for s in expr.free_symbols}
             if symbols & junk_symbols:
@@ -96,8 +105,9 @@ def solve_polynomial(image, llm_url):
 
         if expr is None:
             return (
-                "❌ Could not parse expression even after fallback:\n\n"
-                f"```latex\n{cleaned_latex}\n```"
+                f"❌ Could not parse expression even after fallback:\n\n```latex\n{cleaned_latex}\n```",
+                cleaned_latex,
+                ""
             )
 
         output = (
@@ -122,22 +132,31 @@ def solve_polynomial(image, llm_url):
             simplified = sp.simplify(expr)
             output += f"## ➕ Simplified Expression\n$$ {sp.latex(simplified)} $$"
 
-        return output
+        return output, cleaned_latex, ""
 
     except Exception as e:
-        return f"❌ **Error**: {str(e)}"
+        return f"❌ **Error**: {str(e)}", "", ""
 
-# Gradio UI
+# UI
 with gr.Blocks() as demo:
     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")
     output_box = gr.Markdown(label="📋 Step-by-step Solution")
+    explain_box = gr.Markdown(label="🗣️ Human-style Explanation")
+    hidden_latex = gr.Textbox(visible=False)
+
+    def wrapped_solver(img, url):
+        result, cleaned, _ = solve_polynomial(img, url)
+        return result, cleaned
 
     submit_btn = gr.Button("🔍 Solve")
-    submit_btn.click(fn=solve_polynomial, inputs=[image_input, llm_url], outputs=output_box)
+    submit_btn.click(fn=wrapped_solver, inputs=[image_input, llm_url], outputs=[output_box, hidden_latex])
+
+    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 an image of a polynomial (typed or handwritten). It will be solved step-by-step."
+    demo.description = "Upload a polynomial image (typed or handwritten). View symbolic steps and human-style explanation."
 
 if __name__ == "__main__":
     demo.launch()