Update app.py

app.py

@@ -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"):
@@ -615,269 +628,168 @@ def create_quiz_interface():
                 )
                 certificate_display = gr.Image(label="Your Certificate")
 
-
+        # Helper Functions
         def on_generate_questions(text, num_questions):
-                    success, questions = quiz_app.generate_questions(text, num_questions)
-                    if not success:
-                        return [
-                            "",  # question_display
-                            gr.update(choices=[], visible=False),  # current_options
-                            "",  # question_counter
-                            gr.update(visible=False),  # question_box
-                            [],  # current_questions
-                            0,  # current_question_idx
-                            [None] * 5,  # answer_state
-                            gr.update(selected=1),  # tabs
-                            gr.update(visible=False)  # results_group
-                        ]
-                    
-                    # Initialize first question
-                    initial_answers = [None] * len(questions)
-                    question = questions[0]
-                    
-                    return [
-                        f"""## Question 1
-        {question.question}
-        
-        Please select one answer:""",  # question_display
-                        gr.update(
-                            choices=question.options,
-                            value=None,
-                            visible=True,
-                            label="Select your answer for Question 1:"
-                        ),  # current_options
-                        f"Question 1 of {len(questions)}",  # question_counter
-                        gr.update(visible=True),  # question_box
-                        questions,  # current_questions
-                        0,  # current_question_idx
-                        initial_answers,  # answer_state
-                        gr.update(selected=1),  # tabs
-                        gr.update(visible=False)  # results_group
-                    ]
-        
-                def navigate(direction, current_idx, questions, answers, current_answer):
-                    """
-                    Handle navigation between questions
-                    """
-                    if not questions:  # No questions available
-                        return [
-                            0,  # current_question_idx
-                            answers,  # answer_state
-                            "",  # question_display
-                            gr.update(choices=[], value=None, visible=False),  # current_options
-                            "",  # question_counter
-                            gr.update(visible=False)  # question_box
-                        ]
-                    
-                    # Save current answer
-                    new_answers = list(answers)
-                    if current_answer 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))
-                    
-                    # Get current question
-                    question = questions[new_idx]
-                    
-                    return [
-                        new_idx,  # current_question_idx
-                        new_answers,  # answer_state
-                        f"""## Question {new_idx + 1}
-        {question.question}
-        
-        Please select one answer:""",  # question_display
-                        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 for Question {new_idx + 1}:"
-                        ),  # current_options
-                        f"Question {new_idx + 1} of {len(questions)}",  # question_counter
-                        gr.update(visible=True)  # question_box
-                    ]
+            success, questions = quiz_app.generate_questions(text, num_questions)
+            if not success:
+                return [
+                    "",
+                    gr.update(choices=[], visible=False),
+                    "",
+                    gr.update(visible=False),
+                    [],
+                    0,
+                    [None] * 5,
+                    gr.update(selected=1),
+                    gr.update(visible=False)
+                ]
+            
+            initial_answers = [None] * len(questions)
+            question = questions[0]
+            
+            return [
+                f"## Question 1\n{question.question}\n\nPlease select one answer:",
+                gr.update(
+                    choices=question.options,
+                    value=None,
+                    visible=True,
+                    label="Select your answer for Question 1:"
+                ),
+                f"Question 1 of {len(questions)}",
+                gr.update(visible=True),
+                questions,
+                0,
+                initial_answers,
+                gr.update(selected=1),
+                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=[], value=None, visible=False),
+                    "",
+                    gr.update(visible=False)
+                ]
+            
+            new_answers = list(answers)
+            if current_answer and 0 <= current_idx < len(new_answers):
+                new_answers[current_idx] = current_answer
+            
+            new_idx = max(0, min(len(questions) - 1, current_idx + direction))
+            question = questions[new_idx]
+            
+            return [
+                new_idx,
+                new_answers,
+                f"## Question {new_idx + 1}\n{question.question}\n\nPlease select one answer:",
+                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 for Question {new_idx + 1}:"
+                ),
+                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)
+            if 0 <= idx < len(new_answers):
+                new_answers[idx] = answer
+            return new_answers
+
+       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)
 
         def on_submit(questions, answers, current_idx, current_answer):
-            # Update the current answer in the answers list
             final_answers = list(answers)
             if 0 <= current_idx < len(final_answers):
                 final_answers[current_idx] = current_answer
             
-            # Validate all answers are provided
             if not all(a is not None for a in final_answers[:len(questions)]):
-                return {
-                    feedback_box: "⚠️ Please answer all questions before submitting.",
-                    results_group: gr.update(visible=True),
-                    score_display: 0,
-                    result_message: "",
-                    question_box: gr.update(visible=True),
-                    tabs: gr.update(selected=1)
-                }
+                return [
+                    "⚠️ Please answer all questions before submitting.",
+                    gr.update(visible=True),
+                    0,
+                    "",
+                    gr.update(visible=True),
+                    gr.update(selected=1),
+                    gr.update(visible=False)  # view_cert_btn visibility
+                ]
             
             score, passed, feedback = quiz_app.calculate_score(final_answers[:len(questions)])
             
-            # Generate feedback HTML
             feedback_html = "# Assessment Results\n\n"
             for i, (q, f) in enumerate(zip(questions, feedback)):
                 color = "green" if f.is_correct else "red"
                 symbol = "✅" if f.is_correct else "❌"
