Update app.py

app.py

@@ -45,19 +45,22 @@ def generate_mcqs_from_text(user_text):
 # Function to parse the MCQs and correct answers
 def parse_mcqs_and_answers(mcqs_and_answers):
     try:
-        # Try splitting on "Correct Answers:"
-        parts = mcqs_and_answers.split("Correct Answers:")
-        if len(parts) != 2:
+        # Split based on the "Correct Answers" section
+        correct_answers_start = mcqs_and_answers.find("Correct Answers")
+        if correct_answers_start == -1:
             raise ValueError("Could not find 'Correct Answers:' in the response.")
         
-        # Separate MCQs and correct answers
-        mcqs = parts[0].strip()
+        # Extract MCQs and correct answers
+        mcqs = mcqs_and_answers[:correct_answers_start].strip()
+        correct_answers_section = mcqs_and_answers[correct_answers_start:].strip()
+
+        # Extract the correct answers
         correct_answers = {}
-        for line in parts[1].strip().splitlines():
+        for line in correct_answers_section.splitlines():
             if line.startswith("Question"):
                 q, ans = line.split(":")
                 correct_answers[q.strip()] = ans.strip()
-        
+
         return mcqs, correct_answers
     except Exception as e:
         st.error(f"Failed to parse response: {e}")
@@ -66,7 +69,6 @@ def parse_mcqs_and_answers(mcqs_and_answers):
 
 # Function to evaluate user answers
 def evaluate_answers(mcqs, user_answers):
-    # Here we assume the user_answers is a dictionary of answers for each question
     prompt = f"""
     Here are the user's answers to the following multiple-choice questions. 
     Please evaluate them as an experienced teacher, providing feedback on correctness,