update

recording_tracker.json

@@ -46,5 +46,35 @@
     "audio_hash": "7908731e",
     "filename": "test/prompt2_7908731e_f7c7fa54.wav",
     "timestamp": "49f8bab8-7e44-4cb9-9155-3be03a3afddd"
+  },
+  {
+    "prompt": "",
+    "audio_hash": "fc7c499b",
+    "filename": "pilot_tempo_control6/_fc7c499b_0cf743d9.wav",
+    "timestamp": "8ac3802e-251b-43c6-9f34-258b0bb92648"
+  },
+  {
+    "prompt": "",
+    "audio_hash": "99c0ba0c",
+    "filename": "pilot_tempo_control6/_99c0ba0c_c1f82184.wav",
+    "timestamp": "c81ebd9d-7277-4fda-a413-61855483283e"
+  },
+  {
+    "prompt": "",
+    "audio_hash": "72860f45",
+    "filename": "pilot_tempo_control6/_72860f45_9a25752c.wav",
+    "timestamp": "d5bb27f3-46d7-490b-a27a-f5fff976e9d6"
+  },
+  {
+    "prompt": "",
+    "audio_hash": "866cedd5",
+    "filename": "pilot_tempo_control6/_866cedd5_08f3a22d.wav",
+    "timestamp": "6b38a09a-12d2-4e92-b475-0dd149a2b5d3"
+  },
+  {
+    "prompt": "",
+    "audio_hash": "cc1f9234",
+    "filename": "pilot_tempo_control6/_cc1f9234_2763ff75.wav",
+    "timestamp": "eda94b9d-30e5-44f7-89bd-4b9349b38a81"
   }
 ]

talk_arena/audio_collection.py

@@ -106,51 +106,26 @@ def upload_to_hf(local_path, repo_path):
 def on_submit(audio_input, recording_count):
     """Handle the submission of a recorded audio prompt"""
     if audio_input is None:
-        return (
-            gr.Markdown(f"# Recording {recording_count + 1}/{MAX_RECORDINGS}"),
-            gr.Markdown(USER_INSTRUCTIONS),
-            gr.Markdown(f"### \"{RECORDING_PROMPT}\""),
-            gr.Audio(value=None, label="Record your response"),
-            gr.Button("Submit Recording", interactive=False),
-            gr.Button("Next Recording", visible=False),
-            recording_count
-        )
-    
+        return next_prompt(recording_count)
+
     # Process the audio
     sr, y = audio_input
-    
-    # Generate a hash for this audio
     audio_hash = xxhash.xxh32(bytes(y)).hexdigest()
-
-    # Normalize audio
     y = y.astype(np.float32)
     y /= np.max(np.abs(y)) if np.max(np.abs(y)) > 0 else 1.0
 
-    # Create unique filename
     unique_id = str(uuid.uuid4())[:8]
     clean_prompt = RECORDING_PROMPT.replace(" ", "_").replace(".", "").replace(",", "")[:20]
     local_filename = f"outputs/{clean_prompt}_{audio_hash}_{unique_id}.wav"
-
-    # Save the original (non-resampled) audio
     sf.write(local_filename, y, sr, format="wav")
 
-    # Upload to HF dataset
     hf_path = f"{CATEGORY}/{clean_prompt}_{audio_hash}_{unique_id}.wav"
     upload_to_hf(local_filename, hf_path)
 
-    # Add to tracker
     tracker.add_recording(audio_hash, hf_path)
-    
-    # Show success message
-    return (
-        gr.Markdown(f"# Recording {recording_count + 1}/{MAX_RECORDINGS}"),
-        gr.Markdown(USER_INSTRUCTIONS),
-        gr.Markdown(f"### Recording successfully uploaded!"),
-        gr.Audio(value=None, label="Record your response"),
-        gr.Button("Submit Recording", interactive=False),
-        gr.Button("Next Recording", visible=True),
-        recording_count
-    )
+
+    # After successful upload, immediately move to the next prompt
+    return next_prompt(recording_count)
 
 def next_prompt(recording_count):
     """Move to the next prompt"""