Update app.py

app.py

@@ -2,114 +2,93 @@ import streamlit as st
 from langchain_groq import ChatGroq
 import json
 
-# App Configuration
-st.set_page_config(page_title="AI Quiz Generator", layout="centered")
+# Securely Load API Key (Replace with your actual key if not using secrets)
+GROQ_API_KEY = st.secrets.get("groq_api_key", "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") 
 
-# Title and UI
-st.title("📚 AI Quiz Generator")
-st.subheader("🚀 Test Your Knowledge!")
+# Streamlit App Title
+st.title('AI Quiz Generator 🎯')
+st.subheader('Test Your Knowledge!')
 
-# User Input
-A = st.text_input("📖 Enter Standard")
-B = st.text_input("🧠 Enter a Topic")
+# Input Fields
+standard = st.text_input('Enter Standard/Grade')
+topic = st.text_input('Enter Topic')
 
-# Initialize ChatGroq Model (Replace with actual API key)
+# 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="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"
+    groq_api_key=GROQ_API_KEY
 )
 
-def generate_quiz(standard, topic, difficulty):
-    """
-    Generates a quiz with 5 multiple-choice questions.
-    Returns structured JSON output.
-    """
-    prompt = (
-        f"Generate a quiz with exactly 5 multiple choice questions for a {standard} student on the topic of {topic}. "
-        "Each question should have four options and one correct answer. "
-        "Return the output in JSON format with the following structure: "
-        "{ \"questions\": [ {\"question\": \"...\", \"options\": [\"...\", \"...\", \"...\", \"...\"], \"answer\": \"...\"} ] }"
-    )
+def generate_quiz(standard, topic, quiz_type, difficulty, num_questions):
+    """Generate quiz based on user selection and return JSON data."""
+    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 4 options and one correct answer. Return JSON format: {{'questions': [{{'question': '...', 'options': ['...', '...', '...', '...'], 'answer': '...'}}]}}"
+    elif quiz_type == "True or False":
+        prompt = f"Generate {num_questions} true or false questions for a {standard} student on {topic}. Return JSON format: {{'questions': [{{'question': '...', '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}. Indicate the correct answer in JSON format: {{'questions': [{{'question': '...', 'answer': '...'}}]}}"
+
     response = model.predict(prompt)
     try:
-        return json.loads(response)
+        return json.loads(response)  # Parse JSON response safely
     except json.JSONDecodeError:
-        st.error("⚠️ Failed to generate quiz. Please try again.")
+        st.error("Failed to generate quiz. Try again.")
         return None
 
 def calculate_score(user_answers, correct_answers):
-    """Calculate the score based on user responses."""
+    """Calculate score based on user answers."""
     return sum(1 for q, ans in user_answers.items() if correct_answers.get(q) == ans)
 
-# Session state for quiz management
-if "quiz_data" not in st.session_state:
+# 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:
+if 'user_answers' not in st.session_state:
     st.session_state.user_answers = {}
-if "quiz_submitted" not in st.session_state:
-    st.session_state.quiz_submitted = False
-
-# Difficulty Selection
-difficulty = st.radio("🎯 Select Difficulty Level", ["Low", "Medium", "Hard"])
 
 # Generate Quiz Button
-if st.button("✨ Generate Quiz"):
-    if A and B:
-        st.session_state.quiz_data = generate_quiz(A, B, difficulty)
+if st.button("Generate Quiz"):
+    if standard and topic:
+        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 user answers
-            st.session_state.quiz_submitted = False
-            st.success("✅ Quiz Generated! Scroll down to answer the questions.")
+            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.")
+        st.error("Please enter both the standard and topic.")
 
 # Display Quiz if Available
 if st.session_state.quiz_data:
-    st.write("## 📝 Quiz Questions")
+    st.write("### Quiz Questions:")
     questions = st.session_state.quiz_data.get("questions", [])
 
-    for i, q in enumerate(questions[:5]):  # Limit to 5 questions
-        with st.expander(f"**Q{i+1}: {q['question']}**"):
-            user_answer = st.radio(
-                "Select your answer:",
-                options=q["options"],
-                key=f"question_{i+1}"
-            )
-            st.session_state.user_answers[f"question_{i+1}"] = user_answer
+    for i, q in enumerate(questions):  # Display all selected questions
+        st.write(f"**Q{i+1}: {q['question']}**")
+
+        if quiz_type == "Multiple Choice Questions (MCQ)":
+            user_answer = st.radio(f"Select your answer for Question {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 Question {i+1}", options=["True", "False"], key=f"question_{i+1}")
+        else:  # Fill in the Blanks
+            user_answer = st.text_input(f"Your answer for Question {i+1}", key=f"question_{i+1}")
+
+        st.session_state.user_answers[f"question_{i+1}"] = user_answer
 
 # Submit Quiz Button
-if st.button("📤 Submit Quiz"):
+if st.button("Submit Quiz"):
     if st.session_state.quiz_data:
-        correct_answers = {f"question_{i+1}": q["answer"] for i, q in enumerate(st.session_state.quiz_data["questions"][:5])}
+        correct_answers = {f"question_{i+1}": q["answer"] for i, q in enumerate(st.session_state.quiz_data["questions"])}
         score = calculate_score(st.session_state.user_answers, correct_answers)
         
-        st.session_state.quiz_submitted = True
-        st.session_state.score = score
-
-# Display Results After Submission
-if st.session_state.quiz_submitted:
-    st.write("## 🎯 Quiz Results")
-    
-    # Display correct and user answers side by side
-    col1, col2 = st.columns(2)
-    with col1:
-        st.write("### ✅ Correct Answers")
-        for i, q in enumerate(st.session_state.quiz_data["questions"][:5]):
-            st.write(f"**Q{i+1}:** {q['answer']}")
-    
-    with col2:
-        st.write("### 📝 Your Answers")
-        for i, ans in st.session_state.user_answers.items():
-            st.write(f"**{i.replace('_', ' ').title()}:** {ans}")
-
-    # Final Score Display
-    st.markdown(f"## 🎉 Your Score: **{st.session_state.score}/5**")
-    
-    # Feedback Message
-    if st.session_state.score == 5:
-        st.success("🔥 Perfect! You're a genius!")
-    elif st.session_state.score >= 3:
-        st.info("👍 Great job! Keep practicing.")
+        st.write("### Quiz Results")
+        st.write(f"Your Score: {score}/{num_questions} 🎉")
+        st.success("Great job! Keep practicing!")
     else:
-        st.warning("📚 Keep learning! Try again.")
-
+        st.error("Quiz data not available. Please regenerate the quiz.")