update

pages/arena.py

@@ -2,9 +2,6 @@
 import random
 import gradio as gr
 import json
-import logging
-import gc
-import torch
 
 from utils.data import dataset
 from utils.multiple_stream import stream_data
@@ -39,6 +36,16 @@ def create_arena():
                 show_progress=False
             )
 
+            choice = gr.Radio(label="Choose the best response:", choices=["Response 1", "Response 2", "Response 3"])
+    
+            submit_button = gr.Button("Submit")
+
+            submit_button.click(
+                #fn=lambda response1, response2, response3, choice: save_to_db(eval(choice.lower())),
+                inputs=[choice],
+                outputs=f"Response '{choice}' saved successfully!"
+            )
+
     return demo
 
 if __name__ == "__main__":