-                feedback_html += f"""
-                ### Question {i+1}
-                {q.question}
-                
-                <div style="color: {color}; padding: 10px; margin: 5px 0; border-left: 3px solid {color};">
-                    {symbol} Your answer: {f.selected or "No answer"}
-                    {'' if f.is_correct else f'<br>Correct answer: {f.correct_answer}'}
-                </div>
-                """
+                feedback_html += f"""### Question {i+1}
+{q.question}
+
+<div style="color: {color}; padding: 10px; margin: 5px 0; border-left: 3px solid {color};">
+    {symbol} Your answer: {f.selected or "No answer"}
+    {'' if f.is_correct else f'<br>Correct answer: {f.correct_answer}'}
+</div>
+"""
             
             result_msg = "🎉 Passed!" if passed else "Please try again"
             if not passed:
                 feedback_html += f"""
-                <div style="background-color: #ffe6e6; padding: 20px; margin-top: 20px; border-radius: 10px;">
-                    <h3 style="color: #cc0000;">Please Try Again</h3>
-                    <p>Your score: {score:.1f}%</p>
-                    <p>You need 80% or higher to pass and receive a certificate.</p>
-                </div>
-                """
+<div style="background-color: #ffe6e6; padding: 20px; margin-top: 20px; border-radius: 10px;">
+    <h3 style="color: #cc0000;">Please Try Again</h3>
+    <p>Your score: {score:.1f}%</p>
+    <p>You need 80% or higher to pass and receive a certificate.</p>
+</div>
+"""
             
-            return {
-                feedback_box: feedback_html,
-                results_group: gr.update(visible=True),
-                score_display: score,
-                result_message: result_msg,
-                question_box: gr.update(visible=passed),
-                tabs: gr.update(selected=2 if passed else 1)
-            }
-
-        # Event handlers
-        generate_btn.click(
-            fn=on_generate_questions,
-            inputs=[text_input, num_questions],
-            outputs=[
-                question_display,
-                current_options,
-                question_counter,
-                question_box,
-                current_questions,
-                current_question_idx,
-                answer_state,
-                tabs,
-                results_group
-            ]
-        )
-
-        # Navigation event handlers
-        def handle_prev(current_idx, questions, answers, current_answer):
-            result = navigate(-1, current_idx, questions, answers, current_answer)
-            return [
-                result[current_question_idx],
-                result[answer_state],
-                result[question_display],
-                result[current_options],
-                result[question_counter],
-                result[question_box]
-            ]
-
-        def handle_next(current_idx, questions, answers, current_answer):
-            result = navigate(1, current_idx, questions, answers, current_answer)
             return [
-                result[current_question_idx],
-                result[answer_state],
-                result[question_display],
-                result[current_options],
-                result[question_counter],
-                result[question_box]
-            ]
-
-        prev_btn.click(
-            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=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
-            ]
-        )
-
-        # Update answer state when radio button changes
-        def update_answer_state(answer, idx, current_answers):
-            new_answers = list(current_answers)
-            if 0 <= idx < len(new_answers):
-                new_answers[idx] = answer
-            return new_answers
-
-        current_options.change(
-            fn=update_answer_state,
-            inputs=[current_options, current_question_idx, answer_state],
-            outputs=answer_state
-        )
-
-        # Submission handler
-        def on_submit_wrapper(questions, answers, current_idx, current_answer):
-            result = on_submit(questions, answers, current_idx, current_answer)
-            return [
-                result[feedback_box],
-                result[results_group],
-                result[score_display],
-                result[result_message],
-                result[question_box],
-                result[tabs]
+                feedback_html,
+                gr.update(visible=True),
+                score,
+                result_msg,
+                gr.update(visible=not passed),
+                gr.update(selected=1),
+                gr.update(visible=passed)  # Show view certificate button only if passed
             ]
+        # Event handlers
 
-        submit_btn.click(
-            fn=on_submit_wrapper,
-            inputs=[
-                current_questions,
-                answer_state,
-                current_question_idx,
-                current_options
-            ],
-            outputs=[
-                feedback_box,
-                results_group,
-                score_display,
-                result_message,
-                question_box,
-                tabs
-            ]
-        )
+        generate_btn.click(fn=on_generate_questions, 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]).then(fn=lambda: gr.Tabs(selected=2), outputs=tabs)
+        
+        prev_btn.click(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=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])
+        
+        current_options.change(fn=update_answer_state, inputs=[current_options, current_question_idx, answer_state], outputs=answer_state)
+        
+        submit_btn.click(fn=on_submit, inputs=[current_questions, answer_state, current_question_idx, current_options], outputs=[feedback_box, results_group, score_display, result_message, question_box, 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=lambda: gr.Tabs(selected=3), outputs=tabs)
+        
+        back_to_assessment.click(fn=lambda: gr.Tabs(selected=2), outputs=tabs)
+        
+        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)
 
-        # Certificate generation
-        score_display.change(
-            fn=quiz_app.certificate_generator.generate,
-            inputs=[score_display, name, course_name, company_logo, participant_photo],
-            outputs=certificate_display
-        )
 
     return demo
 
 if __name__ == "__main__":
     demo = create_quiz_interface()
-    demo.launch()    
+    demo.launch() 
+