Update app.py

app.py

@@ -19,16 +19,21 @@ def save_audio(audio, dropdown_label, custom_label, speaker_name):
     if not speaker_name:
         raise gr.Error("User name cannot be empty 💥!", duration=5)
 
+    # Get sample rate and audio data from Gradio
+    sample_rate = audio[0]
+    audio_data = np.array(audio[1])
+
+    # Check if the audio length exceeds 1 second
+    max_length = 1  # in seconds
+    if len(audio_data) / sample_rate > max_length:
+        raise gr.Error("Recording is longer than 1 second 💥!", duration=5)
+
     # Generate speaker_id using a hash function to ensure consistency
     speaker_id = hashlib.sha256(speaker_name.encode()).hexdigest()[:8]
 
     # Generate file name
     filename = f"{datetime.datetime.now().strftime('%Y%m%d_%H%M%S')}_{speaker_id}_{label}.wav"
 
-    # Get sample rate and audio data from Gradio
-    sample_rate = audio[0]
-    audio_data = np.array(audio[1])
-
     # Save the audio file in wav format
     sf.write(filename, audio_data, sample_rate)
 
@@ -52,7 +57,7 @@ def create_interface():
         label_dropdown.change(toggle_custom_label, inputs=label_dropdown, outputs=custom_label)
 
         speaker_name = gr.Textbox(label="Enter User Name")
-        audio = gr.Audio(sources="microphone", type="numpy", label="Record Audio")
+        audio = gr.Audio(source="microphone", type="numpy", label="Record Audio")
 
         submit_button = gr.Button("Submit")