Update app.py

app.py

@@ -288,7 +288,7 @@ STRICT REQUIREMENTS:
         logger.debug("Successfully received response from Anthropic API")
 
         # Execute SymPy code and append results
-        sympy_output = extract_and_run_sympy_code(response_text)
+        sympy_output = extract_and_run_sympy_code_simple(response_text)
 
         if sympy_output:
             # Check if SymPy ran successfully
@@ -313,8 +313,42 @@ STRICT REQUIREMENTS:
             
     except Exception as e:
         logger.error(f"Error generating question: {str(e)}")
-        return f"Error: {str(e)}", None, None
+        return f"Error: {str(e)}", None, Non
 
+def extract_and_run_sympy_code_simple(response_text):
+    """
+    Extract SymPy code from the response and execute it.
+    Returns the output exactly as SymPy would produce it.
+    """
+    try:
+        # Extract code
+        sympy_start = response_text.find('```python')
+        if sympy_start == -1:
+            return "No SymPy code found in the response."
+        
+        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()
+        
+        # Set up basic SymPy environment
+        import io
+        import sympy
+        from contextlib import redirect_stdout
+        
+        # Capture output
+        output_buffer = io.StringIO()
+        with redirect_stdout(output_buffer):
+            exec(sympy_code, {"sympy": sympy, "print": print})
+            
+        return output_buffer.getvalue().strip() or "No output produced"
+    
+    except Exception as e:
+        return f"Error executing SymPy code: {str(e)}"
+        
+        
 def extract_and_run_sympy_code(response_text):
    """
    Extract SymPy code from the response and execute it.