Update app.py

app.py

@@ -570,8 +570,7 @@ def create_quiz_interface():
                     participant_photo = gr.Image(label="Your Photo (Optional)", type="filepath")
                 
                 generate_btn = gr.Button("Generate Assessment", variant="primary", size="lg")
-            
-            # Assessment Tab
+
             with gr.Tab("📝 Step 2: Take Assessment") as assessment_tab:
                 with gr.Column(visible=True) as question_box:
                     # Question section
@@ -616,6 +615,15 @@ def create_quiz_interface():
                             size="lg",
                             visible=False
                         )
+                        back_to_assessment = gr.Button(
+                            "Back to Assessment",
+                            variant="secondary",
+                            size="lg",
+                            visible=True
+                        )
+
+            
+
             
             # Certification Tab
             with gr.Tab("🎓 Step 3: Get Certified"):
@@ -627,43 +635,147 @@ 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:
+
+    def on_generate_questions(text, num_questions):
+                success, questions = quiz_app.generate_questions(text, num_questions)
+                tab_index = 1  # Index for "Take Assessment" tab
+                
+                if not success:
+                    return [
+                        "",
+                        gr.update(choices=[], visible=False),
+                        "",
+                        gr.update(visible=False),
+                        [],
+                        0,
+                        [None] * 5,
+                        gr.update(selected=tab_index),  # Force navigation
+                        gr.update(visible=False),
+                        gr.update(visible=False)  # view_cert_btn visibility
+                    ]
+                
+                initial_answers = [None] * len(questions)
+                question = questions[0]
+                
                 return [
-                    "",
-                    gr.update(choices=[], visible=False),
-                    "",
-                    gr.update(visible=False),
-                    [],
+                    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,
-                    [None] * 5,
-                    gr.update(selected=1),
+                    initial_answers,
+                    gr.update(selected=tab_index),  # Force navigation
                     gr.update(visible=False),
                     gr.update(visible=False)  # view_cert_btn visibility
                 ]
+    
+            def on_submit(questions, answers, current_idx, current_answer):
+                final_answers = list(answers)
+                if 0 <= current_idx < len(final_answers):
+                    final_answers[current_idx] = current_answer
+                
+                if not all(a is not None for a in final_answers[:len(questions)]):
+                    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)])
+                
+                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>
+    """
+                
+                result_msg = "🎉 Passed!" if passed else "Please try again"
+                
+                # Add a message about viewing certificate if passed
+                if passed:
+                    feedback_html += f"""
+    <div style="background-color: #e6ffe6; padding: 20px; margin-top: 20px; border-radius: 10px;">
+        <h3 style="color: #008000;">🎉 Congratulations!</h3>
+        <p>You passed with a score of {score:.1f}%</p>
+        <p>Click the "View Certificate" button below to see your certificate.</p>
+    </div>
+    """
+                else:
+                    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>
+    """
+                
+                return [
+                    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
+                ]
+    
+            def goto_take_assessment():
+                """Navigate to Take Assessment tab"""
+                return gr.update(selected=1)
+    
+            def goto_certificate():
+                """Navigate to Get Certified tab"""
+                return gr.update(selected=2)
+    
+            # 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,
+                    view_cert_btn
+                ]
+            )
+    
+            # Fix View Certificate navigation
+            view_cert_btn.click(
+                fn=goto_certificate,
+                outputs=tabs
+            )
+    
+            # Add "Back to Assessment" button in results view
+            back_to_assessment = gr.Button("Back to Assessment", visible=False)
+            back_to_assessment.click(
+                fn=goto_take_assessment,
+                outputs=tabs
+            )        
             
-            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),
-                gr.update(visible=False)  # view_cert_btn visibility
-            ]
-
+        
         def navigate(direction, current_idx, questions, answers, current_answer):
             """Handle navigation between questions"""
             if not questions: