Update app.py

app.py

@@ -289,13 +289,15 @@ STRICT REQUIREMENTS:
 
         # Execute SymPy code and append results
         sympy_output = extract_and_run_sympy_code(response_text)
+
         if sympy_output:
-            response_text += "\n\nSymPy Verification Results:\n```\n" + sympy_output + "```"
-        
-        # Then continue with the existing code:
-        # Create LaTeX content
-        questions_latex = create_latex_document(response_text, questions_only=True)
-        full_latex = create_latex_document(response_text, questions_only=False)
+            # Check if SymPy ran successfully
+            if "Error" not in sympy_output:
+                resolution = check_and_resolve_discrepancy(response_text, sympy_output)
+                response_text = f"{response_text}\n\nSymPy Verification Results:\n```\n{sympy_output}\n```\n\nVerification Analysis:\n{resolution}"
+            else:
+                # Just append SymPy results if there was an error
+                response_text += f"\n\nSymPy Verification Results:\n```\n{sympy_output}\n```"
         
         # Create LaTeX content
         questions_latex = create_latex_document(response_text, questions_only=True)
@@ -352,7 +354,10 @@ def extract_and_run_sympy_code(response_text):
             "exp": sympy.exp,
             "simplify": sympy.simplify,
             "expand": sympy.expand,
-            "print": print  # Allow print statements
+            "print": print,  # Allow print statements
+            "pi": sympy.pi,
+            "N": sympy.N,
+            "evalf": lambda x: x.evalf()
         }
         
         # Create locals dict to capture new variables
@@ -401,7 +406,50 @@ def extract_and_run_sympy_code(response_text):
     
     except Exception as e:
         return f"Error executing SymPy code: {str(e)}"
+
+def check_and_resolve_discrepancy(initial_response, sympy_output):
+    """
+    Compare the SymPy output with the initial response and resolve any discrepancies
+    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 solutions.
+        
+        If the two answers are consistent with each other
+        then please say that they are consistent and briefly explain why. 
         
+        If the two answers are inconsistent with each other then please:
+
+        1. Identify which solution is correct
+        2. Explain the error in the incorrect solution
+        3. Provide a revised complete solution that fixes any errors
+
+Original solution:
+{initial_response}
+
+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
+            messages=[{
+                "role": "user",
+                "content": resolution_prompt
+            }]
+        )
+        
+        return message.content[0].text
+    except Exception as e:
+        logger.error(f"Error in discrepancy resolution: {str(e)}")
+        return initial_response  # Return original response if resolution fails
+
 # Create Gradio interface
 with gr.Blocks() as interface:
     gr.Markdown("# Advanced Mathematics Question Generator")