Update app.py

app.py

@@ -83,6 +83,17 @@ def handle_answer(answer, mock_index, score, incorrect_questions):
     
     return gr.update(), gr.update(visible=False), gr.update(), gr.update(), gr.update(), gr.update(), mock_index, score, incorrect_questions
 
+def restart_test():
+    mock_index = 0
+    score = 0
+    incorrect_questions = []
+    data = load_mocktest()
+    if data:
+        question, ans1, ans2, ans3, ans4, _ = display_mock_question(mock_index, score, incorrect_questions)
+        return mock_index, score, incorrect_questions, question, ans1, ans2, ans3, ans4, gr.update(visible=False), gr.update(visible=False)
+    else:
+        return mock_index, score, incorrect_questions, "Error loading mock test data", "", "", "", "", gr.update(visible=False), gr.update(visible=False)
+
 def update_question(mock_index, score, incorrect_questions):
     data = load_mocktest()
     if mock_index >= len(data):
@@ -91,10 +102,6 @@ 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, visible=True), gr.update(value=ans2, visible=True), gr.update(value=ans3, visible=True), gr.update(value=ans4, visible=True)
 
-# Function to restart the test
-def restart_test():
-    return 0, "", [], *display_mock_question(0, 0, []), gr.update(visible=False), gr.update(visible=True)
-
 
 # Gradio interface
 with gr.Blocks() as demo:
@@ -158,6 +165,14 @@ with gr.Blocks() as demo:
                 )
 
             # 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])
+            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]
+            ).then(
+                update_question,
+                inputs=[mock_index, score, incorrect_questions],
+                outputs=[question_output, answer_1, answer_2, answer_3, answer_4]
+            )
 
 demo.launch()