Update app.py

app.py

@@ -537,6 +537,7 @@ def create_quiz_interface():
         # State management
         current_questions = gr.State([])
         current_question_idx = gr.State(0)
+        answer_state = gr.State([None] * 5)  # Store all answers
         
         # Header
         gr.Markdown("""
@@ -573,14 +574,14 @@ def create_quiz_interface():
             
             # Assessment Tab
             with gr.Tab("📝 Step 2: Take Assessment") as assessment_tab:
-                with gr.Box(visible=True):
+                with gr.Box(visible=True) as question_box:
                     # Question display
                     question_display = gr.Markdown("", label="Current Question")
                     
                     # Single radio group for current question
                     current_options = gr.Radio(
                         choices=[],
-                        label="Select your answer",
+                        label="Select your answer:",
                         visible=False
                     )
                     
@@ -590,6 +591,8 @@ def create_quiz_interface():
                         question_counter = gr.Markdown("Question 1 of 3")
                         next_btn = gr.Button("Next →", variant="secondary", size="sm")
                     
+                    gr.Markdown("---")  # Separator
+                    
                     submit_btn = gr.Button("Submit Assessment", variant="primary", size="lg")
                 
                 # Results section
@@ -606,27 +609,47 @@ def create_quiz_interface():
                 )
                 certificate_display = gr.Image(label="Your Certificate")
 
-        # Helper functions for navigation
-        def update_question_display(questions, idx):
+        def update_question_display(questions, idx, current_answers):
             if not questions or idx >= len(questions):
                 return {
                     question_display: "",
                     current_options: gr.update(choices=[], value=None, visible=False),
-                    question_counter: ""
+                    question_counter: "",
+                    question_box: gr.update(visible=False)
                 }
             
             question = questions[idx]
             return {
-                question_display: f"### Question {idx + 1}\n{question.question}",
-                current_options: gr.update(choices=question.options, value=None, visible=True),
-                question_counter: f"Question {idx + 1} of {len(questions)}"
+                question_display: f"""## Question {idx + 1}
+{question.question}
+
+Please select one answer:""",
+                current_options: gr.update(
+                    choices=question.options,
+                    value=current_answers[idx] if idx < len(current_answers) else None,
+                    visible=True,
+                    label=f"Select your answer for Question {idx + 1}:"
+                ),
+                question_counter: f"Question {idx + 1} of {len(questions)}",
+                question_box: gr.update(visible=True)
             }
 
-        def navigate(direction, current_idx, questions):
+        def navigate(direction, current_idx, questions, answers, current_answer):
+            # Save current answer
+            new_answers = list(answers)
+            if 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))
-            return new_idx, update_question_display(questions, new_idx)
+            display_updates = update_question_display(questions, new_idx, new_answers)
+            
+            return (
+                new_idx,
+                new_answers,
+                *display_updates.values()
+            )
 
-        # Event handlers
         def on_generate_questions(text, num_questions):
             success, questions = quiz_app.generate_questions(text, num_questions)
             if not success:
@@ -634,58 +657,43 @@ def create_quiz_interface():
                     question_display: "",
                     current_options: gr.update(choices=[], visible=False),
                     question_counter: "",
+                    question_box: gr.update(visible=False),
                     current_questions: [],
                     current_question_idx: 0,
-                    tabs: gr.update(selected=1)
+                    answer_state: [None] * 5,
+                    tabs: gr.update(selected=1),
+                    results_group: gr.update(visible=False)
                 }
             
             # Initialize first question
+            initial_answers = [None] * len(questions)
             return {
-                **update_question_display(questions, 0),
+                **update_question_display(questions, 0, initial_answers),
                 current_questions: questions,
                 current_question_idx: 0,
-                tabs: gr.update(selected=1)
+                answer_state: initial_answers,
+                tabs: gr.update(selected=1),
+                results_group: gr.update(visible=False)
             }
 
-        generate_btn.click(
-            fn=on_generate_questions,
-            inputs=[text_input, num_questions],
-            outputs=[
-                question_display,
-                current_options,
-                question_counter,
-                current_questions,
-                current_question_idx,
-                tabs
-            ]
-        )
-
-        # Navigation event handlers
-        prev_btn.click(
-            fn=navigate,
-            inputs=[-1, current_question_idx, current_questions],
-            outputs=[current_question_idx, question_display, current_options, question_counter]
-        )
-
-        next_btn.click(
-            fn=navigate,
-            inputs=[1, current_question_idx, current_questions],
-            outputs=[current_question_idx, question_display, current_options, question_counter]
-        )
-
-        # Submission handler
-        def on_submit(questions, current_answer):
-            if not current_answer:
+        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=False),
+                    results_group: gr.update(visible=True),
                     score_display: 0,
                     result_message: "",
+                    question_box: gr.update(visible=True),
                     tabs: gr.update(selected=1)
                 }
             
-            answers = [current_answer]  # Collect answers from current question
-            score, passed, feedback = quiz_app.calculate_score(answers)
+            score, passed, feedback = quiz_app.calculate_score(final_answers[:len(questions)])
             
             # Generate feedback HTML
             feedback_html = "# Assessment Results\n\n"
@@ -702,25 +710,113 @@ def create_quiz_interface():
                 </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>
+                """
+            
             return {
                 feedback_box: feedback_html,
                 results_group: gr.update(visible=True),
                 score_display: score,
-                result_message: "🎉 Passed!" if passed else "Please try again",
+                result_message: result_msg,
+                question_box: gr.update(visible=False),
                 tabs: gr.update(selected=2) if passed else gr.update()
             }
 
+        # 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
+        prev_btn.click(
+            fn=navigate,
+            inputs=[
+                -1,
+                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=navigate,
+            inputs=[
+                1,
+                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
+        current_options.change(
+            lambda a, i, old: [old[j] if j != i else a for j in range(len(old))],
+            inputs=[current_options, current_question_idx, answer_state],
+            outputs=[answer_state]
+        )
+
+        # Submission handler
         submit_btn.click(
             fn=on_submit,
-            inputs=[current_questions, current_options],
+            inputs=[
+                current_questions,
+                answer_state,
+                current_question_idx,
+                current_options
+            ],
             outputs=[
                 feedback_box,
                 results_group,
                 score_display,
                 result_message,
+                question_box,
                 tabs
             ]
         )
+
+        # 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