Update app.py

app.py

@@ -19,7 +19,7 @@ def generate_math_quiz(role, topic, difficulty, num_questions):
         difficulty (str): The difficulty level (e.g., Easy, Medium, Hard).
         num_questions (int): The number of questions to generate.
     Returns:
-        tuple: (list, dict) - The generated quiz questions and answers.
+        str: The generated quiz questions.
     """
     # Construct the prompt for the model
     prompt = (
@@ -32,62 +32,7 @@ def generate_math_quiz(role, topic, difficulty, num_questions):
     response = llm.invoke(prompt)
     
     # Extract the content from the response
-    quiz_data = response.content
-    
-    # Debugging: Print the raw quiz data
-    st.write("Raw quiz data:", quiz_data)
-    
-    # Initialize lists and dictionary
-    questions = []
-    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, correct_answers
-
-def display_quiz(questions):
-    """
-    Display quiz questions and collect answers from the user.
-    Args:
-        questions (list): List of quiz questions.
-    Returns:
-        dict: User's answers.
-    """
-    user_answers = {}
-    st.subheader("Quiz Questions")
-    for i, question in enumerate(questions):
-        user_answers[i] = st.text_input(f"Q{i + 1}: {question}", key=f"q{i}")
-    
-    return user_answers
-
-def calculate_score(user_answers, correct_answers):
-    """
-    Calculate the score based on the user's answers.
-    Args:
-        user_answers (dict): User's answers.
-        correct_answers (dict): Correct answers.
-    Returns:
-        int: User's score.
-    """
-    score = 0
-    for i, answer in user_answers.items():
-        if answer.strip().lower() == correct_answers.get(f'A{i + 1}', '').strip().lower():
-            score += 1
-    return score
+    return response.content
 
 # Create a form for user input
 with st.form("quiz_form"):
@@ -110,29 +55,7 @@ with st.form("quiz_form"):
     if submitted:
         if role and topic and difficulty:
             # Generate quiz questions based on the user input
-            questions, correct_answers = generate_math_quiz(role, topic, difficulty, num_questions)
-            
-            if 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)
-                    
-                    # 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.")
+            quiz = generate_math_quiz(role, topic, difficulty, num_questions)
+            st.info(quiz)  # Display the generated quiz questions
         else:
             st.error("Please fill in all fields to generate the quiz.")
-