constant 25 s song length

utils.py

@@ -155,13 +155,21 @@ def random_cut_length(audio_data, max_length, sample_rate):
     audio_data_cut = audio_data[start_point*sample_rate : end_point*sample_rate]
     return audio_data_cut, length_picked
 
+def constant_cut_length(audio_data, max_length, sample_rate, length_picked = 25):
+    if max_length <= length_picked:
+        return audio_data, max_length
+        
+    start_point = np.random.randint(0, max_length - length_picked)
+    end_point = start_point + length_picked
+    audio_data_cut = audio_data[start_point*sample_rate : end_point*sample_rate]
+    return audio_data_cut, length_picked
 
-def generate_songs(state, randomized = True):
+def generate_songs(state, song_cut_function = constant_cut_length):
     idx= pick_and_remove_one()
     state, audio = get_song(idx)
-    if randomized:
+    if song_cut_function is not None:
       audio_data, sample_rate = librosa.load(audio, sr=None)
-      audio_cut, new_length = random_cut_length(audio_data, state.row.duration, sample_rate)
+      audio_cut, new_length = song_cut_function(audio_data, state.row.duration, sample_rate)
       state.update_duration(new_length)
       return state, (sample_rate, audio_cut), "Vote to Reveal Label",