Update app.py

app.py

@@ -40,6 +40,15 @@ speakers = extract_unique_speakers("book.csv")
 # Dictionary to store selected voice actors for each speaker
 selected_voices = {speaker: get_random_voice_for_speaker_fast(speaker) for speaker in speakers}
 
+# Function to find and return a sample audio file for the selected voice actor
+def get_voice_sample(voice_actor):
+    voice_actor_path = os.path.join(voice_actors_folder, voice_actor)
+    # Look for any .wav or .mp3 files in the voice actor's folder
+    for file in os.listdir(voice_actor_path):
+        if file.endswith(".wav") or file.endswith(".mp3"):
+            return os.path.join(voice_actor_path, file)
+    return None
+
 # Function to update the dictionary and display the result
 def update_voice(*args):
     for speaker, selected_voice in zip(speakers, args):
@@ -49,18 +58,40 @@ def update_voice(*args):
     result = "\n".join([f"For {speaker}: selected voice actor {voice}" for speaker, voice in selected_voices.items()])
     return result
 
+# Function to return the voice sample file for a given speaker
+def play_sample(speaker):
+    selected_voice = selected_voices[speaker]
+    sample_file = get_voice_sample(selected_voice)
+    if sample_file:
+        return sample_file
+    else:
+        return "No audio sample available."
+
 # Creating Gradio interface
 def create_interface():
     inputs = []
+    play_buttons = []
+    
     for speaker in speakers:
         # Create a dropdown for each speaker with the list of all voice actors
         dropdown = gr.Dropdown(choices=voice_actors, label=f"Select voice actor for {speaker}", value=selected_voices[speaker])
         inputs.append(dropdown)
+        
+        # Add a button to play the sample for each selected voice actor
+        play_buttons.append(gr.Audio(source="filepath", label=f"Play sample for {speaker}", type="filepath", interactive=True, value=None))
     
+    def update_interface(*args):
+        # Update voice selection
+        update_voice(*args)
+        
+        # Update the play button with the audio file
+        audio_samples = [play_sample(speaker) for speaker in speakers]
+        return update_voice(*args), audio_samples
+
     interface = gr.Interface(
-        fn=update_voice,
+        fn=update_interface,
         inputs=inputs,
-        outputs="text",
+        outputs=["text"] + play_buttons,
         live=True
     )
     return interface