Update app.py

app.py

@@ -6,10 +6,11 @@ import hashlib
 
 
 MAX_LENGTH = 1.5  # in seconds
-LABLES = ["Hi, doctor", "docker home", "Ask doctor", "Hey, daughter", "Hi, gemtek", "Hi, AI"]
+LABLES = ["Hi, doctor", "doctor home", "Ask doctor", "Hey, daughter", "Hi, gemtek", "Hi, AI", "Custom"]
+
+# Initialize a dictionary to store file paths by speaker_id
+uploaded_files = {}
 
-# Initialize a list to store file paths
-uploaded_files = []
 help_en = f"""
 ### Usage Instructions:
 1. **Select a Label**: Choose a label from the dropdown menu, or select "Custom" to enter your own label.
@@ -39,6 +40,9 @@ def save_audio(audio, dropdown_label, custom_label, speaker_name):
     if not speaker_name:
         raise gr.Error("User name cannot be empty 💥!", duration=5)
 
+    # Generate speaker_id using a hash function to ensure consistency
+    speaker_id = hashlib.sha256(speaker_name.encode()).hexdigest()[:8]
+
     # Get sample rate and audio data from Gradio
     sample_rate = audio[0]
     audio_data = np.array(audio[1])
@@ -50,26 +54,21 @@ def save_audio(audio, dropdown_label, custom_label, speaker_name):
     if audio_length > MAX_LENGTH:
         raise gr.Error(f"Recording is longer than {MAX_LENGTH} 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"
 
     # Save the audio file in wav format
     sf.write(filename, audio_data, sample_rate)
 
-    # Add the new file path to the list of uploaded files
-    uploaded_files.append(filename)
+    # Add the new file path to the list of uploaded files by speaker_id
+    if speaker_id not in uploaded_files:
+        uploaded_files[speaker_id] = []
+    uploaded_files[speaker_id].append(filename)
 
     # Create an info message
     info_message = f"Recorded audio length: {audio_length:.2f} seconds."
 
-    # Display the info message using gr.Info
-    gr.Info(info_message, duration=5)
-
-    # Return the list of all file paths for download
-    return uploaded_files
+    return uploaded_files[speaker_id], info_message
 
 # Interface design using gr.Blocks
 def create_interface():
@@ -95,16 +94,23 @@ def create_interface():
         # Create a list to display all uploaded files
         file_list = gr.Files(label="Download your recordings")
 
-        # Create an info display area
-        info_display = gr.Info()
+        # Create an info display area for feedback messages
+        info_display = gr.Markdown()
 
         # Add the info output to display recording length
         submit_button.click(
             fn=save_audio,
             inputs=[audio, label_dropdown, custom_label, speaker_name],
-            outputs=[file_list],
+            outputs=[file_list, info_display],
         )
 
+        # Update file list based on the current speaker
+        def get_user_files(speaker_name):
+            speaker_id = hashlib.sha256(speaker_name.encode()).hexdigest()[:8]
+            return uploaded_files.get(speaker_id, [])
+
+        speaker_name.submit(fn=get_user_files, inputs=speaker_name, outputs=file_list)
+
         # Add usage instructions in both English and Chinese
         gr.Markdown(help_en)
         gr.Markdown(help_zh_tw)