Update app.py

app.py

@@ -85,8 +85,13 @@ def solve_polynomial(image, llm_url):
 
         try:
             expr = parse_latex(cleaned_latex)
-            expr = expr.replace(sp.pi, sp.Float(3.1416))
-            expr = expr.replace(sp.E, sp.Float(2.7183))
+
+            # ✅ Replace pi and e by name
+            for symbol in expr.free_symbols:
+                if str(symbol) == 'pi':
+                    expr = expr.subs(symbol, sp.Float(3.1416))
+                elif str(symbol) == 'e':
+                    expr = expr.subs(symbol, sp.Float(2.7183))
 
             if not isinstance(expr, sp.Equality):
                 raise ValueError("Expression is not an equation.")
@@ -104,8 +109,13 @@ def solve_polynomial(image, llm_url):
             cleaned_latex = clean_latex(fixed_latex)
             try:
                 expr = parse_latex(cleaned_latex)
-                expr = expr.replace(sp.pi, sp.Float(3.1416))
-                expr = expr.replace(sp.E, sp.Float(2.7183))
+
+                # ✅ Repeat fix in fallback too
+                for symbol in expr.free_symbols:
+                    if str(symbol) == 'pi':
+                        expr = expr.subs(symbol, sp.Float(3.1416))
+                    elif str(symbol) == 'e':
+                        expr = expr.subs(symbol, sp.Float(2.7183))
 
                 if not isinstance(expr, sp.Equality):
                     raise ValueError("Expression is not an equation.")