Update src/app.py

src/app.py

@@ -488,9 +488,24 @@ with tab3:
                     
                     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 "UNABLE" 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
+                        response_text, usage = query_model_universal([{"role": "user", "content": prompt}], 300, model_choice, st.session_state.get("user_openai_key"))
+                        
+                        # PARSE OUTPUT
+                        if "UNABLE" not in response_text and "QUOTE:" in response_text:
+                            # Split into Question and Quote
+                            parts = response_text.split("QUOTE:")
+                            q_text = parts[0].replace("QUESTION:", "").strip()
+                            quote_text = parts[1].strip()
+                            
+                            # SAFETY CHECK: Is the quote actually in the text?
+                            # We use a loose check (in case of minor whitespace diffs)
+                            # We take the first 20 chars of the quote to verify location
+                            if quote_text[:20] in q_ctx["context_text"]:
+                                valid_question_found = True
+                                qs["active"] = True
+                                qs["question_data"] = q_ctx
+                                qs["generated_question_text"] = q_text
+                                qs["feedback"] = None
                 
                 if not valid_question_found:
                     if focus_topic: st.warning(f"No documents found containing '{focus_topic}'. Try a different keyword.")