Update app.py

app.py

@@ -14,6 +14,9 @@ st.subheader('🚀 Test Your Knowledge!')
 standard = st.text_input('📚 Enter Standard/Grade')
 topic = st.text_input('📝 Enter Topic')
 
+# 🔹 User Selects Quiz Type
+quiz_type = st.radio("🎭 Select Quiz Type", ["MCQ", "True/False", "Fill in the Blanks"])
+
 # 🔹 Difficulty Level
 difficulty = st.radio("📊 Select Difficulty Level", ["Low", "Medium", "Hard"])
 
@@ -27,30 +30,50 @@ model = ChatGroq(
     model_name="llama-3.3-70b-versatile"
 )
 
-def detect_quiz_type(topic):
-    """AI determines the best quiz type for the topic."""
-    prompt = f"""
-    Given the topic '{topic}', select the most suitable quiz type:
-    - 'MCQ' for subjects like Science, Math, History.
-    - 'True/False' for fact-based topics.
-    - 'Fill in the Blanks' for language-based topics.
-
-    Return only one of these: "MCQ", "True/False", "Fill in the Blanks".
-    """
-    response = model.predict(prompt).strip()
-    if response in ["MCQ", "True/False", "Fill in the Blanks"]:
-        return response
-    return "MCQ"  # Default to MCQ
-
 def generate_quiz(standard, topic, quiz_type, difficulty, num_questions):
-    """Generate quiz using AI with strict JSON formatting."""
+    """Generate topic-specific quiz using AI in JSON format."""
+    
+    format_example = {
+        "MCQ": '''
+        {
+            "questions": [
+                {
+                    "question": "What is the chemical symbol for water?",
+                    "options": ["A) O2", "B) CO2", "C) H2O", "D) N2"],
+                    "answer": "C) H2O"
+                }
+            ]
+        }
+        ''',
+        "True/False": '''
+        {
+            "questions": [
+                {
+                    "question": "The Sun is a planet.",
+                    "answer": "False"
+                }
+            ]
+        }
+        ''',
+        "Fill in the Blanks": '''
+        {
+            "questions": [
+                {
+                    "question": "The process by which plants make food is called ____.",
+                    "answer": "Photosynthesis"
+                }
+            ]
+        }
+        '''
+    }
+
     prompt = f"""
-    Generate {num_questions} {quiz_type} questions for a {standard} student on {topic}. 
+    Generate {num_questions} {quiz_type} questions for a {standard} grade student on the topic '{topic}' with {difficulty} difficulty.
+    
+    Ensure the quiz is strictly about '{topic}'.
 
-    Format:
-    {{"questions": [
-        {{"question": "Example?", "options": ["A) Option1", "B) Option2"], "answer": "A) Option1"}} 
-    ]}}
+    Return strictly in JSON format like this:
+    {format_example[quiz_type]}
     """
 
     try:
@@ -86,11 +109,10 @@ if 'user_answers' not in st.session_state:
 # 🔹 Generate Quiz Button
 if st.button("🚀 Generate Quiz"):
     if standard and topic:
-        auto_quiz_type = detect_quiz_type(topic)  # AI selects quiz type
-        st.session_state.quiz_data = generate_quiz(standard, topic, auto_quiz_type, difficulty, num_questions)
+        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 = {}  
-            st.success(f"✅ {auto_quiz_type} Quiz Generated on '{topic}' with {num_questions} Questions!")
+            st.success(f"✅ {quiz_type} Quiz Generated on '{topic}' with {num_questions} Questions!")
             st.balloons()
     else:
         st.error("⚠️ Please enter both the standard and topic.")
@@ -103,10 +125,12 @@ if st.session_state.quiz_data:
     for i, q in enumerate(questions):
         st.write(f"**Q{i+1}: {q['question']}**")
 
-        if "options" in q:  
+        if quiz_type == "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/False":
+            user_answer = st.radio(f"🧐 Select True or False for Question {i+1}", options=["True", "False"], key=f"question_{i+1}")
         else:  
-            user_answer = st.text_input(f"🧐 Your answer for Question {i+1}", key=f"question_{i+1}")
+            user_answer = st.text_input(f"🧐 Fill in the blank for Question {i+1}", key=f"question_{i+1}")
 
         st.session_state.user_answers[f"question_{i+1}"] = user_answer