Update app.py

app.py

@@ -530,7 +530,7 @@ STRICT REQUIREMENTS:
         logger.debug("Successfully created temporary files")
         
         return response_text, questions_path, full_path
-        
+            
     except Exception as e:
         logger.error(f"Error generating question: {str(e)}")
         return f"Error: {str(e)}", None, None
@@ -605,7 +605,7 @@ def check_and_resolve_discrepancy(initial_response, sympy_output):
         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 from the beginning that fixes any errors and does not refer to SymPy
+        3. Provide a revised complete solution that fixes any errors and does not refer to SymPy
 
 Original solution:
 {initial_response}
@@ -634,31 +634,26 @@ Please maintain the same LaTeX formatting as the original solution."""
             resolution_text += "\n\nNew SymPy Verification Results:\n```\n" + new_sympy_output + "\n```"
         
         # Determine if there was a discrepancy that required a revised solution
-        # Check for various phrasings that indicate a revised solution
-        revision_phrases = ["revised complete solution", "revised solution", "correct complete solution", "corrected solution"]
-        has_discrepancy = False
-        found_phrase = None
-        
-        for phrase in revision_phrases:
-            if phrase in resolution_text.lower():
-                has_discrepancy = True
-                found_phrase = phrase
-                break
-        
-        # Extract revised solution if it exists
+        # First check if there's an inconsistency mentioned
+        has_discrepancy = "inconsistent" in resolution_text.lower() or "inconsistency" in resolution_text.lower()
+        
+        # Look for the exact phrase we required in the prompt
         revised_solution = None
-        if has_discrepancy and found_phrase:
-            # Look for the revised solution after the explanation
-            solution_parts = resolution_text.split(found_phrase, maxsplit=1)
-            if len(solution_parts) > 1:
-                revised_solution = solution_parts[1].strip()
+        if has_discrepancy:
+            marker = "Here is the revised complete solution:"
+            parts = resolution_text.split(marker, maxsplit=1)
+            if len(parts) > 1:
+                revised_solution = parts[1].strip()
             
-            # Add additional check for solutions that might appear after "here's the"
-            if not revised_solution and "here's the" in solution_parts[0].lower():
-                solution_parts = resolution_text.split("here's the", maxsplit=1)
-                if len(solution_parts) > 1:
-                    revised_solution = solution_parts[1].strip()
-                    has_discrepancy = True
+            if not revised_solution:
+                # Fallback check for common revision phrases
+                revision_phrases = ["revised complete solution:", "revised solution:", "correct solution:", "corrected solution:"]
+                for phrase in revision_phrases:
+                    if phrase in resolution_text.lower():
+                        parts = resolution_text.split(phrase, maxsplit=1)
+                        if len(parts) > 1:
+                            revised_solution = parts[1].strip()
+                            break
         
         return resolution_text, has_discrepancy, revised_solution
             
@@ -684,7 +679,7 @@ Please:
 
 If you find any issues:
 1. Clearly explain what is incorrect or missing
-2. Provide a COMPLETE CORRECTED solution
+2. Provide a complete corrected solution
 3. Maintain the same LaTeX formatting as the original
 4. Include any missing assumptions or conditions
 5. If relevant, provide corrected SymPy code
@@ -717,6 +712,7 @@ Please ensure any corrected solution maintains proper LaTeX formatting with $ fo
     except Exception as e:
         logger.error(f"Error in final verification: {str(e)}")
         return f"Error in final verification: {str(e)}"
+    
 
 # Create Gradio interface
 with gr.Blocks() as interface: