Update app.py

app.py

@@ -1,82 +1,97 @@
-import os
-import random
 import gradio as gr
-from demo_questions import QUESTIONS  # demo questions (60 across 6 subjects)
+from demo_questions import QUESTIONS
 
-# -----------------------------
-# Helper functions
-# -----------------------------
+# Store student answers
+student_answers = {}
 
-def load_exam(subject):
-    """Load 10 random questions from the selected subject."""
-    all_questions = QUESTIONS.get(subject, [])
-    if not all_questions:
-        return [f"**Q{i+1}:** (No Question)" for i in range(10)]
 
-    questions = random.sample(all_questions, min(10, len(all_questions)))
-    outputs = [f"**Q{i+1}:** {q}" for i, q in enumerate(questions)]
+def load_exam(subject, paper):
+    """Load selected subject & paper questions."""
+    if subject not in QUESTIONS or paper not in QUESTIONS[subject]:
+        return [gr.Markdown("⚠️ No questions available for this selection.")]
+
+    qlist = QUESTIONS[subject][paper]
+    student_answers.clear()  # reset before new exam
+
+    outputs = []
+    for i, q in enumerate(qlist):
+        # Store internally but only show the text to student
+        student_answers[i] = {"id": q.get("id"), "question": q.get("text"),
+                              "choices": q.get("choices"), "answer": q.get("answer"),
+                              "student_answer": None}
+
+        # Display question text
+        outputs.append(gr.Markdown(f"**Q{i+1}:** {q['text']}"))
+
+        if q.get("choices"):
+            # Multiple-choice → student selects option
+            outputs.append(gr.Radio(choices=q["choices"], label=f"Your Answer for Q{i+1}",
+                                    interactive=True, key=f"q{i+1}"))
+        else:
+            # Subjective → free text box
+            outputs.append(gr.Textbox(label=f"Your Answer for Q{i+1}", interactive=True, key=f"q{i+1}"))
+
     return outputs
 
 
-def submit_exam_answers(*answers):
-    """Process submitted answers. For now, just echo them back."""
-    submitted = [f"Q{i+1}: {ans}" for i, ans in enumerate(answers, 1)]
-    return "\n".join(submitted)
+def submit_exam(*responses):
+    """Process submission, calculate score, and show results."""
+    score = 0
+    total = len(student_answers)
+
+    # Store student answers
+    for i, response in enumerate(responses):
+        student_answers[i]["student_answer"] = response
+
+    # Evaluate
+    review = []
+    for i, q in student_answers.items():
+        correct = q["answer"]
+        student = q["student_answer"]
+
+        if correct and student == correct:
+            score += 1
+            result = f"✅ Q{i+1}: Correct! (Your Answer: {student})"
+        elif correct:
+            result = f"❌ Q{i+1}: Wrong (Your Answer: {student}, Correct: {correct})"
+        else:
+            result = f"✏️ Q{i+1}: Essay submitted. (Your Answer: {student})"
 
+        review.append(result)
+
+    summary = f"## Exam Finished!\n\n**Score: {score} / {total}**\n\n" + "\n".join(review)
+    return summary
 
-# -----------------------------
-# Gradio UI
-# -----------------------------
 
 with gr.Blocks() as demo:
-    gr.Markdown("# 📘 SPM Exam Simulator")
+    gr.Markdown("# 🎓 SPM Exam Simulator")
 
-    subject = gr.Dropdown(
-        choices=list(QUESTIONS.keys()),
-        label="Select Subject",
-        value="BM"
-    )
+    with gr.Row():
+        subject = gr.Dropdown(choices=list(QUESTIONS.keys()), label="Select Subject")
+        paper = gr.Dropdown(choices=["Paper1", "Paper2"], label="Select Paper")
 
     load_btn = gr.Button("Load Exam")
 
-    question_boxes = []
-    answer_boxes = []
-
-    with gr.Column():
-        for i in range(10):
-            q_box = gr.Markdown(f"**Q{i+1}:** (empty)")
-            a_box = gr.Textbox(
-                label=f"Answer Q{i+1}",
-                lines=2,
-                placeholder="Type your answer here (e.g., A/B/C/D or essay)..."
-            )
-            question_boxes.append(q_box)
-            answer_boxes.append(a_box)
+    exam_area = gr.Column()
 
     submit_btn = gr.Button("Submit Answers")
-    result = gr.Textbox(label="Submission Result", interactive=False, lines=15)
-
-    # -----------------------------
-    # Button bindings
-    # -----------------------------
-    load_btn.click(
-        fn=load_exam,
-        inputs=subject,
-        outputs=question_boxes
-    )
+    result_box = gr.Markdown("")
 
+    def render_exam(subject, paper):
+        return load_exam(subject, paper)
+
+    load_btn.click(render_exam, inputs=[subject, paper], outputs=exam_area)
+
+    # Capture student responses dynamically
     submit_btn.click(
-        fn=submit_exam_answers,
-        inputs=answer_boxes,
-        outputs=result
+        submit_exam,
+        inputs=exam_area,  # all answer widgets will be collected
+        outputs=result_box
     )
 
-# -----------------------------
-# Launch
-# -----------------------------
-
 if __name__ == "__main__":
-    demo.launch(server_name="0.0.0.0", server_port=7860, share=False, ssr_mode=True)
+    demo.launch(server_name="0.0.0.0", server_port=7860, ssr_mode=True)
+