add new prompts

app.py

@@ -1,19 +1,27 @@
 import numpy as np
 import gradio as gr
 from bark import SAMPLE_RATE, generate_audio, preload_models
+from bark.generation import SUPPORTED_LANGS
 
 DEBUG_MODE = False
 
 if not DEBUG_MODE:
     _ = preload_models()
 
+AVAILABLE_PROMPTS = ["Unconditional", "Announcer"]
+PROMPT_LOOKUP = {}
+for _, lang in SUPPORTED_LANGS:
+    for n in range(10):
+        label = f"Speaker {n} ({lang})"
+        AVAILABLE_PROMPTS.append(label)
+        PROMPT_LOOKUP[label] = f"{lang}_speaker_{n}"
+PROMPT_LOOKUP["Unconditional"] = None
+PROMPT_LOOKUP["Announcer"] = "announcer"
+
 default_text = "Hello, my name is Suno. And, uh — and I like pizza. [laughs]\nBut I also have other interests such as playing tic tac toe."
 
 def gen_tts(text, history_prompt, temp_semantic, temp_waveform):
-    if history_prompt == "Unconditional":
-        history_prompt = None
-    else:
-        history_prompt = history_prompt.lower().replace(" ", "_").replace("speaker", "speech")
+    history_prompt = PROMPT_LOOKUP[history_prompt]
     if DEBUG_MODE:
         audio_arr = np.zeros(SAMPLE_RATE)
     else:
@@ -26,10 +34,7 @@ iface = gr.Interface(
     fn=gen_tts, 
     inputs=[
         gr.Textbox(label="Input Text", lines=3, value=default_text), 
-        gr.Dropdown(
-            ["Unconditional"] + [f"Speaker {n}" for n in range(8)] + [f"Music {n}" for n in range(6)], 
-            value="None", label="Acoustic Prompt", info="This choice primes the model on how to condition the generated audio."
-        ),
+        gr.Dropdown(AVAILABLE_PROMPTS, value="None", label="Acoustic Prompt", info="This choice primes the model on how to condition the generated audio."),
         gr.Slider(minimum=0, maximum=1, step=0.01, value=0.7, label="Temp 1", info="Gen. temperature of semantic tokens. (lower is more conservative, higher is more diverse)"),
         gr.Slider(minimum=0, maximum=1, step=0.01, value=0.7, label="Temp 2", info="Gen. temperature of waveform tokens. (lower is more conservative, higher is more diverse)"),
     ],