Update app.py

app.py

@@ -346,14 +346,14 @@ def extract_and_run_sympy_code(response_text):
         logger.debug(f"Code after removing imports:\n{sympy_code}")
         
         # Create globals dict with all SymPy symbols
-        globals_dict = {
-            '__builtins__': __builtins__,
-            'sympy': sympy,  # Add the sympy module itself
-        }
+        import sys
+        globals_dict = {}
+        
+        # Add built-ins
+        globals_dict['__builtins__'] = __builtins__
         
-        # Add all sympy attributes to globals
-        for name in dir(sympy):
-            globals_dict[name] = getattr(sympy, name)
+        # Add sympy itself by loading it from sys.modules
+        globals_dict.update({name: value for name, value in globals().items() if name in dir(sys.modules['sympy'])})
         
         # Create locals dict to store variables
         local_vars = {}
@@ -380,7 +380,7 @@ def extract_and_run_sympy_code(response_text):
         import traceback
         logger.error(f"Full traceback:\n{traceback.format_exc()}")
         return f"Error executing SymPy code: {str(e)}"
-        
+
         
 # Create Gradio interface
 with gr.Blocks() as interface: