update

pages/arena.py

@@ -16,6 +16,8 @@ def random_data_selection():
 
 # Function to handle user selection and disable the radio
 def lock_selection(selected_option):
+    global global_selected_choice
+    global_selected_choice = selected_option  # Store the selected choice in the variable
     return gr.update(visible=True), selected_option, gr.update(interactive=False), gr.update(interactive=False)
 
 def create_arena():
@@ -71,14 +73,14 @@ Once the streaming is complete, you can choose the best response.\u2764\ufe0f"""
 
             submit_button.click(fn=lock_selection, inputs=choice, outputs=[output, output, choice, submit_button])
 
-            if choice == "Response 1":
+            if global_selected_choice == "Response 1":
                 prompt_id = random_selected_prompts[0]
-            elif choice == "Response 2":
+            elif global_selected_choice == "Response 2":
                 prompt_id = random_selected_prompts[1]
-            elif choice == "Response 3":
+            elif global_selected_choice == "Response 3":
                 prompt_id = random_selected_prompts[2]
             else:
-                raise ValueError(f"No corresponding response of {choice}")
+                raise ValueError(f"No corresponding response of {global_selected_choice}")
             
             for i in range(len(prompts)):
                 if prompts[i]['id'] == prompt_id: