logic fixes

app.py

@@ -44,7 +44,7 @@ with gr.Blocks(title="TAI: AI Teacher Assistant") as demo:
     with gr.Row():
 
         # Left column: chat
-        with gr.Column(scale=2):
+        with gr.Column(scale=1):
             chatbot = gr.ChatInterface(
                 fn=lambda message, history: chat(llm, message, history),
                 examples=[
@@ -56,7 +56,7 @@ with gr.Blocks(title="TAI: AI Teacher Assistant") as demo:
             )
         
         # Right column: quiz
-        with gr.Column(scale=1):
+        with gr.Column(scale=2):
             gr.Markdown("## Test Yourself")
             
             start_btn = gr.Button("Start Quiz", variant="primary")
@@ -66,45 +66,63 @@ with gr.Blocks(title="TAI: AI Teacher Assistant") as demo:
             progress_md = gr.Markdown("")
 
             # Answer buttons
-            btn_A = gr.Button("A")
-            btn_B = gr.Button("B")
-            btn_C = gr.Button("C")
-            btn_D = gr.Button("D")
-            retry_btn = gr.Button("Retry")
+            btn_A = gr.Button("A", visible=False)
+            btn_B = gr.Button("B", visible=False)
+            btn_C = gr.Button("C", visible=False)
+            btn_D = gr.Button("D", visible=False)
+            retry_btn = gr.Button("Retry", visible=False)
 
             # States
             idx_state = gr.State(0)
             score_state = gr.State(0)
 
             # Start quiz
-            start_fn = partial(start_quiz, parsed_quiz)
+            def start_quiz_ui():
+                q_text, feedback, progress, idx, score, _ = start_quiz(parsed_quiz)
+                return (
+                    q_text, feedback, progress, idx, score,
+                    gr.update(visible=False),  # hide start button
+                    gr.update(visible=True),   # show A
+                    gr.update(visible=True),   # show B
+                    gr.update(visible=True),   # show C
+                    gr.update(visible=True),   # show D
+                    gr.update(visible=True)    # show retry
+                )
 
             start_btn.click(
-                fn=start_fn,
+                fn=start_quiz_ui,
                 inputs=[],
-                outputs=[question_md, idx_state, score_state, feedback_md, progress_md, start_btn]
+                outputs=[question_md, feedback_md, progress_md, idx_state, score_state,
+                         start_btn, btn_A, btn_B, btn_C, btn_D, retry_btn]
             )
 
             # Answer button clicks
-            answer_fn_A = partial(answer_question, parsed_quiz, "A")
-            answer_fn_B = partial(answer_question, parsed_quiz, "B")
-            answer_fn_C = partial(answer_question, parsed_quiz, "C")
-            answer_fn_D = partial(answer_question, parsed_quiz, "D")
-
-            btn_A.click(fn=answer_fn_A, inputs=[idx_state, score_state],
-                        outputs=[question_md, idx_state, score_state, feedback_md, progress_md])
-            btn_B.click(fn=answer_fn_B, inputs=[idx_state, score_state],
-                        outputs=[question_md, idx_state, score_state, feedback_md, progress_md])
-            btn_C.click(fn=answer_fn_C, inputs=[idx_state, score_state],
-                        outputs=[question_md, idx_state, score_state, feedback_md, progress_md])
-            btn_D.click(fn=answer_fn_D, inputs=[idx_state, score_state],
-                        outputs=[question_md, idx_state, score_state, feedback_md, progress_md])
+            for letter, btn in zip(["A", "B", "C", "D"], [btn_A, btn_B, btn_C, btn_D]):
+                btn_fn = partial(answer_question, parsed_quiz, letter)
+                btn.click(
+                    fn=btn_fn,
+                    inputs=[idx_state, score_state],
+                    outputs=[question_md, idx_state, score_state, feedback_md, progress_md]
+                )
 
             # Retry button
+            def retry_quiz_ui():
+                q_text, feedback, progress, idx, score, _ = start_quiz(parsed_quiz)
+                return (
+                    q_text, feedback, progress, idx, score,
+                    gr.update(visible=True),    # show start
+                    gr.update(visible=False),   # hide A
+                    gr.update(visible=False),   # hide B
+                    gr.update(visible=False),   # hide C
+                    gr.update(visible=False),   # hide D
+                    gr.update(visible=False)    # hide retry
+                )
+
             retry_btn.click(
-                fn=retry_quiz,
+                fn=retry_quiz_ui,
                 inputs=[],
-                outputs=[question_md, idx_state, score_state, feedback_md, progress_md, start_btn]
+                outputs=[question_md, feedback_md, progress_md, idx_state, score_state,
+                         start_btn, btn_A, btn_B, btn_C, btn_D, retry_btn]
             )
 
 

quiz.py

@@ -68,16 +68,11 @@ def parse_quiz(text):
 
     return parsed
 
-def start_quiz(quiz_raw, quiz_parsed):
+def start_quiz(quiz_parsed):
     """Initialize quiz state, return first question display."""
     idx = 0
     q = quiz_parsed[idx]
-    return {
-        "question_text": q["q"],
-        "feedback": "",
-        "progress": f"{idx+1}/{len(quiz_parsed)}",
-        "show_start": False
-    }, idx, 0  # idx, score
+    return q["q"], "", f"{idx+1}/{len(quiz_parsed)}", idx, 0, False
 
 def answer_question(parsed_quiz, selected, idx, score):
     current = parsed_quiz[idx]