added placeholder for flagging

app.py

@@ -1,10 +1,9 @@
 import gradio as gr
 from utils import *
 import os
-from huggingface_hub import snapshot_download
 
 def build_demo():
-    
+
 
     with gr.Blocks() as demo:
       state0 = gr.State()
@@ -23,9 +22,15 @@ def build_demo():
 
                 with gr.Row():
                     with gr.Column():
-                        audio_a = gr.Audio(label="Model A")
+                        leftheard_btn = gr.Button(
+                        value="🚩 Heard this song before", visible=False, interactive=False
+                    )
+                        audio_a = gr.Audio(label="Model A", max_length = 100, show_download_button = False)
                     with gr.Column():
-                        audio_b = gr.Audio(label="Model B")
+                        rightheard_btn = gr.Button(
+                        value="🚩 Heard this song before", visible=False, interactive=False
+                    )
+                        audio_b = gr.Audio(label="Model B", max_length = 100, show_download_button = False)
                 with gr.Row():
                     leftvote_btn = gr.Button(
                         value="👈  A is Fake", visible=False, interactive=False
@@ -38,29 +43,40 @@ def build_demo():
                         value="👎  Both are Real", visible=False, interactive=False
                     )
 
-        btn_list = [leftvote_btn, rightvote_btn, tie_btn, bothbad_btn]
+        btn_list = [leftvote_btn, rightvote_btn, tie_btn, bothbad_btn, leftheard_btn, rightheard_btn]
 
         leftvote_btn.click(
           leftvote_last_response,
           inputs=[state0, state1],
-          outputs=[leftvote_btn, rightvote_btn, tie_btn, bothbad_btn, model_selector_left, model_selector_right]
-      )
+          outputs=[leftvote_btn, rightvote_btn, tie_btn, bothbad_btn,  leftheard_btn, rightheard_btn, model_selector_left, model_selector_right]
+        )
         rightvote_btn.click(
           rightvote_last_response,
           inputs=[state0, state1],
-          outputs=[leftvote_btn, rightvote_btn, tie_btn, bothbad_btn, model_selector_left, model_selector_right]
-      )
+          outputs=[leftvote_btn, rightvote_btn, tie_btn, bothbad_btn, leftheard_btn, rightheard_btn, model_selector_left, model_selector_right]
+        )
         tie_btn.click(
           tievote_last_response,
           inputs=[state0, state1],
-          outputs=[leftvote_btn, rightvote_btn, tie_btn, bothbad_btn, model_selector_left, model_selector_right]
-      )
+          outputs=[leftvote_btn, rightvote_btn, tie_btn, bothbad_btn, leftheard_btn, rightheard_btn, model_selector_left, model_selector_right]
+        )
         bothbad_btn.click(
           bothbadvote_last_response,
           inputs=[state0, state1],
-          outputs=[leftvote_btn, rightvote_btn, tie_btn, bothbad_btn, model_selector_left, model_selector_right]
-      )
+          outputs=[leftvote_btn, rightvote_btn, tie_btn, bothbad_btn, leftheard_btn, rightheard_btn, model_selector_left, model_selector_right]
+        )
 
+        leftheard_btn.click(
+            leftheard_last_response,
+            inputs=state0,
+            outputs=[leftvote_btn, rightvote_btn, tie_btn, bothbad_btn, leftheard_btn, rightheard_btn, model_selector_left, model_selector_right]
+        )
+
+        rightheard_btn.click(
+            rightheard_last_response,
+            inputs=state1,
+            outputs=[leftvote_btn, rightvote_btn, tie_btn, bothbad_btn, leftheard_btn, rightheard_btn, model_selector_left, model_selector_right]
+        )
 
         new_round_button = gr.Button("New Round")
         new_round_button.click(generate_songs, [state0, state1], [state0, audio_a, state1, audio_b, model_selector_left, model_selector_right]).then(
@@ -77,7 +93,8 @@ def build_demo():
 if __name__ == "__main__":
 
     SONG_SOURCE = os.getenv("SONG_SOURCE")
-    snapshot_download(repo_id=SONG_SOURCE, repo_type="dataset", token = access_token, local_dir = '/data/')
+    TOKEN = os.getenv("TOKEN")
+    snapshot_download(repo_id=SONG_SOURCE, repo_type="dataset", token = TOKEN, local_dir = '/data/')
     # elo_results_file, leaderboard_table_file = load_elo_results(elo_results_dir)
     demo = build_demo()
     demo.queue(max_size=20).launch(server_name="0.0.0.0")