Update app.py

app.py

@@ -6,9 +6,6 @@ from sympy.parsing.latex import parse_latex
 import re
 import requests
 
-# ✅ Predeclare common variables
-x, y, z = sp.symbols('x y z')
-
 # Preprocessing
 def preprocess_handwritten_image(pil_img):
     return pil_img.convert('RGB')
@@ -21,7 +18,6 @@ def clean_latex(latex):
     latex = re.sub(r'\\(cal|mathcal)\s*X', 'x', latex)
     latex = re.sub(r'\\(cal|mathcal)\s*Y', 'y', latex)
     latex = re.sub(r'\\(cal|mathcal)\s*Z', 'z', latex)
-    latex = re.sub(r'\\(hat|bf|tau|enspace)', '', latex)
     latex = latex.replace('{', '').replace('}', '')
     latex = latex.strip().rstrip(',.')
     latex = re.sub(r'(\d+)\s*\\pi', r'(\1*3.1416)', latex)
@@ -40,6 +36,7 @@ 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
     )
@@ -63,36 +60,37 @@ def solve_polynomial(image, llm_url):
         cleaned_latex = clean_latex(latex_result)
 
         try:
-            expr = parse_latex(cleaned_latex, local_dict={'x': x, 'y': y, 'z': z})
-        
+            expr = parse_latex(cleaned_latex)
+
             if not isinstance(expr, sp.Equality):
                 raise ValueError("Expression is not an equation.")
-        
+
             lhs_minus_rhs = expr.lhs - expr.rhs
             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:
                 raise ValueError("Expression contains junk symbols.")
-        
+
         except Exception:
             fixed_latex = request_llm_fallback(cleaned_latex, llm_url)
             cleaned_latex = clean_latex(fixed_latex)
             try:
-                expr = parse_latex(cleaned_latex, local_dict={'x': x, 'y': y, 'z': z})
-        
+                expr = parse_latex(cleaned_latex)
                 if not isinstance(expr, sp.Equality):
                     raise ValueError("Expression is not an equation.")
-        
+
                 lhs_minus_rhs = expr.lhs - expr.rhs
                 if not lhs_minus_rhs.is_polynomial():
                     raise ValueError("Expression is not a polynomial.")
-                
+
                 symbols = {str(s) for s in expr.free_symbols}
                 if symbols & junk_symbols:
                     raise ValueError("Expression contains junk symbols.")
+
             except Exception:
                 expr = None
 
@@ -134,13 +132,9 @@ 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")
-    state = gr.State()
-
-    def wrapped_solver(img, url):
-        return solve_polynomial(img, url)
 
     submit_btn = gr.Button("🔍 Solve")
-    submit_btn.click(fn=wrapped_solver, inputs=[image_input, llm_url], outputs=output_box)
+    submit_btn.click(fn=solve_polynomial, inputs=[image_input, llm_url], outputs=output_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."