Update app.py

app.py

@@ -1,53 +1,62 @@
-# app.py
 import gradio as gr
 import numpy as np
 import matplotlib.pyplot as plt
 import sympy as sp
-import yaml
 import requests
+import yaml
 
 x, y = sp.symbols('x y')
 
-# Load theorems database
-def load_theorems(file="theorems.yaml"):
-    with open(file, 'r') as f:
-        return yaml.safe_load(f)
-
-theorems_db = load_theorems()
-
-# Helper to construct LLM explanation prompt
-def build_llm_prompt(context, eq_type="polynomial"):
-    selected = [t for t in theorems_db if ("polynomial" in t["tags"] if eq_type == "polynomial" else "linear" in t["tags"])]
-    theory_context = "\n".join([f"{t['name']}: {t['short_explanation']}" for t in selected])
-    return f"Context:\n{theory_context}\n\nQuestion:\nExplain how the following was solved:\n{context}"
-
-# Call LLM microservice
-def get_llm_explanation(context, url, eq_type):
+# Load theorem database
+def load_theorem_db():
+    try:
+        with open("theorems.yaml", "r") as f:
+            return yaml.safe_load(f)
+    except:
+        return []
+
+def generate_context(task_type):
+    theorems = load_theorem_db()
+    relevant = []
+    if task_type == "polynomial":
+        relevant = [t for t in theorems if "polynomial" in t["tags"]]
+    elif task_type == "linear":
+        relevant = [t for t in theorems if "linear" in t["tags"]]
+    context = ""
+    for t in relevant:
+        context += f"Theorem: {t['name']}\nExplanation: {t['short_explanation']}\n\n"
+    return context
+
+def explain_with_llm(problem, task_type, llm_url):
     try:
-        prompt = build_llm_prompt(context, eq_type)
-        resp = requests.post(f"{url}/explain", json={"prompt": prompt})
-        if resp.status_code == 200:
-            return resp.json().get("explanation", "❌ No explanation returned.")
-        return f"❌ LLM returned status {resp.status_code}"
+        context = generate_context(task_type)
+        payload = {
+            "prompt": f"Using the theorems below, explain this math solution in depth:\n\n{context}\n\nProblem: {problem}"
+        }
+        if llm_url.strip() == "":
+            raise ValueError("No LLM URL provided.")
+        response = requests.post(f"{llm_url}/explain", json=payload)
+        if response.status_code == 200:
+            return response.json().get("explanation", "✅ Processed but no explanation returned.")
+        return f"❌ LLM request failed: {response.status_code}"
     except Exception as e:
         return f"❌ LLM request failed: {e}"
 
-# Generate polynomial template
+# Polynomial Solver
+
 def generate_polynomial_template(degree):
     terms = [f"a{i}*x^{degree - i}" for i in range(degree)] + [f"a{degree}"]
     return " + ".join(terms) + " = 0"
 
-# Solve and plot polynomial
 def solve_polynomial(degree, coeff_string):
     try:
         coeffs = [sp.sympify(s) for s in coeff_string.strip().split()]
         if len(coeffs) != degree + 1:
-            return f"⚠️ Please enter exactly {degree + 1} coefficients.", None, None
+            return f"⚠️ Please enter exactly {degree + 1} coefficients.", None, None, ""
 
         poly = sum([coeffs[i] * x**(degree - i) for i in range(degree + 1)])
         simplified = sp.simplify(poly)
 
-        # Factor step-by-step
         factored_steps = []
         current_expr = simplified
         while True:
@@ -94,12 +103,13 @@ def solve_polynomial(degree, coeff_string):
 
         ax.legend()
 
-        return steps_output, fig, steps_output
+        return steps_output, fig, "", steps_output
 
     except Exception as e:
-        return f"❌ Error: {e}", None, ""
+        return f"❌ Error: {e}", None, "", ""
+
+# Linear Solver
 
-# Solve linear system
 def solve_linear_system(eq1_str, eq2_str):
     try:
         eq1 = sp.sympify(eq1_str)
@@ -147,44 +157,43 @@ def solve_linear_system(eq1_str, eq2_str):
     except Exception as e:
         return f"❌ Error: {e}", None, ""
 
