added random song sampling

utils.py

@@ -8,7 +8,8 @@ import time
 import os
 from huggingface_hub import CommitScheduler
 from functools import partial
-
+import pandas as pd
+import numpy as np
 
 def enable_buttons_side_by_side():
     return tuple(gr.update(visible=True, interactive=True) for i in range(6))
@@ -19,6 +20,8 @@ def disable_buttons_side_by_side():
 
 os.makedirs('data', exist_ok = True)
 LOG_FILENAME = os.path.join('data', f'log_{datetime.now().isoformat()}.json')
+FLAG_FILENAME = os.path.join('data', f'flagged_{datetime.now().isoformat()}.json')
+
 enable_btn = gr.update(interactive=True, visible=True)
 disable_btn = gr.update(interactive=False)
 invisible_btn = gr.update(interactive=False, visible=False)
@@ -36,6 +39,31 @@ scheduler = CommitScheduler(
     every = 10,
 )
 
+df = pd.read_csv('/data/singfake_english.csv')
+df.filename = '/data/Songs/' + df.filename + '.mp3'
+
+
+indices = list(df.index)
+main_indices = indices.copy()
+
+def init_indices():
+    global indices, main_indices
+    indices = main_indices
+
+
+
+def pick_and_remove_two():
+    global indices
+    if len(indices) < 2:
+        init_indices()
+
+    np.random.shuffle(indices)
+    sel_indices = indices[:2].copy()
+    indices = indices[2:]
+    return sel_indices
+
+
+
 
 def vote_last_response(state, vote_type, request: gr.Request):
     with scheduler.lock:
@@ -49,6 +77,16 @@ def vote_last_response(state, vote_type, request: gr.Request):
           }
           fout.write(json.dumps(data) + "\n")
 
+def flag_last_response(state, vote_type, request: gr.Request):
+    with scheduler.lock:
+      with open(FLAG_FILENAME, "a") as fout:
+          data = {
+              "tstamp": round(time.time(), 4),
+              "type": vote_type,
+              "state": state.dict(),
+              "ip": get_ip(request),
+          }
+          fout.write(json.dumps(data) + "\n")
 
 
 class AudioStateIG:
@@ -74,15 +112,16 @@ def get_ip(request: gr.Request):
     return ip
 
 
-def get_song():
-    audio_a = "marine.mp3"
-    state = AudioStateIG("Random_Model")
-    return state, audio_a
+def get_song(idx, df = df):
+    row = df.loc[idx]
+    audio_path = row.filename
+    state = AudioStateIG(row['Bonafide Or Spoof'])
+    return state, audio_path
 
 def generate_songs(state0, state1):
-
-    state0, audio_a = get_song()
-    state1, audio_b = get_song()
+    idx0, idx1 = pick_and_remove_two()
+    state0, audio_a = get_song(idx0)
+    state1, audio_b = get_song(idx1)
 
     return state0, audio_a, state1, audio_b, "Model A: Vote to Reveal", "Model B: Vote to Reveal"
 
@@ -98,9 +137,9 @@ def leftvote_last_response(
     vote_last_response(
         [state0, state1], "leftvote", request
     )
-    return (disable_btn,) * 4 + (
-        gr.Markdown(f"### Model A: {state0.model_name.split('_')[1]}", visible=True),
-        gr.Markdown(f"### Model B: {state1.model_name.split('_')[1]}", visible=True))
+    return (disable_btn,) * 6 + (
+        gr.Markdown(f"### Model A: {state0.model_name}", visible=True),
+        gr.Markdown(f"### Model B: {state1.model_name}", visible=True))
 
 def rightvote_last_response(
     state0, state1, request: gr.Request
@@ -108,9 +147,9 @@ def rightvote_last_response(
     vote_last_response(
         [state0, state1], "rightvote", request
     )
-    return (disable_btn,) * 4 + (
-        gr.Markdown(f"### Model A: {state0.model_name.split('_')[1]}", visible=True),
-        gr.Markdown(f"### Model B: {state1.model_name.split('_')[1]}", visible=True))
+    return (disable_btn,) * 6 + (
+        gr.Markdown(f"### Model A: {state0.model_name}", visible=True),
+        gr.Markdown(f"### Model B: {state1.model_name}", visible=True))
 
 def tievote_last_response(
     state0, state1, request: gr.Request
@@ -118,9 +157,9 @@ def tievote_last_response(
     vote_last_response(
         [state0, state1], "tievote", request
     )
-    return (disable_btn,) * 4 + (
-        gr.Markdown(f"### Model A: {state0.model_name.split('_')[1]}", visible=True),
-        gr.Markdown(f"### Model B: {state1.model_name.split('_')[1]}", visible=True))
+    return (disable_btn,) * 6 + (
+        gr.Markdown(f"### Model A: {state0.model_name}", visible=True),
+        gr.Markdown(f"### Model B: {state1.model_name}", visible=True))
 
 def bothbadvote_last_response(
     state0, state1, request: gr.Request
@@ -128,8 +167,28 @@ def bothbadvote_last_response(
     vote_last_response(
         [state0, state1], "bothbadvote", request
     )
-    return (disable_btn,) * 4 + (
-        gr.Markdown(f"### Model A: {state0.model_name.split('_')[1]}", visible=True),
-        gr.Markdown(f"### Model B: {state1.model_name.split('_')[1]}", visible=True))
+    return (disable_btn,) * 6 + (
+        gr.Markdown(f"### Model A: {state0.model_name}", visible=True),
+        gr.Markdown(f"### Model B: {state1.model_name}", visible=True))
+
+def leftheard_last_response(
+    state, request: gr.Request
+):
+    vote_last_response(
+        [state], "leftheard", request
+    )
+    return (disable_btn,) * 6 + (
+        gr.Markdown(f"### Model A: {state0.model_name}", visible=True),
+        gr.Markdown(f"### Model B: {state1.model_name}", visible=True))
+
 
+def rightheard_last_response(
+    state, request: gr.Request
+):
+    vote_last_response(
+        [state], "rightheard", request
+    )
+    return (disable_btn,) * 6 + (
+        gr.Markdown(f"### Model A: {state0.model_name}", visible=True),
+        gr.Markdown(f"### Model B: {state1.model_name}", visible=True))