Update app.py

app.py

@@ -594,7 +594,7 @@ def create_quiz_interface():
                 generate_btn = gr.Button("Generate Assessment", variant="primary", size="lg")
             
             # Assessment Tab
-            with gr.Tab(id=2, label="📝 Step 2: Take Assessment") as assessment_tab:
+            with gr.Tab(id=2, label="📝 Step 2: Take Assessment"):
                 with gr.Column() as main_container:
                     # Questions Section
                     with gr.Column(visible=True) as question_box:
@@ -610,16 +610,20 @@ def create_quiz_interface():
                             question_counter = gr.Markdown("Question 1")
                             next_btn = gr.Button("Next →", variant="secondary", size="sm")
                         
+                        gr.Markdown("---")  # Separator
+                        
                         submit_btn = gr.Button(
                             "Submit Assessment",
                             variant="primary",
                             size="lg"
                         )
                     
-                    # Results Section
+                    # Results Section in Assessment Tab
                     with gr.Column(visible=False) as results_group:
                         feedback_box = gr.Markdown("")
-                        with gr.Row():
+                        gr.Markdown("---")  # Separator
+                        with gr.Row(equal_height=True):
+                            # Only one of these buttons will be visible based on pass/fail
                             reset_btn = gr.Button(
                                 "Reset Quiz",
                                 variant="secondary",
@@ -635,27 +639,25 @@ def create_quiz_interface():
             
             # Certification Tab
             with gr.Tab(id=3, label="🎓 Step 3: Get Certified"):
-                with gr.Column():
-                    score_display = gr.Number(label="Your Score", visible=False)
-                    course_name = gr.Textbox(
-                        label="Certification Title",
-                        value="Professional Assessment Certification"
-                    )
-                    certificate_display = gr.Image(label="Your Certificate")
-        
+                score_display = gr.Number(label="Your Score", visible=False)
+                course_name = gr.Textbox(
+                    label="Certification Title",
+                    value="Professional Assessment Certification"
+                )
+                certificate_display = gr.Image(label="Your Certificate")
+
         # Helper Functions
         def on_generate_questions(text, num_questions):
-            """Generate quiz questions and setup initial state"""
             if not text.strip():
                 return [
-                    gr.Markdown("⚠️ Please enter some text content to generate questions."),
+                    "",
                     gr.update(visible=False),
                     gr.update(choices=[], visible=False),
                     "",
                     [],
                     0,
                     [None] * 5,
-                    gr.Tabs(selected=1),
+                    gr.update(selected=1),
                     gr.update(visible=False),
                     gr.update(visible=False)
                 ]
@@ -663,25 +665,24 @@ def create_quiz_interface():
             success, questions = quiz_app.generate_questions(text, num_questions)
             if not success or not questions:
                 return [
-                    gr.Markdown("❌ Failed to generate questions. Please try again."),
+                    "",
                     gr.update(visible=False),
                     gr.update(choices=[], visible=False),
                     "",
                     [],
                     0,
                     [None] * 5,
-                    gr.Tabs(selected=1),
+                    gr.update(selected=1),
                     gr.update(visible=False),
                     gr.update(visible=False)
                 ]
             
-            # Setup initial question
             question = questions[0]
-            question_html = f"""### Question 1
-            {question.question}"""
+            question_md = f"""### Question 1
+{question.question}"""
             
             return [
-                gr.Markdown(question_html),
+                question_md,
                 gr.update(visible=True),
                 gr.update(
                     choices=question.options,
@@ -693,47 +694,38 @@ def create_quiz_interface():
                 questions,
                 0,
                 [None] * len(questions),
-                gr.Tabs(selected=2),
+                gr.update(selected=2),
                 gr.update(visible=False),
                 gr.update(visible=False)
             ]
 
         def navigate(direction, current_idx, questions, answers, current_answer):
-            """Handle navigation between questions"""
             if not questions:
                 return [0, answers, "", gr.update(choices=[], visible=False), "", gr.update(visible=False)]
             
-            # Update current answer in state
             new_answers = list(answers)
             if current_answer is not None and 0 <= current_idx < len(new_answers):
                 new_answers[current_idx] = current_answer
             
-            # Calculate new index
             new_idx = max(0, min(len(questions) - 1, current_idx + direction))
             question = questions[new_idx]
             
-            # Format question display
-            question_html = f"""### Question {new_idx + 1}
-            {question.question}"""
+            question_md = f"""### Question {new_idx + 1}
+{question.question}"""
             
             return [
                 new_idx,
                 new_answers,
-                gr.Markdown(question_html),
+                question_md,
                 gr.update(
                     choices=question.options,
                     value=new_answers[new_idx] if new_idx < len(new_answers) else None,
                     visible=True,
-                    label=f"Select your answer:"
+                    label="Select your answer:"
                 ),
                 f"Question {new_idx + 1} of {len(questions)}",
                 gr.update(visible=True)
             ]
-        def handle_prev(current_idx, questions, answers, current_answer):
-            return navigate(-1, current_idx, questions, answers, current_answer)
-
-        def handle_next(current_idx, questions, answers, current_answer):
-            return navigate(1, current_idx, questions, answers, current_answer)
 
         def update_answer_state(answer, idx, current_answers):
             new_answers = list(current_answers)
@@ -747,7 +739,7 @@ def create_quiz_interface():
 
         def view_certificate():
             """Navigate to certificate tab"""
-            return gr.Tabs(selected=2)
+            return gr.update(selected=3)
 
         def on_submit(questions, answers, current_idx, current_answer):
             """Handle quiz submission with proper Markdown rendering"""
@@ -812,6 +804,7 @@ def create_quiz_interface():
                 gr.update(visible=True),        # back_to_assessment
                 gr.update(visible=False)        # profile_tab
             ] 
+            
         # Event Handlers
         generate_btn.click(
             fn=on_generate_questions,
@@ -831,13 +824,13 @@ def create_quiz_interface():
         )
         
         prev_btn.click(
-            fn=lambda *args: navigate(-1, *args),
+            fn=handle_prev,
             inputs=[current_question_idx, current_questions, answer_state, current_options],
             outputs=[current_question_idx, answer_state, question_display, current_options, question_counter, question_box]
         )
         
         next_btn.click(
-            fn=lambda *args: navigate(1, *args),
+            fn=handle_next,
             inputs=[current_question_idx, current_questions, answer_state, current_options],
             outputs=[current_question_idx, answer_state, question_display, current_options, question_counter, question_box]
         )
@@ -859,7 +852,7 @@ def create_quiz_interface():
         )
         
         reset_btn.click(
-            fn=on_generate_questions,
+            fn=reset_quiz,
             inputs=[text_input, num_questions],
             outputs=[
                 question_display,
@@ -876,16 +869,22 @@ def create_quiz_interface():
         )
         
         view_cert_btn.click(
-            fn=lambda: gr.Tabs(selected=3),
+            fn=lambda: gr.update(selected=3),
             outputs=tabs
         )
         
+        current_options.change(
+            fn=update_answer_state,
+            inputs=[current_options, current_question_idx, answer_state],
+            outputs=answer_state
+        )
+        
         score_display.change(
             fn=lambda s, n, c, l, p: quiz_app.certificate_generator.generate(s, n, c, l, p) or gr.update(value=None),
             inputs=[score_display, name, course_name, company_logo, participant_photo],
             outputs=certificate_display
         )
-
+        
     return demo
 
 if __name__ == "__main__":