Update app.py

app.py

@@ -15,9 +15,11 @@ class ExamPrepApp:
     def reset(self):
         random.shuffle(self.all_topics)
         self.exam_questions = self.all_topics[:NUM_EXAM_QUESTIONS]
-        self.answered = {}  # qidx: score (0-3)
-        self.answered_yes = set()  # Topics you said "Yes" (fully learned)
+        self.answered = {}   # qidx: score (0-3)
+        self.answered_yes = set()
+        self.answered_no = set()
         self.current_index = 0
+        self.last_answer = ""
 
     def _parse_flashcards(self) -> List[Dict[str, str]]:
         return [
@@ -376,27 +378,54 @@ Features:
                 "answer": """Range, bandwidth, interference, security."""
             }
         ]
+
     def get_current_card(self) -> Dict[str, str]:
         qidx = self.exam_questions[self.current_index]
         return self.flashcards[qidx]
 
     def show_question(self):
         card = self.get_current_card()
-        return card["question"], "", gr.update(visible=True), gr.update(visible=False), gr.update(visible=False), self.get_progress(), gr.update(visible=False), self.get_pass_probability()
+        answer_to_show = self.last_answer if self.last_answer else ""
+        return (
+            card["question"],
+            answer_to_show,
+            gr.update(visible=True),
+            gr.update(visible=False),
+            gr.update(visible=False),
+            self.get_stats(),
+            gr.update(visible=False),
+            self.get_pass_probability()
+        )
 
     def show_answer(self):
         card = self.get_current_card()
-        return card["question"], card["answer"], gr.update(visible=False), gr.update(visible=True), gr.update(visible=True), self.get_progress(), gr.update(visible=True), self.get_pass_probability()
+        return (
+            card["question"],
+            card["answer"],
+            gr.update(visible=False),
+            gr.update(visible=True),
+            gr.update(visible=True),
+            self.get_stats(),
+            gr.update(visible=True),
+            self.get_pass_probability()
+        )
 
     def handle_yes(self, score):
         qidx = self.exam_questions[self.current_index]
         self.answered_yes.add(qidx)
         self.answered[qidx] = int(score)
+        self.last_answer = self.flashcards[qidx]["answer"]
+        if qidx in self.answered_no:
+            self.answered_no.remove(qidx)
         return self.goto_next()
 
     def handle_no(self, score):
         qidx = self.exam_questions[self.current_index]
+        self.answered_no.add(qidx)
         self.answered[qidx] = int(score)
+        self.last_answer = self.flashcards[qidx]["answer"]
+        if qidx in self.answered_yes:
+            self.answered_yes.remove(qidx)
         return self.goto_next()
 
     def goto_next(self):
@@ -404,13 +433,14 @@ Features:
         for _ in range(n):
             self.current_index = (self.current_index + 1) % n
             qidx = self.exam_questions[self.current_index]
-            if qidx not in self.answered_yes and qidx not in self.answered:
+            if qidx not in self.answered:
                 return self.show_question()
         return self.show_summary()
 
     def show_summary(self):
         total = sum(self.answered.get(qidx, 0) for qidx in self.exam_questions)
-        # Show question, answer, and user score for each
+        correct = len(self.answered_yes)
+        wrong = len(self.answered_no)
         details = "\n\n".join([
             f"**Q{i+1}: {self.flashcards[q]['question']}**\n"
             f"**A:** {self.flashcards[q]['answer']}\n"
@@ -418,12 +448,23 @@ Features:
             for i, q in enumerate(self.exam_questions)
         ])
         status = "✅ You PASSED!" if total >= PASS_POINTS else "❌ You did NOT pass."
-        summary = f"**Your Score:** {total}/24\n**Result:** {status}\n\n**Full breakdown:**\n\n{details}"
+        summary = (
+            f"**Your Score:** {total}/24\n"
+            f"**Result:** {status}\n"
+            f"✅ Remembered: {correct}  |  ❌ Not remembered: {wrong}\n\n"
+            f"**Full breakdown:**\n\n{details}"
+        )
         return ("Exam Finished!", summary, gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), summary, gr.update(visible=False), self.get_pass_probability())
 
