Update app.py

app.py

@@ -514,36 +514,58 @@ def extract_and_run_sympy_code_simple(response_text):
         from contextlib import redirect_stdout
         
         output_buffer = io.StringIO()
-        with redirect_stdout(output_buffer):
-            exec(modified_code, globals_dict)
-            
-        return output_buffer.getvalue().strip() or "No output produced"
+        try:
+            with redirect_stdout(output_buffer):
+                exec(modified_code, globals_dict)
+            return output_buffer.getvalue().strip() or "No output produced"
+        except NonInvertibleMatrixError:
+            return "ERROR_NONINVERTIBLE_MATRIX: The system of equations has either no solution or infinitely many solutions. Matrix is singular (determinant = 0)."
+        except Exception as e:
+            if "det == 0" in str(e) or "not invertible" in str(e):
+                return "ERROR_NONINVERTIBLE_MATRIX: The system of equations has either no solution or infinitely many solutions. Matrix is singular (determinant = 0)."
+            return f"Error executing SymPy code: {str(e)}"
     
     except Exception as e:
-        return f"Error executing SymPy code: {str(e)}"
+        return f"Error in code extraction: {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.
+    Compare the SymPy output with the initial response and resolve any discrepancies.
+    Now handles singular matrix cases specifically.
     """
     try:
-        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.
-        
-        If the SymPy Verification answer is consistent with the final answer Original solution,
-        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
+        if "ERROR_NONINVERTIBLE_MATRIX" in sympy_output:
+            resolution_prompt = f"""The original solution claims to solve a system of linear equations, but SymPy indicates the coefficient matrix is not invertible (singular). This means:
+1. The system has either no solution or infinitely many solutions
+2. The vectors are linearly dependent
+3. The solution given cannot be unique
+
+Please:
+1. Verify if the system has no solutions or infinitely many solutions
+2. If solutions exist, express them in parametric form
+3. Provide a complete revised solution that correctly addresses this case
+
 Original solution:
 {initial_response}
-SymPy Verification Results:
-{sympy_output}
-Please maintain the same LaTeX formatting as the original solution."""
+
+Please maintain the same LaTeX formatting as the original solution and include correct SymPy code that uses row reduction instead of matrix inversion."""
+        else:
+            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.
+            
+            If the SymPy Verification answer is consistent with the final answer Original solution,
+            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(
@@ -560,13 +582,13 @@ Please maintain the same LaTeX formatting as the original solution."""
         
         # Check if resolution contains new SymPy code
         if "```python" in resolution_text:
-            new_sympy_output = extract_and_run_sympy_code(resolution_text)
+            new_sympy_output = extract_and_run_sympy_code_simple(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 f"Error in resolution: {str(e)}"
         
 # Create Gradio interface
 with gr.Blocks() as interface: