Update app.py

app.py

@@ -66,23 +66,27 @@ def display_mock_question(index, score, incorrect_questions):
     return question, all_answers[0], all_answers[1], all_answers[2], all_answers[3], correct_answer
 
 
-def handle_answer(index, answer, score, incorrect_questions):
+def handle_answer(answer, mock_index, score, incorrect_questions):
     data = load_mocktest()
-    correct_answer = data[index]["correct-answer"]
+    correct_answer = data[mock_index]["correct-answer"]
     
     if answer == correct_answer:
         score += 1
     else:
-        incorrect_questions.append(data[index]["question"])
+        incorrect_questions.append(data[mock_index]["question"])
     
-    index += 1
-    if index >= len(data):
+    mock_index += 1
+    if mock_index >= len(data):
         incorrect_str = "Questions to review:\n\n" + "\n\n".join(incorrect_questions) if incorrect_questions else "No questions to review."
         result = f"Your score: {score}/{len(data)}\n\n{incorrect_str}"
-        return gr.update(visible=False), result, gr.update(visible=True), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), score, incorrect_questions
+        return gr.update(value=result), gr.update(visible=True), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), mock_index, score, incorrect_questions
     
-    question, ans1, ans2, ans3, ans4, correct = display_mock_question(index, score, incorrect_questions)
-    return gr.update(value=question), ans1, ans2, ans3, ans4, gr.update(visible=False), gr.update(visible=False), score, incorrect_questions
+    question, ans1, ans2, ans3, ans4, correct = display_mock_question(mock_index, score, incorrect_questions)
+    return gr.update(value=question), gr.update(visible=False), gr.update(value=ans1), gr.update(value=ans2), gr.update(value=ans3), gr.update(value=ans4), mock_index, score, incorrect_questions
+
+def update_question(mock_index, score, incorrect_questions):
+    question, ans1, ans2, ans3, ans4, correct = display_mock_question(mock_index, score, incorrect_questions)
+    return gr.update(value=question), gr.update(value=ans1), gr.update(value=ans2), gr.update(value=ans3), gr.update(value=ans4)
 
 # Function to restart the test
 def restart_test():
@@ -136,17 +140,21 @@ with gr.Blocks() as demo:
             restart_button = gr.Button("Restart Test", visible=False)
             
             # Initialize the first question
-            demo.load(display_mock_question, inputs=[mock_index, score, incorrect_questions], outputs=[question_output, answer_1, answer_2, answer_3, answer_4])
+            demo.load(update_question, inputs=[mock_index, score, incorrect_questions], outputs=[question_output, answer_1, answer_2, answer_3, answer_4])
 
             # Handle answer selection
             for answer_btn in [answer_1, answer_2, answer_3, answer_4]:
                 answer_btn.click(
                     handle_answer, 
-                    inputs=[mock_index, answer_btn, score, incorrect_questions], 
-                    outputs=[question_output, answer_1, answer_2, answer_3, answer_4, result_output, restart_button, score, incorrect_questions]
+                    inputs=[answer_btn, mock_index, score, incorrect_questions],
+                    outputs=[question_output, restart_button, answer_1, answer_2, answer_3, answer_4, mock_index, score, incorrect_questions]
+                ).then(
+                    update_question,
+                    inputs=[mock_index, score, incorrect_questions],
+                    outputs=[question_output, answer_1, answer_2, answer_3, answer_4]
                 )
 
             # Restart test button
             restart_button.click(restart_test, inputs=[], outputs=[mock_index, score, incorrect_questions, question_output, answer_1, answer_2, answer_3, answer_4, result_output, restart_button])
-# Launch the app
+
 demo.launch()