added random length cutting

utils.py

@@ -11,6 +11,7 @@ from functools import partial
 import pandas as pd
 import numpy as np
 from huggingface_hub import snapshot_download
+import librosa
 
 def enable_buttons_side_by_side():
     return tuple(gr.update(visible=True, interactive=True) for i in range(6))
@@ -94,19 +95,23 @@ class AudioStateIG:
     def __init__(self, row):
         self.conv_id = uuid4().hex
         self.row = row
+        self.new_duration = None
 
     def dict(self):
         base = {
             "conv_id": self.conv_id,
             "label": self.row.label,
             "filename": self.row.filename,
-            "duration": self.row.duration, 
+            "duration": self.row.duration if self.new_duration is None else self.new_duration, 
             "song_id": str(self.row.id), 
             "source": self.row.source, 
-            "algorithm": self.row.algorithm
+            "algorithm": self.row.algorithm,
             }
         return base
 
+    def update_duration(self, duration):
+        self.new_duration = duration
+
 def get_ip(request: gr.Request):
     if request:
         if "cf-connecting-ip" in request.headers:
@@ -124,13 +129,31 @@ def get_song(idx, df = df, filenames = filenames):
     state = AudioStateIG(row)
     return state, audio_path
 
-def generate_songs(state):
-    idx= pick_and_remove_one()
-    state, audio = get_song(idx)
+def random_cut_length(audio_data, max_length, sample_rate):
+    if max_length > 125:
+      options = [125, 55, 25]
+
+    elif max_length > 55:
+      options = [55, 25]
     
+    elif max_length > 25:
+      options = [25]
 
-    return state, audio, "Vote to Reveal Label", 
+    else:
+      return audio_data, max_length
 
+
+def generate_songs(state, randomized = True):
+    idx= pick_and_remove_one()
+    state, audio = get_song(idx)
+    if randomized:
+      audio_data, sample_rate = librosa.load(audio, sr=None)
+      audio_cut, new_length = random_cut_length(audio_data, state.row.duration, sample_rate)
+      state.update_duration(new_length)
+      return state, (sample_rate, audio_cut), "Vote to Reveal Label",
+        
+    return state, audio, "Vote to Reveal Label",
+    
 def fake_last_response(
     state, request: gr.Request
 ):