Update app.py

app.py

@@ -115,12 +115,46 @@ def display_videos(type, prompt, model_name_1, model_name_2):
 
 def record_user_feedback_a(model_name1, model_name2, type, prompt):
     # 0 means model A  better, 1 means model B better 
-    with open("user_feedback.csv",'a') as f:
+    hf_api.hf_hub_download(
+            repo_id = "Vchitect/VBench_human_annotation",
+            filename = "arena_feedback.csv",
+            repo_type = "dataset",
+            local_dir = './'
+        )
+    with open("arena_feedback.csv",'a') as f:
         f.write(f"{model_name1}\t{model_name2}\t{type}\t{prompt}\t{0}\n")
+    hf_api.upload_file(
+        path_or_fileobj="arena_feedback.csv",
+        path_in_repo="arena_feedback.csv",
+        repo_id="Vchitect/VBench_human_annotation",
+        token=hf_token,
+        repo_type="dataset",
+        commit_message="[From VBench Arena] user feedback",
+    )
+    return gr.update(visible=False),gr.update(visible=False)
+    
 def record_user_feedback_b(model_name1, model_name2, type, prompt):
     # 0 means model A  better, 1 means model B better 
-    with open("user_feedback.csv",'a') as f:
+    hf_api.hf_hub_download(
+            repo_id = "Vchitect/VBench_human_annotation",
+            filename = "arena_feedback.csv",
+            repo_type = "dataset",
+            local_dir = './'
+        )
+    with open("arena_feedback.csv",'a') as f:
         f.write(f"{model_name1}\t{model_name2}\t{type}\t{prompt}\t{1}\n")  
+    hf_api.upload_file(
+        path_or_fileobj="arena_feedback.csv",
+        path_in_repo="arena_feedback.csv",
+        repo_id="Vchitect/VBench_human_annotation",
+        token=hf_token,
+        repo_type="dataset",
+        commit_message="[From VBench Arena] user feedback",
+    )
+    return gr.update(visible=False),gr.update(visible=False)
+
+def show_feedback_button():
+    return gr.update(visible=True),gr.update(visible=True)
 
 with gr.Blocks() as interface:
     gr.Markdown("# VBench Video Arena")
@@ -144,8 +178,8 @@ If you are interested, you can also leave your comments.""")
             model_name_2_output = gr.Dropdown(label="Model Name 2", choices=model_names, value=model_names[1])
             video_output_2 = gr.Video(label="Video 2")
     with gr.Row():
-        feed0 = gr.Button("👈 Model A is better")
-        feed1 = gr.Button("👉 Model B is better")
+        feed0 = gr.Button("👈 Model A is better",visible=False)
+        feed1 = gr.Button("👉 Model B is better",visible=False)
     type_output.change(fn=update_prompt_options, inputs=[type_output], outputs=[prompt_output])
     
     
@@ -154,9 +188,14 @@ If you are interested, you can also leave your comments.""")
         outputs=[video_output_1, video_output_2,model_name_1_output, model_name_2_output, type_output, prompt_placeholder]
     ).then(fn=update_prompt_options,
           inputs=[type_output],
-          outputs=[prompt_output]).then(fn=update_prompt_options,
+          outputs=[prompt_output]
+    ).then(fn=update_prompt_options,
           inputs=[type_output,prompt_placeholder],
-          outputs=[prompt_output])
+          outputs=[prompt_output]
+    ).then(
+        fn= show_feedback_button,
+        outputs=[feed0, feed1]
+    )
 
 
     display_button.click(
@@ -167,11 +206,13 @@ If you are interested, you can also leave your comments.""")
 
     feed0.click(
         fn = record_user_feedback_a,
-        inputs=[model_name_1_output, model_name_2_output, type_output, prompt_placeholder]
+        inputs=[model_name_1_output, model_name_2_output, type_output, prompt_placeholder],
+        outputs=[feed0, feed1]
     )
     feed1.click(
         fn = record_user_feedback_b,
-        inputs=[model_name_1_output, model_name_2_output, type_output, prompt_placeholder]
+        inputs=[model_name_1_output, model_name_2_output, type_output, prompt_placeholder],
+        outputs=[feed0, feed1]
     )
 
 interface.launch()