Update app.py

app.py

@@ -540,35 +540,30 @@ def get_solution_for_verification(response_text, sympy_correct, final_verificati
             # If we can't even split paragraphs, just return the whole text as both question and solution
             return response_text.strip(), response_text.strip()
         
-    # Extract question and solution sections
+    # Extract question
     question = response_text[q_start:q_end].strip()
-    original_solution = response_text[q_end:]
-    
-    # If no sympy verification was done, treat as correct and use original
-    if sympy_correct is None:
-        sympy_correct = True
     
-    if sympy_correct:
-        # Use original solution - need to extract before SymPy code
-        sympy_start = original_solution.find('```python')
-        if sympy_start != -1:
-            solution = original_solution[:sympy_start].strip()
+    # If we have a final verification solution, use that
+    if final_verification:
+        marker = "Here is the complete verified solution:"
+        if marker in final_verification:
+            solution = final_verification.split(marker)[1].strip()
+            logger.debug("Using final verified solution for verification")
+            return question, solution
         else:
-            solution = original_solution.strip()
+            # If marker not found in final verification, extract solution part
+            logger.debug("Marker not found in final verification, using full verification text")
+            return question, final_verification.strip()
+            
+    # Otherwise, extract original solution (before SymPy code if present)
+    original_solution = response_text[q_end:]
+    sympy_start = original_solution.find('```python')
+    if sympy_start != -1:
+        solution = original_solution[:sympy_start].strip()
     else:
-        # Use final verified solution
-        if final_verification:
-            # Extract the solution after the marker
-            marker = "Here is the complete verified solution:"
-            if marker in final_verification:
-                solution = final_verification.split(marker)[1].strip()
-            else:
-                # Fallback to using the whole verification text
-                solution = final_verification.strip()
-        else:
-            # Fallback to original if no final verification
-            solution = original_solution.strip()
+        solution = original_solution.strip()
     
+    logger.debug("Using original solution for verification")
     return question, solution
     
 def verify_with_chatgpt(question, solution):