From fd53cdb529fb8eb5f3ad56d623cd1f7b9b54742c Mon Sep 17 00:00:00 2001
From: Ayo <ayo@ayco.io>
Date: Wed, 3 Sep 2025 22:59:28 +0200
Subject: [PATCH] feat: allow using all voices available

---
 tts.py | 12 +++++-------
 1 file changed, 5 insertions(+), 7 deletions(-)

diff --git a/tts.py b/tts.py
index dc7bffa..b1fc029 100644
--- a/tts.py
+++ b/tts.py
@@ -44,12 +44,10 @@ def parse_args():
 def main():
     args=parse_args()
     pipeline = KPipeline(lang_code='a', device=args.device, repo_id='hexgrad/Kokoro-82M')
-    voice=voices[args.voice]
-    if voice is None:
-        if args.voice is None:
-            voice=voices['pro']
-        else:
-            voice=args.voice
+    if args.voice in voices:
+        voice=voices[args.voice]
+    else:
+        voice=voices['pro'] if args.voice is None else args.voice
 
     # filename argument
     file_path = args.input
@@ -59,7 +57,7 @@ def main():
 
     file = open(file_path, "r")
     text = file.read()
-    generator = pipeline(text, voice=voices[args.voice])
+    generator = pipeline(text, voice=voice)
 
     output_files = []
     length = 0