Update app.py

app.py

@@ -906,19 +906,58 @@ def create_voice_mix_ui():
 
 demo4 = create_voice_mix_ui()
 
+# display_text = "  \n".join(voice_list)
+
+# with gr.Blocks() as demo5:
+#     gr.Markdown("[Install on Windows/Linux](https://github.com/NeuralFalconYT/Kokoro-82M-WebUI)")
+#     gr.Markdown(f"# Voice Names \n{display_text}")
+    
+
 
 
 
 
+import os
+import json
+
+def get_voice_names():
+    male_voices, female_voices, other_voices = [], [], []
+    
+    for filename in os.listdir("./KOKORO/voices"):
+        if filename.endswith('.pt'):
+            name = os.path.splitext(filename)[0]
+            if "m_" in name:
+                male_voices.append(name)
+            elif name=="af":
+                female_voices.append(name)
+            elif "f_" in name:
+                female_voices.append(name)
+            else:
+                other_voices.append(name)
+    
+    # Sort the lists by the length of the voice names
+    male_voices = sorted(male_voices, key=len)
+    female_voices = sorted(female_voices, key=len)
+    other_voices = sorted(other_voices, key=len)
 
+    return json.dumps({
+        "male_voices": male_voices,
+        "female_voices": female_voices,
+        "other_voices": other_voices
+    }, indent=4)
 
 
-display_text = "  \n".join(voice_list)
+# display_text = "  \n".join(voice_list)
 
 with gr.Blocks() as demo5:
+    gr.Markdown(f"# Voice Names")
     gr.Markdown("[Install on Windows/Linux](https://github.com/NeuralFalconYT/Kokoro-82M-WebUI)")
-    gr.Markdown(f"# Voice Names \n{display_text}")
-    
+    get_voice_button = gr.Button("Get Voice Names")
+    voice_names = gr.Textbox(label="Voice Names", placeholder="Click 'Get Voice Names' to display the list of available voice names", lines=10)
+    get_voice_button.click(get_voice_names, outputs=[voice_names])
+
+
+