Update app.py

app.py

@@ -2,41 +2,61 @@ import gradio as gr
 import soundfile as sf
 import datetime
 import numpy as np
+import hashlib
 
+# Function to save audio with label and speaker name
+def save_audio(audio, dropdown_label, custom_label, speaker_name):
+    # Determine the final label
+    label = custom_label if dropdown_label == "Custom" else dropdown_label
 
-# 錄音函數
-def save_audio(audio, label, sample_rate, bit_depth):
-    # 產生檔案名稱
-    filename = f"{datetime.datetime.now().strftime('%Y%m%d_%H%M%S')}_{label}_{sample_rate}Hz_{bit_depth}bit.wav"
+    if not label:
+        raise gr.Error("Label cannot be empty 💥!", duration=5)
+    if not speaker_name:
+        raise gr.Error("User name cannot be empty 💥!", duration=5)
 
-    # 將音頻數據轉換成 numpy 陣列
-    audio_data = np.array(audio[1])
+    # 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"
 
-    # 將錄音儲存為 wav 檔案
-    sf.write(filename, audio_data, int(sample_rate), subtype=f'PCM_{bit_depth}')
+    # Get sample rate and audio data from Gradio
+    sample_rate = audio[0]
+    audio_data = np.array(audio[1])
 
-    # 返回檔案名作為下載連結
-    return filename, gr.File(filename)
+    # Save the audio file in wav format
+    sf.write(filename, audio_data, sample_rate)
 
+    # Return only the download link
+    return gr.File(filename)
 
-# 介面設計
+# Interface design using gr.Blocks
 def create_interface():
-    sample_rate = gr.Dropdown(["44100", "48000"], label="選擇採樣率", value="44100")
-    bit_depth = gr.Dropdown(["16", "24"], label="選擇位深度", value="16")
-    label = gr.Textbox(label="輸入標記")
-    audio = gr.Audio(sources="microphone", type="numpy", label="錄音")
-
-    interface = gr.Interface(
-        fn=save_audio,
-        inputs=[audio, label, sample_rate, bit_depth],
-        outputs=["text", "file"],
-        title="錄音程式",
-        description="錄音最大長度為 1 秒，儲存為 WAV 格式。",
-    )
-    return interface
-
-
-# 啟動介面
+    with gr.Blocks() as demo:
+        labels = ["Label1", "Label2", "Label3", "Custom"]  # Add a "Custom" option for custom label
+        label_dropdown = gr.Dropdown(choices=labels, label="Select Label")
+        custom_label = gr.Textbox(label="Enter Custom Label", visible=False)  # Initially hidden
+
+        # Function to show/hide custom label textbox based on dropdown selection
+        def toggle_custom_label(selected_label):
+            return gr.update(visible=True) if selected_label == "Custom" else gr.update(visible=False)
+
+        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")
+
+        submit_button = gr.Button("Submit")
+
+        submit_button.click(
+            fn=save_audio,
+            inputs=[audio, label_dropdown, custom_label, speaker_name],
+            outputs=[gr.File()],
+        )
+
+    return demo
+
+# Launch the interface
 if __name__ == "__main__":
     interface = create_interface()
     interface.launch()