Update app.py

app.py

@@ -345,9 +345,10 @@ def extract_and_run_sympy_code(response_text):
         # Create string buffer for output
         output_buffer = io.StringIO()
         
-        # Create globals dict with all SymPy symbols
-        sympy_globals = {}
+        # Create globals dict with all SymPy symbols and add sympy module itself
+        sympy_globals = {'sympy': sympy}  # Add sympy module itself
         sympy_globals.update({name: getattr(sympy, name) for name in dir(sympy)})
+        sympy_globals['__builtins__'] = __builtins__  # Add built-ins
         
         # Create locals dict to capture new variables
         local_vars = {}
@@ -360,7 +361,9 @@ def extract_and_run_sympy_code(response_text):
             print("SymPy Calculation Results:")
             print("-" * 25)
             for var_name, value in local_vars.items():
-                if not var_name.startswith('__') and not hasattr(value, '__module__'):
+                if (not var_name.startswith('__') and 
+                    not hasattr(value, '__module__') and 
+                    var_name != 'sympy'):  # Don't print the sympy module itself
                     print(f"{var_name}: {value}")
         
         return output_buffer.getvalue()