-    def get_progress(self):
-        done = len(self.answered)
-        return f"Question {done+1} of {NUM_EXAM_QUESTIONS} | Mark yourself (0-3 points per question)."
+    def get_stats(self):
+        total = sum(self.answered.get(qidx, 0) for qidx in self.exam_questions)
+        correct = len(self.answered_yes)
+        wrong = len(self.answered_no)
+        return (
+            f"✅ Remembered: {correct} | ❌ Not remembered: {wrong} | "
+            f"🎯 Your Score: {total}/24 | "
+            f"Question {len(self.answered)+1} of {NUM_EXAM_QUESTIONS}"
+        )
 
     def get_pass_probability(self):
         if len(self.answered) == NUM_EXAM_QUESTIONS:
@@ -463,7 +504,7 @@ with gr.Blocks(title="Network Exam Simulator", theme=gr.themes.Soft()) as demo:
     with gr.Row():
         with gr.Column(scale=3):
             question = gr.Textbox(label="❓ Question", interactive=False, lines=3)
-            answer = gr.Textbox(label="💡 Answer", interactive=False, visible=False, lines=8)
+            answer = gr.Textbox(label="💡 Answer (for previous question)", interactive=False, visible=True, lines=8)
             with gr.Row():
                 show_answer_btn = gr.Button("👀 Show Answer", variant="primary")
                 yes_btn = gr.Button("✅ I remember this", visible=False)
@@ -471,21 +512,32 @@ with gr.Blocks(title="Network Exam Simulator", theme=gr.themes.Soft()) as demo:
             with gr.Row():
                 score_input = gr.Radio(choices=[0,1,2,3], value=0, label="How many points would you get for this question?", interactive=True, visible=False)
         with gr.Column(scale=1):
-            stats = gr.Markdown(app.get_progress())
+            stats = gr.Markdown(app.get_stats())
             prob = gr.Markdown(app.get_pass_probability())
             reset_btn = gr.Button("♻️ Start New Exam", variant="stop")
-    # Event handlers
+
     show_answer_btn.click(
-        lambda: app.show_answer(),
+        fn=app.show_answer,
+        outputs=[question, answer, show_answer_btn, yes_btn, no_btn, stats, score_input, prob]
+    )
+    yes_btn.click(
+        fn=app.handle_yes,
+        inputs=[score_input],
+        outputs=[question, answer, show_answer_btn, yes_btn, no_btn, stats, score_input, prob]
+    )
+    no_btn.click(
+        fn=app.handle_no,
+        inputs=[score_input],
+        outputs=[question, answer, show_answer_btn, yes_btn, no_btn, stats, score_input, prob]
+    )
+    reset_btn.click(
+        fn=app.reset_exam,
+        outputs=[question, answer, show_answer_btn, yes_btn, no_btn, stats, score_input, prob]
+    )
+    demo.load(
+        fn=app.show_question,
         outputs=[question, answer, show_answer_btn, yes_btn, no_btn, stats, score_input, prob]
     )
-    def yes_submit(score): return app.handle_yes(score)
-    yes_btn.click(yes_submit, inputs=[score_input], outputs=[question, answer, show_answer_btn, yes_btn, no_btn, stats, score_input, prob])
-    def no_submit(score): return app.handle_no(score)
-    no_btn.click(no_submit, inputs=[score_input], outputs=[question, answer, show_answer_btn, yes_btn, no_btn, stats, score_input, prob])
-    def on_reset(): return app.reset_exam()
-    reset_btn.click(on_reset, outputs=[question, answer, show_answer_btn, yes_btn, no_btn, stats, score_input, prob])
-    demo.load(app.show_question, outputs=[question, answer, show_answer_btn, yes_btn, no_btn, stats, score_input, prob])
 
 if __name__ == "__main__":
     demo.launch()