Update app.py

app.py

@@ -34,16 +34,30 @@ def generate_math_quiz(role, topic, difficulty, num_questions):
     # Extract the content from the response
     quiz_data = response.content
     
-    # For demonstration, let's assume quiz_data is in the format "Q1: question1, A1: answer1; Q2: question2, A2: answer2"
+    # Debugging: Print the raw quiz data
+    st.write("Raw quiz data:", quiz_data)
+    
+    # Initialize lists and dictionary
     questions = []
-    answers = {}
-    for item in quiz_data.split(';'):
-        if 'Q' in item and 'A' in item:
-            q, a = item.split(', ')
-            questions.append(q.split(': ')[1])
-            answers[q.split(': ')[0]] = a.split(': ')[1]
+    correct_answers = {}
+    
+    try:
+        # Example format parsing
+        items = quiz_data.split('\n')
+        for item in items:
+            if item.startswith("Q"):
+                # This is a question
+                q_number, q_text = item.split(': ', 1)
+                questions.append(q_text)
+            elif item.startswith("A"):
+                # This is an answer
+                a_number, a_text = item.split(': ', 1)
+                correct_answers[a_number] = a_text
+        
+    except Exception as e:
+        st.error(f"Error parsing quiz data: {e}")
     
-    return questions, answers
+    return questions, correct_answers
 
 def display_quiz(questions):
     """
@@ -71,7 +85,7 @@ def calculate_score(user_answers, correct_answers):
     """
     score = 0
     for i, answer in user_answers.items():
-        if answer.strip().lower() == correct_answers.get(f'Q{i + 1}', '').strip().lower():
+        if answer.strip().lower() == correct_answers.get(f'A{i + 1}', '').strip().lower():
             score += 1
     return score
 
@@ -98,25 +112,27 @@ with st.form("quiz_form"):
             # Generate quiz questions based on the user input
             questions, correct_answers = generate_math_quiz(role, topic, difficulty, num_questions)
             
-            # Display the quiz
-            user_answers = display_quiz(questions)
-            
-            # Button to submit answers
-            submit_answers = st.button("Submit Answers")
-            
-            if submit_answers:
-                # Calculate score
-                score = calculate_score(user_answers, correct_answers)
+            if questions:
+                # Display the quiz
+                user_answers = display_quiz(questions)
                 
-                # Display results
-                st.success(f"Your score: {score} / {len(questions)}")
-                st.balloons()
+                # Button to submit answers
+                submit_answers = st.button("Submit Answers")
                 
-                # Optional: Show correct answers
-                st.info("Correct Answers:")
-                for i, (question, answer) in enumerate(correct_answers.items()):
-                    st.write(f"{question}: {answer}")
+                if submit_answers:
+                    # Calculate score
+                    score = calculate_score(user_answers, correct_answers)
+                    
+                    # Display results
+                    st.success(f"Your score: {score} / {len(questions)}")
+                    st.balloons()
+                    
+                    # Optional: Show correct answers
+                    st.info("Correct Answers:")
+                    for i, (question, answer) in enumerate(correct_answers.items()):
+                        st.write(f"{question}: {answer}")
+            else:
+                st.error("No questions generated. Please try again.")
         else:
             st.error("Please fill in all fields to generate the quiz.")
 
-