Update app.py

app.py

@@ -6,6 +6,9 @@ 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')
@@ -18,6 +21,7 @@ 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)
@@ -36,7 +40,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,7 +51,6 @@ def request_llm_fallback(bad_latex, llm_url):
         print(f"LLM fallback failed: {e}")
     return pre_cleaned
 
-
 # Main function
 def solve_polynomial(image, llm_url):
     try:
@@ -61,9 +63,8 @@ def solve_polynomial(image, llm_url):
         cleaned_latex = clean_latex(latex_result)
 
         try:
-            expr = parse_latex(cleaned_latex)
+            expr = parse_latex(cleaned_latex, local_dict={'x': x, 'y': y, 'z': z})
         
-            # Trigger fallback if not a polynomial
             if not isinstance(expr, sp.Equality):
                 raise ValueError("Expression is not an equation.")
         
@@ -71,18 +72,16 @@ def solve_polynomial(image, llm_url):
             if not lhs_minus_rhs.is_polynomial():
                 raise ValueError("Expression is not a polynomial.")
             
-            # 🔍 Fallback trigger if junk variables 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:
-            # Fallback to LLM microservice
             fixed_latex = request_llm_fallback(cleaned_latex, llm_url)
             cleaned_latex = clean_latex(fixed_latex)
             try:
-                expr = parse_latex(cleaned_latex)
+                expr = parse_latex(cleaned_latex, local_dict={'x': x, 'y': y, 'z': z})
         
                 if not isinstance(expr, sp.Equality):
                     raise ValueError("Expression is not an equation.")
@@ -91,7 +90,6 @@ def solve_polynomial(image, llm_url):
                 if not lhs_minus_rhs.is_polynomial():
                     raise ValueError("Expression is not a polynomial.")
                 
-                # 🔁 Fallback again if junk remains
                 symbols = {str(s) for s in expr.free_symbols}
                 if symbols & junk_symbols:
                     raise ValueError("Expression contains junk symbols.")