Restart test on the end

app.py

@@ -30,7 +30,7 @@ def question_interface(question_state, answer):
         name = f"Вітаємо, Ви {test.full_names[best_id]}!"
         title = test.titles[best_id]
         image_path = test.get_photos_path(best_id)
-        return None, name, radio, title, image_path
+        return test.zero_state, name, radio, title, image_path
 
     # Display the current question
     question = test.get_question(current_question_index)
@@ -50,7 +50,7 @@ def question_interface(question_state, answer):
 
 
 with gr.Blocks() as blocks:
-    question_state = gr.State(value = {"value":-1, "responses":[]})
+    question_state = gr.State(value = test.zero_state)
     gr.Markdown("<h1 style='text-align: center; margin-bottom: 1rem'>"
                 + title
                 + "</h1>")