Update app.py

app.py

@@ -526,7 +526,7 @@ class QuizApp:
             <p>You need 80% or higher to pass and receive a certificate.</p>
         </div>
         """
-
+------------------
 def create_quiz_interface():
     if not os.getenv("GROQ_API_KEY"):
         raise EnvironmentError("Please set your GROQ_API_KEY environment variable")
@@ -595,15 +595,28 @@ def create_quiz_interface():
                         
                         gr.Markdown("---")  # Separator
                         
-                        submit_btn = gr.Button(
-                            "Submit Assessment",
-                            variant="primary",
-                            size="lg"
-                        )
+                        with gr.Row():
+                            submit_btn = gr.Button(
+                                "Submit Assessment",
+                                variant="primary",
+                                size="lg"
+                            )
+                            reset_btn = gr.Button(
+                                "Reset Quiz",
+                                variant="secondary",
+                                size="lg"
+                            )
                 
                 # Results section
                 with gr.Group(visible=False) as results_group:
                     feedback_box = gr.Markdown("")
+                    with gr.Row():
+                        view_cert_btn = gr.Button(
+                            "View Certificate",
+                            variant="primary",
+                            size="lg",
+                            visible=False
+                        )
             
             # Certification Tab
             with gr.Tab("🎓 Step 3: Get Certified"):
@@ -614,8 +627,7 @@ def create_quiz_interface():
                     value="Professional Assessment Certification"
                 )
                 certificate_display = gr.Image(label="Your Certificate")
-
-        # Helper Functions
+# Helper Functions
         def on_generate_questions(text, num_questions):
             success, questions = quiz_app.generate_questions(text, num_questions)
             if not success:
@@ -628,7 +640,8 @@ def create_quiz_interface():
                     0,
                     [None] * 5,
                     gr.update(selected=1),
-                    gr.update(visible=False)
+                    gr.update(visible=False),
+                    gr.update(visible=False)  # view_cert_btn visibility
                 ]
             
             initial_answers = [None] * len(questions)
@@ -648,7 +661,8 @@ def create_quiz_interface():
                 0,
                 initial_answers,
                 gr.update(selected=1),
-                gr.update(visible=False)
+                gr.update(visible=False),
+                gr.update(visible=False)  # view_cert_btn visibility
             ]
 
         def navigate(direction, current_idx, questions, answers, current_answer):
@@ -682,10 +696,10 @@ def create_quiz_interface():
                 ),
                 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)
+            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)
@@ -708,7 +722,8 @@ def create_quiz_interface():
                     0,
                     "",
                     gr.update(visible=True),
-                    gr.update(selected=1)
+                    gr.update(selected=1),
+                    gr.update(visible=False)  # view_cert_btn visibility
                 ]
             
             score, passed, feedback = quiz_app.calculate_score(final_answers[:len(questions)])
@@ -742,10 +757,19 @@ def create_quiz_interface():
                 score,
                 result_msg,
                 gr.update(visible=not passed),
-                gr.update(selected=2 if passed else 1)
+                gr.update(selected=1),
+                gr.update(visible=passed)  # Show view certificate button only if passed
             ]
 
-        # Event handlers
+        def reset_quiz(text, num_questions):
+            """Handle quiz reset"""
+            return on_generate_questions(text, num_questions)
+
+        def view_certificate():
+            """Navigate to certificate tab"""
+            return gr.update(selected=2)
+
+# Event handlers
         generate_btn.click(
             fn=on_generate_questions,
             inputs=[text_input, num_questions],
@@ -758,8 +782,12 @@ def create_quiz_interface():
                 current_question_idx,
                 answer_state,
                 tabs,
-                results_group
+                results_group,
+                view_cert_btn
             ]
+        ).then(
+            lambda: gr.update(selected=1),  # Navigate to Take Assessment tab
+            outputs=tabs
         )
 
         prev_btn.click(
@@ -818,10 +846,34 @@ def create_quiz_interface():
                 score_display,
                 result_message,
                 question_box,
-                tabs
+                tabs,
+                view_cert_btn
             ]
         )
 
+        reset_btn.click(
+            fn=reset_quiz,
+            inputs=[text_input, num_questions],
+            outputs=[
+                question_display,
+                current_options,
+                question_counter,
+                question_box,
+                current_questions,
+                current_question_idx,
+                answer_state,
+                tabs,
+                results_group,
+                view_cert_btn
+            ]
+        )
+
+        view_cert_btn.click(
+            fn=view_certificate,
+            outputs=tabs
+        )
+
+        # Certificate generation
         score_display.change(
             fn=quiz_app.certificate_generator.generate,
             inputs=[score_display, name, course_name, company_logo, participant_photo],