update

pages/arena.py

@@ -7,6 +7,12 @@ from utils.data import dataset
 from utils.multiple_stream import stream_data
 from pages.summarization_playground import get_model_batch_generation
 
+def random_data_selection():
+    datapoint = random.choice(dataset)
+    datapoint = datapoint['section_text'] + '\n\nDialogue:\n' + datapoint['dialogue']
+
+    return datapoint
+
 def create_arena():
     with open("prompt/prompt.json", "r") as file:
         json_data = file.read()
@@ -14,14 +20,21 @@ def create_arena():
 
     with gr.Blocks() as demo:
         with gr.Group():
-            datapoint = random.choice(dataset)
-            datapoint = datapoint['section_text'] + '\n\nDialogue:\n' + datapoint['dialogue']
+            datapoint = random_data_selection()
             gr.Markdown("""This arena is designed to compare different prompts. Click the button to stream responses from randomly shuffled prompts. Each column represents a response generated from one randomly selected prompt.
 
 Once the streaming is complete, you can choose the best response.\u2764\ufe0f""")
 
-            gr.Textbox(label="datapoint", lines=10, placeholder="Datapoints to test...", value=datapoint)
-            submit_button = gr.Button("✨ Click to Streaming ✨")
+            data_textbox = gr.Textbox(label="Data", lines=10, placeholder="Datapoints to test...", value=datapoint)
+            with gr.row():
+                random_selection_button = gr.Button("Change Data")
+                submit_button = gr.Button("✨ Click to Streaming ✨")
+
+            random_selection_button.click(
+                fn=random_data_selection,
+                inputs=[],
+                outputs=data_textbox
+            )
     
             with gr.Row():
                 columns = [gr.Textbox(label=f"Prompt {i+1}", lines=10) for i in range(len(prompts))]