Update app.py

app.py

@@ -1,52 +1,54 @@
 import gradio as gr
-from kokoro import TTS
+from kokoro import Kokoro
 import torch
 
-# Load model once at startup
-device = "cuda" if torch.cuda.is_available() else "cpu"
 
-# You can pick a voice that exists in Kokoro (e.g. 'af_heart', 'af_calypso', 'am_adam', etc.)
-VOICE = "af_heart"
+device = "cuda" if torch.cuda.is_available() else "cpu"
 
-print("🔄 Loading Kokoro-TTS model...")
-tts = TTS(voice=VOICE, device=device)
-print("✅ Model loaded successfully!")
+print("🔄 Loading Kokoro model...")
+model = Kokoro(device=device)
+print("✅ Kokoro loaded!")
 
 
-def generate_audio(text):
+def generate_audio(text, voice):
     if not text.strip():
         return None, "Please enter some text."
 
     try:
-        print("🎤 Generating audio...")
-        audio = tts.generate(text)          # returns numpy array (float32)
-        return (44100, audio), "Success!"
+        print("🎤 Generating...")
+        audio = model.tts(text, voice=voice)     # numpy float32 array
+        return (24000, audio), "Success!"
     except Exception as e:
         return None, f"Error: {str(e)}"
 
 
-### GRADIO UI ###
-with gr.Blocks(title="Kokoro-TTS Text → Audio") as demo:
-    gr.Markdown("## 🎧 Kokoro-TTS: Text to Speech Demo")
+voices = [
+    "af_heart",
+    "af_bella",
+    "af_nicole",
+    "am_adam",
+    "am_michael",
+    "bf_emma",
+    "bm_george"
+]
 
-    with gr.Row():
-        input_text = gr.Textbox(
-            label="Enter text",
-            placeholder="Type something to speak...",
-            lines=4
-        )
+
+with gr.Blocks(title="Kokoro-TTS") as demo:
+    gr.Markdown("## 🎧 Kokoro Text → Speech")
 
     with gr.Row():
-        output_audio = gr.Audio(label="Generated Audio")
-        status = gr.Textbox(label="Status")
+        txt = gr.Textbox(lines=4, label="Input Text")
+        voice_select = gr.Dropdown(voices, value="af_heart", label="Voice")
+
+    audio_out = gr.Audio(label="Generated Audio")
+    status = gr.Textbox(label="Status")
 
-    generate_button = gr.Button("Generate")
+    btn = gr.Button("Generate Audio")
 
-    generate_button.click(
+    btn.click(
         fn=generate_audio,
-        inputs=input_text,
-        outputs=[output_audio, status]
+        inputs=[txt, voice_select],
+        outputs=[audio_out, status]
     )
 
-if __name__ == "__main__":
-    demo.launch()
+demo.launch()