-# UI
-def build_ui():
-    with gr.Blocks() as demo:
-        gr.Markdown("## 🔢 Polynomial & Linear System Solver with Step-by-Step Explanation")
+with gr.Blocks() as demo:
+    gr.Markdown("## 🔢 Polynomial Solver with Step-by-Step Factorization and Graph")
+
+    with gr.Row():
+        degree_slider = gr.Slider(1, 8, value=3, step=1, label="Degree of Polynomial")
+        template_display = gr.Textbox(label="Polynomial Template (Fill in Coefficients)", interactive=False)
+
+    coeff_input = gr.Textbox(label="Enter Coefficients (space-separated, supports pi, e, sqrt(2), I)", placeholder="e.g. 1 -3 sqrt(2) -pi")
+    llm_url = gr.Textbox(label="LLM Microservice URL (optional)", placeholder="https://your-llm.ngrok.app")
 
-        with gr.Tab("Polynomial"):
-            with gr.Row():
-                degree_slider = gr.Slider(1, 8, value=3, step=1, label="Degree")
-                template_display = gr.Textbox(label="Template", interactive=False)
+    steps_md = gr.Markdown()
+    plot_output = gr.Plot()
+    error_box = gr.Textbox(visible=False)
+    poly_full_solution = gr.Textbox(visible=False)
 
-            coeff_input = gr.Textbox(label="Coefficients", placeholder="e.g. 1 -3 sqrt(2) -pi")
-            steps_md = gr.Markdown()
-            plot_output = gr.Plot()
-            error_box = gr.Textbox(visible=False)
-            explanation_md = gr.Markdown()
-            llm_url = gr.Textbox(label="LLM URL", placeholder="https://your-llm.ngrok-free.app")
+    with gr.Row():
+        solve_button = gr.Button("Plot Polynomial", variant="primary")
+        explain_poly = gr.Button("Explain Polynomial with LLM")
 
-            degree_slider.change(generate_polynomial_template, degree_slider, template_display)
-            solve_btn = gr.Button("Solve Polynomial", variant="primary")
-            solve_btn.click(solve_polynomial, [degree_slider, coeff_input], [steps_md, plot_output, error_box])
+    degree_slider.change(fn=generate_polynomial_template, inputs=degree_slider, outputs=template_display)
+    solve_button.click(fn=solve_polynomial, inputs=[degree_slider, coeff_input], outputs=[steps_md, plot_output, error_box, poly_full_solution])
+    explain_poly.click(fn=lambda sol, url: explain_with_llm(sol, "polynomial", url), inputs=[poly_full_solution, llm_url], outputs=steps_md)
 
-            explain_btn = gr.Button("Explain Polynomial Solution", variant="primary")
-            explain_btn.click(lambda context, url: get_llm_explanation(context, url, "polynomial"), [steps_md, llm_url], explanation_md)
+    gr.Markdown("## 📐 Solve Linear System (2 Equations, 2 Variables)")
 
-        with gr.Tab("Linear System"):
-            eq1_input = gr.Textbox(label="Equation 1", placeholder="2*x + 3*y - 6")
-            eq2_input = gr.Textbox(label="Equation 2", placeholder="-x + y - 2")
-            sys_steps = gr.Markdown()
-            sys_plot = gr.Plot()
-            sys_explain = gr.Markdown()
+    eq1_input = gr.Textbox(label="Equation 1 (in x and y)", placeholder="e.g. 2*x + 3*y - 6")
+    eq2_input = gr.Textbox(label="Equation 2 (in x and y)", placeholder="e.g. -x + y - 2")
+    sys_steps = gr.Markdown()
+    sys_plot = gr.Plot()
+    linear_full_solution = gr.Textbox(visible=False)
 
-            solve_sys_btn = gr.Button("Solve Linear System", variant="primary")
-            solve_sys_btn.click(solve_linear_system, [eq1_input, eq2_input], [sys_steps, sys_plot])
+    with gr.Row():
+        solve_sys_button = gr.Button("Solve Linear System", variant="primary")
+        explain_linear = gr.Button("Explain Linear System with LLM")
 
-            explain_sys_btn = gr.Button("Explain Linear Solution", variant="primary")
-            explain_sys_btn.click(lambda context, url: get_llm_explanation(context, url, "linear"), [sys_steps, llm_url], sys_explain)
+    solve_sys_button.click(fn=solve_linear_system, inputs=[eq1_input, eq2_input], outputs=[sys_steps, sys_plot, linear_full_solution])
+    explain_linear.click(fn=lambda sol, url: explain_with_llm(sol, "linear", url), inputs=[linear_full_solution, llm_url], outputs=sys_steps)
 
-        return demo
+    demo.launch()
 
-if __name__ == "__main__":
-    build_ui().launch()