Update app.py

app.py

@@ -2,6 +2,7 @@ import os
 from dotenv import load_dotenv
 import streamlit as st
 from groq import Groq
+import re
 
 # Load environment variables from .env file
 load_dotenv()
@@ -27,8 +28,8 @@ def generate_mcqs_from_text(user_text):
     B. [Option 2]  
     C. [Option 3]  
     D. [Option 4]
-    
-    Additionally, provide the correct answers in a separate section titled "Correct Answers" in the format:  
+
+    Correct Answers:
     Question 1: [Correct Option]  
     Question 2: [Correct Option], etc.
 
@@ -42,33 +43,37 @@ def generate_mcqs_from_text(user_text):
     print("API Response:\n", response)  # Debugging: Inspect the raw response
     return response
 
-# Function to parse the MCQs and correct answers
+# Function to parse MCQs and answers with a flexible approach
 def parse_mcqs_and_answers(mcqs_and_answers):
     try:
-        # 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.")
-        
-        # 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 correct_answers_section.splitlines():
-            if line.startswith("Question"):
-                q, ans = line.split(":")
-                correct_answers[q.strip()] = ans.strip()
-
-        return mcqs, correct_answers
+        # Regular expression to match MCQs and answers
+        mcq_pattern = r"(Question \d+:.+?)(?=Question \d+|$)"
+        answer_pattern = r"(Question \d+: .+?)\n(A|B|C|D)\. .+"
+
+        # Extract questions
+        questions = re.findall(mcq_pattern, mcqs_and_answers)
+        # Extract answers
+        answers = re.findall(answer_pattern, mcqs_and_answers)
+
+        if len(questions) == 0 or len(answers) == 0:
+            raise ValueError("Failed to extract questions or answers.")
+
+        # Parse questions and answers
+        mcqs = {}
+        for q, ans in zip(questions, answers):
+            question_text = q.strip()
+            answer_text = ans.strip()
+            mcqs[question_text] = answer_text
+
+        return mcqs
     except Exception as e:
         st.error(f"Failed to parse response: {e}")
         st.write("Raw response:", mcqs_and_answers)  # Display raw output for troubleshooting
-        return None, None
+        return None
 
 # 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, 
@@ -99,14 +104,14 @@ if st.button("Generate MCQs"):
     mcqs_and_answers = generate_mcqs_from_text(user_text)
     
     # Parse the MCQs and correct answers
-    mcqs, correct_answers = parse_mcqs_and_answers(mcqs_and_answers)
+    mcqs = parse_mcqs_and_answers(mcqs_and_answers)
     
-    if mcqs and correct_answers:
+    if mcqs:
         # Display the first MCQ question
-        questions = mcqs.splitlines()
+        questions = list(mcqs.keys())
         question = questions[0]
         
-        # Show question options
+        # Show question options (Here we assume options are A, B, C, D)
         st.write(f"Question: {question}")
         
         options = ["A", "B", "C", "D"]  # Example options, you may adjust according to your response