Update app.py

app.py

@@ -1,40 +1,60 @@
 import gradio as gr
 import pandas as pd
+import os
+import random
 
-# Predefined list of voices
-voices = ["Voice1", "Voice2", "Voice3"]
+# Path to voice actors folder
+voice_actors_folder = "tortoise/voices/"
 
-# Function to extract unique names from CSV
-def extract_unique_names(csv_file):
-    # Read the CSV
+# Get the list of voice actors
+voice_actors = [va for va in os.listdir(voice_actors_folder) if va != "cond_latent_example" and va != ".DS_Store"]
+male_voice_actors = [va for va in voice_actors if va.endswith(".M")]
+female_voice_actors = [va for va in voice_actors if va.endswith(".F")]
+
+# Function to extract unique speakers from CSV
+def extract_unique_speakers(csv_file):
     df = pd.read_csv(csv_file)
+    # Extract unique speakers from "Speaker" column
+    unique_speakers = df["Speaker"].unique().tolist()
+    return unique_speakers
+
+# Function to get a random default voice actor for each speaker
+def get_random_voice_for_speaker_fast(speaker):
+    selected_voice_actors = voice_actors  # default to all voice actors
+
+    if speaker.endswith(".M") and male_voice_actors: 
+        selected_voice_actors = male_voice_actors
+    elif speaker.endswith(".F") and female_voice_actors:
+        selected_voice_actors = female_voice_actors
+    elif speaker.endswith(".?"):
+        selected_voice_actors = male_voice_actors + female_voice_actors
     
-    # Extract unique names from "Speaker" column
-    unique_names = df["Speaker"].unique().tolist()
+    if not selected_voice_actors:  # If list is empty, default to all voice actors
+        selected_voice_actors = male_voice_actors + female_voice_actors
     
-    return unique_names
+    return random.choice(selected_voice_actors)
 
-# Extract names from the CSV file
-names = extract_unique_names("book.csv")
+# Extract speakers from the CSV file
+speakers = extract_unique_speakers("book.csv")
 
-# Dictionary to store selected voices
-selected_voices = {name: None for name in names}
+# 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 update the dictionary and display the result
 def update_voice(*args):
-    for name, selected_voice in zip(names, args):
-        selected_voices[name] = selected_voice
+    for speaker, selected_voice in zip(speakers, args):
+        selected_voices[speaker] = selected_voice
     
     # Create a result string to display
-    result = "\n".join([f"For {name}: selected voice {voice}" for name, voice in selected_voices.items()])
+    result = "\n".join([f"For {speaker}: selected voice actor {voice}" for speaker, voice in selected_voices.items()])
     return result
 
 # Creating Gradio interface
 def create_interface():
     inputs = []
-    for name in names:
-        # Create a dropdown for each name with the predefined voices
-        dropdown = gr.Dropdown(choices=voices, label=f"Select voice for {name}")
+    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)
     
     interface = gr.Interface(