Update app.py

app.py

@@ -347,7 +347,7 @@ def extract_and_run_sympy_code(response_text):
         
         # Create globals dict
         import sys
-        module = sys.modules[__name__]  # Get the current module
+        module = sys.modules[__name__]
         globals_dict = {
             '__builtins__': __builtins__,
         }
@@ -369,6 +369,14 @@ def extract_and_run_sympy_code(response_text):
         # Execute with try/except to catch specific error
         try:
             exec(sympy_code, globals_dict, local_vars)
+            
+            # Print all variables that were created
+            print("SymPy Calculation Results:")
+            print("-" * 25)
+            for var_name, value in local_vars.items():
+                if (not var_name.startswith('__') and 
+                    not hasattr(value, '__module__')):
+                    print(f"{var_name} = {value}")
         except NameError as e:
             logger.error(f"NameError during execution: {str(e)}")
             logger.debug(f"Available names: {list(globals_dict.keys())}")
@@ -379,18 +387,6 @@ def extract_and_run_sympy_code(response_text):
             logger.error(f"Other error during execution: {str(e)}")
             return f"Error executing SymPy code: {str(e)}"
         
-        logger.debug("Code executed successfully")
-        logger.debug(f"Local variables created: {list(local_vars.keys())}")
-        
-        # Print results
-        with redirect_stdout(output_buffer):
-            print("SymPy Calculation Results:")
-            print("-" * 25)
-            for var_name, value in local_vars.items():
-                if (not var_name.startswith('__') and 
-                    not hasattr(value, '__module__')):
-                    print(f"{var_name}: {value}")
-        
         result = output_buffer.getvalue()
         logger.debug(f"Final output:\n{result}")
         return result
@@ -401,6 +397,7 @@ def extract_and_run_sympy_code(response_text):
         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:
     gr.Markdown("# Advanced Mathematics Question Generator")