fix so we don't use a shared quiz state

app.py

@@ -5,10 +5,9 @@ from quiz import BenchmarkQuiz, BENCHMARKS, QuestionData
 
 mpl.rcParams["figure.dpi"] = 300
 
-quiz = BenchmarkQuiz()
 
 
-def update_quiz_screen(question_data: QuestionData):
+def update_quiz_screen(quiz: BenchmarkQuiz, question_data: QuestionData):
     quiz_state = quiz.state
     return {
         quiz_screen: gr.update(visible=True),
@@ -68,16 +67,18 @@ def update_score_screen(plot, results_data):
 
 
 def start_quiz_handler(benchmark_name):
+    quiz = BenchmarkQuiz()
     quiz.start_quiz(benchmark_name)
     question_data = quiz.update_question()
     return {
         start_screen: gr.update(visible=False),
         score_screen: gr.update(visible=False),
-        **update_quiz_screen(question_data),
+        **update_quiz_screen(quiz, question_data),
+        quiz_state: quiz,
     }
 
 
-def next_question_handler(answer_input, free_text_input):
+def next_question_handler(quiz, answer_input, free_text_input):
     answer = (
         answer_input
         if BENCHMARKS[quiz.state.benchmark_name]["type"] == "multiple_choice"
@@ -88,14 +89,16 @@ def next_question_handler(answer_input, free_text_input):
         return {
             quiz_screen: gr.update(visible=False),
             **update_score_screen(result["plot"], result["results_data"]),
+            quiz_state: quiz,
         }
     else:
-        return update_quiz_screen(result["question_data"])
+        return {**update_quiz_screen(quiz, result["question_data"]), quiz_state: quiz}
 
 
-def previous_question_handler():
+
+def previous_question_handler(quiz):
     question_data = quiz.previous_question()
-    return update_quiz_screen(question_data)
+    return {**update_quiz_screen(quiz, question_data), quiz_state: quiz}
 
 
 def reset_quiz_handler():
@@ -160,6 +163,8 @@ with demo:
             for card in result_cards:
                 with card:
                     gr.Markdown("")
+    
+    quiz_state = gr.State()
 
     for benchmark_name, button in benchmark_buttons.items():
         button.click(
@@ -175,12 +180,13 @@ with demo:
                 free_text_input,
                 next_button,
                 previous_button,
+                quiz_state,
             ],
         )
 
     next_button.click(
         fn=next_question_handler,
-        inputs=[answer_input, free_text_input],
+        inputs=[quiz_state, answer_input, free_text_input],
         outputs=[
             quiz_screen,
             score_screen,
@@ -194,12 +200,13 @@ with demo:
             results_container,
             *result_cards,
             *[child for card in result_cards for child in card.children],
+            quiz_state,
         ],
     )
 
     previous_button.click(
         fn=previous_question_handler,
-        inputs=[],
+        inputs=[quiz_state],
         outputs=[
             quiz_screen,
             question_number,
@@ -208,6 +215,7 @@ with demo:
             free_text_input,
             next_button,
             previous_button,
+            quiz_state,
         ],
     )