Update app.py

app.py

@@ -277,6 +277,12 @@ STRICT REQUIREMENTS:
    - Print intermediate steps and final answers
    - Print variable values after they are computed"""
 
+#Consider
+   #When writing SymPy code:
+   #- Use FiniteSet(1, 2, 3) instead of Set([1, 2, 3]) for finite sets
+   #- Import specific functions instead of using 'from sympy import *'
+   #- Print results of each calculation step
+        
         logger.debug("Sending request to Anthropic API")
         message = anthropic.messages.create(
             model=difficulty_params['model'],
@@ -362,7 +368,6 @@ def check_and_resolve_discrepancy(initial_response, sympy_output):
     by making another API call to Claude.
     """
     try:
-        # Create prompt showing both solutions and asking for resolution
         resolution_prompt = f"""Here is a mathematics question with two answers.
         The first, called Original solution, is a complete solution.
         The second, called SymPy Verification, will only provide the final answer.
@@ -382,25 +387,31 @@ Original solution:
 SymPy Verification Results:
 {sympy_output}
 
-
 Please maintain the same LaTeX formatting as the original solution."""
 
         # Make API call for resolution
         message = anthropic.messages.create(
             model="claude-3-5-sonnet-20241022",
             max_tokens=4096,
-            temperature=0.2,  # Lower temperature for verification
+            temperature=0.2,
             messages=[{
                 "role": "user",
                 "content": resolution_prompt
             }]
         )
         
-        return message.content[0].text
+        resolution_text = message.content[0].text
+        
+        # Check if resolution contains new SymPy code
+        if "```python" in resolution_text:
+            new_sympy_output = extract_and_run_sympy_code(resolution_text)
+            resolution_text += "\n\nNew SymPy Verification Results:\n```\n" + new_sympy_output + "\n```"
+            
+        return resolution_text
     except Exception as e:
         logger.error(f"Error in discrepancy resolution: {str(e)}")
-        return initial_response  # Return original response if resolution fails
-
+        return initial_response
+        
 # Create Gradio interface
 with gr.Blocks() as interface:
     gr.Markdown("# Advanced Mathematics Question Generator")