Upload app.py

app.py

@@ -5,9 +5,8 @@ import anthropic
 from sympy import (
     symbols, diff, integrate, limit, solve, simplify, expand, factor,
     latex, sympify, oo, sin, cos, tan, exp, log, sqrt, pi, E,
-    Matrix, det, eigenvals, Symbol, Function, dsolve, Eq,
-    series, Sum, Product, Rational, Integer, Float, Abs,
-    standard_transformations
+    Matrix, Symbol, Function, dsolve, Eq,
+    series, Sum, Product, Rational, Integer, Float, Abs
 )
 from sympy.parsing.sympy_parser import (
     parse_expr,
@@ -249,7 +248,7 @@ def run_sympy(problem: str) -> dict:
                     raw = p.split(kw, 1)[-1].strip()
                     break
             raw = re.sub(r"\s*(dx|with\s*respect\s*to\s*x).*$", "", raw).strip()
-            expr = parse_expr(clean(raw), transformations=tfms, local_dict={"x": x})
+            expr = parse_expr(clean(raw), transformations=tfms, local_dict={"x": x, "e": sp.E, "E": sp.E, "pi": sp.pi, "ln": sp.log, "inf": sp.oo})
             result = sp.diff(expr, x)
             return {"type": "Derivative", "result": str(result), "latex": sp.latex(result)}
 
@@ -261,7 +260,7 @@ def run_sympy(problem: str) -> dict:
                     raw = p.split(kw, 1)[-1].strip()
                     break
             raw = re.sub(r"\s*dx.*$", "", raw).strip()
-            expr = parse_expr(clean(raw), transformations=tfms, local_dict={"x": x})
+            expr = parse_expr(clean(raw), transformations=tfms, local_dict={"x": x, "e": sp.E, "E": sp.E, "pi": sp.pi, "ln": sp.log, "inf": sp.oo})
             result = sp.integrate(expr, x)
             return {"type": "Integral", "result": str(result), "latex": sp.latex(result)}
 
@@ -275,7 +274,7 @@ def run_sympy(problem: str) -> dict:
                 raw_expr = clean(match.group(1))
                 point_str = match.group(2).strip()
                 point = sp.oo if point_str in ("inf", "infinity", "oo") else sp.sympify(point_str)
-                expr = parse_expr(raw_expr, transformations=tfms, local_dict={"x": x})
+                expr = parse_expr(raw_expr, transformations=tfms, local_dict={"x": x, "e": sp.E, "E": sp.E, "pi": sp.pi, "ln": sp.log, "inf": sp.oo})
                 result = sp.limit(expr, x, point)
                 return {"type": "Limit", "result": str(result), "latex": sp.latex(result)}
 
@@ -284,11 +283,11 @@ def run_sympy(problem: str) -> dict:
             raw = re.sub(r"(solve|find x|roots of|roots|the equation)", "", p).strip().strip(": ")
             if "=" in raw:
                 lhs_s, rhs_s = raw.split("=", 1)
-                lhs_e = parse_expr(clean(lhs_s), transformations=tfms, local_dict={"x": x})
-                rhs_e = parse_expr(clean(rhs_s), transformations=tfms, local_dict={"x": x})
+                lhs_e = parse_expr(clean(lhs_s), transformations=tfms, local_dict={"x": x, "e": sp.E, "E": sp.E, "pi": sp.pi, "ln": sp.log, "inf": sp.oo})
+                rhs_e = parse_expr(clean(rhs_s), transformations=tfms, local_dict={"x": x, "e": sp.E, "E": sp.E, "pi": sp.pi, "ln": sp.log, "inf": sp.oo})
                 expr = lhs_e - rhs_e
             else:
-                expr = parse_expr(clean(raw), transformations=tfms, local_dict={"x": x})
+                expr = parse_expr(clean(raw), transformations=tfms, local_dict={"x": x, "e": sp.E, "E": sp.E, "pi": sp.pi, "ln": sp.log, "inf": sp.oo})
             if x in expr.free_symbols:
                 sol = sp.solve(expr, x)
                 sol_latex = ", ".join([sp.latex(s) for s in sol])
@@ -509,4 +508,4 @@ if send_btn and problem and problem != st.session_state.last_submitted:
 
     st.session_state.messages.append({"role": "user", "content": problem})
     st.session_state.messages.append({"role": "assistant", "content": answer})
-    st.rerun()
+    st.rerun()