Update src/app.py

src/app.py

@@ -569,6 +569,10 @@ with tab3:
     quiz = QuizEngine()
     qs = st.session_state.quiz_state
     
+    # NEW: Trigger State for "Next Question" chaining
+    if "quiz_trigger" not in st.session_state:
+        st.session_state.quiz_trigger = False
+
     # Display Streak
     with col_streak:
         st.metric("Current Streak", qs["streak"])
@@ -576,51 +580,60 @@ with tab3:
 
     st.divider()
 
-    # 2. START BUTTON
-    if not qs["active"]:
-        if st.button("🚀 Generate New Question", type="primary"):
-            
-            with st.spinner("Consulting the Board..."):
-                # MODE A: ACRONYMS
-                if "Acronym" in quiz_mode:
-                    q_data = quiz.get_random_acronym()
-                    if q_data:
-                        qs["active"] = True
-                        qs["question_data"] = q_data
-                        qs["feedback"] = None
-                        qs["generated_question_text"] = q_data["question"]
-                        st.rerun()
-                    else:
-                        st.error("No acronyms found! Run the extractor first.")
-
-                # MODE B: DOCUMENTS
+    # --- GENERATION FUNCTION (Reused for Button AND Auto-Trigger) ---
+    def generate_question():
+        with st.spinner("Consulting the Board..."):
+            # MODE A: ACRONYMS
+            if "Acronym" in quiz_mode:
+                q_data = quiz.get_random_acronym()
+                if q_data:
+                    qs["active"] = True
+                    qs["question_data"] = q_data
+                    qs["feedback"] = None
+                    qs["generated_question_text"] = q_data["question"]
                 else:
-                    valid_question_found = False
-                    attempts = 0
+                    st.error("No acronyms found! Run the extractor first.")
+
+            # MODE B: DOCUMENTS
+            else:
+                valid_question_found = False
+                attempts = 0
+                
+                while not valid_question_found and attempts < 3:
+                    attempts += 1
+                    q_ctx = quiz.get_document_context(st.session_state.username)
                     
-                    while not valid_question_found and attempts < 3:
-                        attempts += 1
-                        q_ctx = quiz.get_document_context(st.session_state.username)
+                    if q_ctx:
+                        prompt = quiz.construct_question_generation_prompt(q_ctx["context_text"])
+                        question_text, usage = query_model_universal(
+                            [{"role": "user", "content": prompt}], 
+                            300, model_choice, st.session_state.get("user_openai_key")
+                        )
                         
-                        if q_ctx:
-                            prompt = quiz.construct_question_generation_prompt(q_ctx["context_text"])
-                            question_text, usage = query_model_universal(
-                                [{"role": "user", "content": prompt}], 
-                                300, model_choice, st.session_state.get("user_openai_key")
-                            )
-                            
-                            if "SKIP" not in question_text and len(question_text) > 10:
-                                valid_question_found = True
-                                qs["active"] = True
-                                qs["question_data"] = q_ctx
-                                qs["generated_question_text"] = question_text
-                                qs["feedback"] = None
-                                st.rerun()
-                    
-                    if not valid_question_found:
-                        st.warning("Tried to find a good engineering question but the available text was too administrative. Try uploading denser technical notes!")
+                        if "SKIP" not in question_text and len(question_text) > 10:
+                            valid_question_found = True
+                            qs["active"] = True
+                            qs["question_data"] = q_ctx
+                            qs["generated_question_text"] = question_text
+                            qs["feedback"] = None
+                
+                if not valid_question_found:
+                    st.warning("Tried to find a good engineering question but the available text was too administrative. Try uploading denser technical notes!")
 
-    # 3. QUIZ INTERFACE
+    # 2. AUTO-TRIGGER CHECK
+    # If the user clicked "Next Question", this runs immediately on rerun
+    if st.session_state.quiz_trigger:
+        st.session_state.quiz_trigger = False # Reset flag
+        generate_question()
+        st.rerun() # Rerun to display the new question
+
+    # 3. MANUAL START BUTTON
+    if not qs["active"]:
+        if st.button("🚀 Generate New Question", type="primary"):
+            generate_question()
+            st.rerun()
+
+    # 4. QUIZ INTERFACE
     if qs["active"]:
         st.markdown(f"### {qs['generated_question_text']}")
         
@@ -658,6 +671,35 @@ with tab3:
                     
                 st.rerun()
 
+    # 5. FEEDBACK AREA
+    if qs["feedback"]:
+        st.divider()
+        if "PASS" in qs["feedback"]:
+            st.success("✅ CORRECT")
+        else:
+            if "FAIL" in qs["feedback"]:
+                st.error("❌ INCORRECT")
+            else:
+                st.warning("⚠️ PARTIAL / COMMENTARY")
+            
+        st.markdown(qs["feedback"])
+        
+        # Display Correct Answer
+        data = qs["question_data"]
+        if data["type"] == "acronym":
+             st.info(f"**Official Definition:** {data['correct_definition']}")
+        elif data["type"] == "document":
+            with st.expander("Show Source Text (Answer Key)"):
+                st.info(data["context_text"])
+        
+        # FIXED: "Next Question" now sets the trigger flag instead of just resetting state
+        if st.button("Next Question ➡️"):
+            st.session_state.quiz_trigger = True
+            qs["active"] = False
+            qs["question_data"] = None
+            qs["feedback"] = None
+            st.rerun()
+
     # 4. FEEDBACK AREA (MERGED & FIXED)
     if qs["feedback"]:
         st.divider()