Update app.py

app.py

@@ -7,6 +7,7 @@ import random
 import logging
 import tempfile
 from pathlib import Path
+from sympy import *
 
 # Set up logging
 logging.basicConfig(
@@ -302,6 +303,58 @@ STRICT REQUIREMENTS:
         logger.error(f"Error generating question: {str(e)}")
         return f"Error: {str(e)}", None, None
 
+from sympy import *  # Import all SymPy functionality
+
+def extract_and_run_sympy_code(response_text):
+    """
+    Extract SymPy code from the response and execute it.
+    Returns the execution output as a string.
+    """
+    try:
+        # Find the SymPy code block
+        sympy_start = response_text.find('```python')
+        if sympy_start == -1:
+            return "No SymPy code found in the response."
+        
+        # Extract the code (excluding the ```python and ``` markers)
+        code_start = response_text.find('\n', sympy_start) + 1
+        code_end = response_text.find('```', code_start)
+        if code_end == -1:
+            return "Malformed SymPy code block."
+        
+        sympy_code = response_text[code_start:code_end].strip()
+        
+        # Create a string buffer to capture print output
+        import io
+        import sys
+        from contextlib import redirect_stdout
+        
+        # Create string buffer for output
+        output_buffer = io.StringIO()
+        
+        # Create globals dict with all SymPy symbols
+        sympy_globals = {}
+        sympy_globals.update({name: getattr(sympy, name) for name in dir(sympy)})
+        
+        # Create locals dict to capture new variables
+        local_vars = {}
+        
+        # Execute the original code
+        exec(sympy_code, sympy_globals, local_vars)
+        
+        # Print all variables that were created (excluding built-ins and modules)
+        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}")
+        
+        return output_buffer.getvalue()
+        
+    except Exception as e:
+        return f"Error executing SymPy code: {str(e)}"
+
 # Create Gradio interface
 with gr.Blocks() as interface:
     gr.Markdown("# Advanced Mathematics Question Generator")