Update app.py

app.py

@@ -422,8 +422,7 @@ STRICT REQUIREMENTS:
    - Print variable values after they are computed
    - Format numerical results using evalf() for decimal
 9. When writing SymPy code to verify solutions:
-   - Start with essential imports: from sympy import * 
-   - Or use specific imports as needed: from sympy import Symbol, solve, limit, Abs, integrate, Function, diff, sin, cos
+   - Start with "from sympy import" and include eall necessary imports
    - DO NOT use NumPy, SciPy, or other libraries
    - For numerical integration, use integrate() instead of scipy.integrate.quad()
    - For any numerical computations, use native SymPy functions
@@ -518,12 +517,37 @@ def extract_and_run_sympy_code_simple(response_text):
         # Set up basic SymPy environment
         import io
         import sympy
-        from contextlib import redirect_stdout
+        from  sympy import (Symbol, symbols, solve, limit, Abs, integrate,
+                         Function, diff, sin, cos, exp, log, sqrt,
+                         Matrix, Derivative, Integral, series)
+        
+        # Create globals dict with all necessary imports
+        globals_dict = {
+            'sympy': sympy,
+            'Symbol': Symbol,
+            'symbols': symbols,
+            'solve': solve,
+            'limit': limit,
+            'Abs': Abs,
+            'integrate': integrate,
+            'Function': Function,
+            'diff': diff,
+            'sin': sin,
+            'cos': cos,
+            'exp': exp,
+            'log': log,
+            'sqrt': sqrt,
+            'Matrix': Matrix,
+            'Derivative': Derivative,
+            'Integral': Integral,
+            'series': series,
+            'print': print
+        }
         
         # Capture output
         output_buffer = io.StringIO()
         with redirect_stdout(output_buffer):
-            exec(sympy_code, {"sympy": sympy, "print": print})
+            exec(sympy_code, globals_dict)
             
         return output_buffer.getvalue().strip() or "No output produced"