Update app.py

app.py

@@ -346,28 +346,32 @@ def extract_and_run_sympy_code(response_text):
         logger.debug(f"Code after removing imports:\n{sympy_code}")
         
         # Create globals dict
-        sympy_globals = {
-            'sympy': sympy,
-            '__builtins__': __builtins__
+        import sys
+        module = sys.modules[__name__]  # Get the current module
+        globals_dict = {
+            '__builtins__': __builtins__,
         }
         
-        # Add all sympy attributes
-        for name in dir(sympy):
-            sympy_globals[name] = getattr(sympy, name)
+        # Add all imported SymPy names to globals
+        for name in dir(module):
+            if not name.startswith('_'):  # Skip private names
+                globals_dict[name] = getattr(module, name)
         
-        logger.debug(f"Global namespace keys: {list(sympy_globals.keys())}")
+        logger.debug(f"Global namespace keys: {list(globals_dict.keys())}")
         
         # Create locals dict and output buffer
         local_vars = {}
+        import io
+        from contextlib import redirect_stdout
         output_buffer = io.StringIO()
         
         logger.debug("Attempting to execute code")
         # Execute with try/except to catch specific error
         try:
-            exec(sympy_code, sympy_globals, local_vars)
+            exec(sympy_code, globals_dict, local_vars)
         except NameError as e:
             logger.error(f"NameError during execution: {str(e)}")
-            # Log the actual error traceback
+            logger.debug(f"Available names: {list(globals_dict.keys())}")
             import traceback
             logger.error(f"Full traceback:\n{traceback.format_exc()}")
             return f"Error executing SymPy code: {str(e)}"
@@ -384,8 +388,7 @@ def extract_and_run_sympy_code(response_text):
             print("-" * 25)
             for var_name, value in local_vars.items():
                 if (not var_name.startswith('__') and 
-                    not hasattr(value, '__module__') and 
-                    var_name != 'sympy'):
+                    not hasattr(value, '__module__')):
                     print(f"{var_name}: {value}")
         
         result = output_buffer.getvalue()