Update app.py

app.py

@@ -8,36 +8,30 @@ GROQ_API_KEY = "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"
 
 # Streamlit App Title
 st.title('🎯 AI Quiz Generator')
-st.subheader('🚀 Test Your Knowledge and Have Fun!')
+st.subheader('🚀 Test Your Knowledge!')
 
-# Input Fields
+# User Inputs
 standard = st.text_input('📚 Enter Standard/Grade')
 topic = st.text_input('📝 Enter Topic')
-
-# Quiz Type Selection
 quiz_type = st.selectbox("🎭 Select Quiz Type", ["Multiple Choice Questions (MCQ)", "True or False", "Fill in the Blanks"])
-
-# Difficulty Level Selection
 difficulty = st.radio("📊 Select Difficulty Level", ["Low", "Medium", "Hard"])
-
-# Number of Questions Selection (1 to 10)
 num_questions = st.slider("🔢 Select Number of Questions", min_value=1, max_value=10, value=5)
 
 # Initialize AI Model
 model = ChatGroq(
     temperature=0.6,
     groq_api_key=GROQ_API_KEY,
-    model_name="llama-3.3-70b-versatile" 
+    model_name="llama-3.3-70b-versatile"
 )
 
 def generate_quiz(standard, topic, quiz_type, difficulty, num_questions):
-    """Generate quiz based on user selection and return formatted questions."""
-    
+    """Generate quiz questions dynamically based on user selection."""
+
     if quiz_type == "Multiple Choice Questions (MCQ)":
         prompt = f"""
         Generate {num_questions} multiple choice questions (MCQs) for a {standard} student on {topic}. 
         Each question should have exactly 4 options (A, B, C, D) and one correct answer.
-        Return in a structured text format:
+        Return the result in this format:
         
         1. Question text?
         A) Option 1
@@ -45,50 +39,107 @@ def generate_quiz(standard, topic, quiz_type, difficulty, num_questions):
         C) Option 3
         D) Option 4
         
-        Answer: Correct option (A/B/C/D)
+        Answer: Correct Option (A, B, C, or D)
         """
     elif quiz_type == "True or False":
         prompt = f"""
         Generate {num_questions} True or False questions for a {standard} student on {topic}.
-        Return in a structured text format:
-        
-        1. Question text? (True/False)
-        Answer: True/False
+        Return the result in this format:
+
+        1. Question text?
+        Answer: True or False
         """
     else:  # Fill in the Blanks
         prompt = f"""
         Generate {num_questions} fill-in-the-blank questions for a {standard} student on {topic}.
-        Return in a structured text format:
-        
-        1. Question text with ___ blank.
-        Answer: Correct Answer
+        Return the result in this format:
+
+        1. The capital of France is __.
+        Answer: Paris
         """
 
     try:
-        response = model.predict(prompt).strip()
-        st.write("🔹 **Raw AI Response:**", response)  # Debugging
-        return response
+        response = model.predict(prompt).strip()  # Ensure clean output
+        quiz_lines = response.split("\n")  # Split response into lines
+        questions = []
+        current_question = None
+
+        for line in quiz_lines:
+            if line.strip() and not line.startswith("Answer:"):
+                if line[0].isdigit():  # New question detected
+                    if current_question:
+                        questions.append(current_question)
+                    current_question = {"question": line, "options": [], "answer": ""}
+                elif line[0] in "ABCD)" and quiz_type == "Multiple Choice Questions (MCQ)":
+                    current_question["options"].append(line)
+            elif line.startswith("Answer:"):
+                current_question["answer"] = line.split("Answer: ")[1]
+
+        if current_question:
+            questions.append(current_question)
+
+        return questions
     except Exception as e:
-        st.error(f"⚠️ Error: {str(e)}")
+        st.error(f"⚠️ Failed to generate quiz. Error: {str(e)}")
         return None
 
+# Session state to store quiz data
+if 'quiz_data' not in st.session_state:
+    st.session_state.quiz_data = None
+if 'user_answers' not in st.session_state:
+    st.session_state.user_answers = {}
+
 # Generate Quiz Button
 if st.button("🚀 Generate Quiz"):
     if standard and topic:
-        quiz_text = generate_quiz(standard, topic, quiz_type, difficulty, num_questions)
-        if quiz_text:
-            st.session_state.quiz_text = quiz_text  # Store quiz text
-            st.success(f"{quiz_type} Quiz Generated!")
-            st.balloons()
+        st.session_state.quiz_data = generate_quiz(standard, topic, quiz_type, difficulty, num_questions)
+        if st.session_state.quiz_data:
+            st.session_state.user_answers = {}  # Reset answers
+            st.success(f"{quiz_type} Quiz Generated with {num_questions} Questions! 🎉")
     else:
         st.error("⚠️ Please enter both the standard and topic.")
 
 # Display Quiz if Available
-if 'quiz_text' in st.session_state:
+if st.session_state.quiz_data:
     st.write("### 📝 Quiz Questions:")
-    st.text(st.session_state.quiz_text)  # Display formatted text output
+    for i, q in enumerate(st.session_state.quiz_data):
+        st.write(f"**Q{i+1}: {q['question']}**")
+
+        if quiz_type == "Multiple Choice Questions (MCQ)":
+            user_answer = st.radio(f"🧐 Select your answer for Q{i+1}", options=q["options"], key=f"question_{i+1}")
+        elif quiz_type == "True or False":
+            user_answer = st.radio(f"🧐 Select your answer for Q{i+1}", options=["True", "False"], key=f"question_{i+1}")
+        else:  # Fill in the Blanks
+            user_answer = st.text_input(f"🧐 Your answer for Q{i+1}", key=f"question_{i+1}")
+
+        st.session_state.user_answers[f"question_{i+1}"] = user_answer
 
-# 🎉 Submission Handling
+# Submit Quiz Button
 if st.button("✅ Submit Quiz"):
-    st.success("✅ Your answers have been submitted! Review the answers and improve.")
+    if st.session_state.quiz_data:
+        correct_answers = {f"question_{i+1}": q["answer"] for i, q in enumerate(st.session_state.quiz_data)}
+        score = sum(1 for q_id, user_ans in st.session_state.user_answers.items() if user_ans == correct_answers.get(q_id, ""))
 
+        st.write("### 🎯 Quiz Results")
+        st.write(f"Your Score: **{score}/{num_questions}** 🎉")
+
+        # Show correct & incorrect answers
+        for q_id, user_ans in st.session_state.user_answers.items():
+            correct_ans = correct_answers.get(q_id, "")
+            if user_ans == correct_ans:
+                st.success(f"✅ {q_id.replace('_', ' ').title()} - Correct! ({correct_ans})")
+            else:
+                st.error(f"❌ {q_id.replace('_', ' ').title()} - Incorrect! Your Answer: {user_ans} | Correct Answer: {correct_ans}")
+
+        # 🎉 Animations based on score
+        time.sleep(1)
+        if score == num_questions:
+            st.snow()  # 🎉 Perfect score effect
+            st.success("🏆 Perfect Score! You're a genius! 🚀")
+        elif score / num_questions >= 0.7:
+            st.success("🎉 Great job! Keep practicing! 💪")
+        else:
+            st.warning("😃 Don't worry! Learn from mistakes and try again! 📚")
+
+    else:
+        st.error("⚠️ Quiz data not available. Please regenerate the quiz.")