Update app.py

app.py

@@ -2,51 +2,71 @@ import streamlit as st
 from transformers import pipeline
 import random
 
-st.set_page_config(page_title="Text-to-Quiz Generator", layout="centered")
-
-st.title("📘 Text-to-Quiz Generator")
-st.write("Enter a passage below and let AI generate a quiz for you!")
-
-# Text input
-text_input = st.text_area("Paste your text here:", height=200)
-
-# Load the pipeline
+# Load the pipeline with caching
 @st.cache_resource
 def load_pipeline():
     return pipeline("text2text-generation", model="valhalla/t5-small-e2e-qg")
 
+# Initialize the pipeline
 qg_pipeline = load_pipeline()
 
-# Generate question using the model
-def generate_questions(text):
-    input_text = f"generate questions: {text}"
-    results = qg_pipeline(input_text, max_length=128, do_sample=True, top_k=50, top_p=0.95, num_return_sequences=3)
-    return [res['generated_text'] for res in results]
-
-# Interface
-if st.button("Generate Quiz") and text_input:
-    st.subheader("📝 Generated Quiz")
-
-    try:
-        generated_questions = generate_questions(text_input)
-
-        for i, q in enumerate(generated_questions):
-            # You can split question and answer if model formats them well
-            question = q.split("?")[0] + "?"
-            correct_answer = "Answer based on context"  # Placeholder
-            choices = [correct_answer, "Random Option A", "Random Option B", "Random Option C"]
-            random.shuffle(choices)
+# UI Header
+st.markdown("# Text-to-Quiz Generator")
+st.markdown("Enter a passage below and let AI create an interactive quiz for you!")
+st.markdown("---")
 
-            st.markdown(f"**Q{i+1}. {question}**")
-            user_answer = st.radio("Choose an answer:", choices, key=i)
+# Input section
+passage = st.text_area("Paste your text here:", height=150)
 
-            if user_answer == correct_answer:
-                st.success("✅ Correct!")
-            else:
-                st.error(f"❌ Incorrect. Correct answer: **{correct_answer}**")
-            st.markdown("---")
+if st.button("Generate Quiz"):
+    with st.spinner("Generating your quiz..."):
+        # Generate question-answer pairs (adjust based on actual model output format)
+        generated = qg_pipeline(passage, max_length=512)
+        
+        # For this example, assume the model outputs a list of strings like "Q: ... A: ..."
+        # Parse the output into questions and answers (modify this based on your model's actual output)
+        quiz_data = []
+        for item in generated:
+            if "Q:" in item["generated_text"] and "A:" in item["generated_text"]:
+                q, a = item["generated_text"].split("A:")
+                question = q.replace("Q:", "").strip()
+                answer = a.strip()
+                quiz_data.append({"question": question, "answer": answer})
+        
+        if not quiz_data:
+            st.error("No valid questions generated. Please try a different passage.")
+        else:
+            # Store quiz data in session state
+            st.session_state["quiz_data"] = quiz_data
+            st.session_state["user_answers"] = {}
 
-    except Exception as e:
-        st.error(f"Something went wrong: {e}")
-else:
-    st.info("👆 Paste some text and click 'Generate Quiz'")
+    # Display quiz
+    if "quiz_data" in st.session_state:
+        st.markdown("### Your Quiz")
+        questions = [item["question"] for item in st.session_state["quiz_data"]]
+        answers = [item["answer"] for item in st.session_state["quiz_data"]]
+        
+        # Generate distractors for each question
+        for i, question in enumerate(questions):
+            correct_answer = answers[i]
+            distractors = [a for j, a in enumerate(answers) if j != i]
+            options = [correct_answer] + random.sample(distractors, min(3, len(distractors)))
+            random.shuffle(options)
+            
+            st.write(f"**Question {i+1}:** {question}")
+            st.session_state["user_answers"][question] = st.radio(
+                "Select your answer:", options, key=f"q{i}"
+            )
+        
+        # Submit button
+        if st.button("Submit Answers"):
+            score = 0
+            for i, question in enumerate(questions):
+                user_answer = st.session_state["user_answers"][question]
+                correct_answer = answers[i]
+                if user_answer == correct_answer:
+                    st.success(f"**Question {i+1}:** Correct! The answer is '{correct_answer}'.")
+                    score += 1
+                else:
+                    st.error(f"**Question {i+1}:** Incorrect. The correct answer is '{correct_answer}', not '{user_answer}'.")
+            st.markdown(f"### Your Score: {score}/{len(questions)}")