Update app.py

app.py

@@ -4,6 +4,7 @@ import asyncio
 import tempfile
 import os
 from typing import Dict
+from collections import defaultdict
 
 async def text_to_speech(text: str, voice: str) -> str:
     output_file = tempfile.NamedTemporaryFile(delete=False, suffix=".mp3")
@@ -15,18 +16,34 @@ async def list_voices() -> Dict[str, Dict]:
     voices = await edge_tts.list_voices()
     return {v['ShortName']: {'name': v['ShortName'], 'language': v['Locale']} for v in voices}
 
+def process_voices(voices: Dict[str, Dict]) -> Dict[str, Dict[str, str]]:
+    processed_voices = defaultdict(dict)
+    for full_name, details in voices.items():
+        lang, speaker_name = full_name.split('-')[:2]
+        processed_voices[lang][speaker_name] = full_name
+    return dict(processed_voices)
+
 async def main():
     st.title("OpenSpeech-TTS")
     st.write("An API to generate high quality speech with minimal hardware fast and efficiently.")
     st.write("https://github.com/PantelisDeveloping/openspeech-tts/")
 
-    # Get voices
+    # Get voices and process them
     voices = await list_voices()
+    processed_voices = process_voices(voices)
 
     # Text-to-Speech
     st.header("Text-to-Speech")
     text_input = st.text_area("Enter text to convert to speech:")
-    selected_voice = st.selectbox("Select a voice:", list(voices.keys()))
+
+    # Two-step voice selection
+    col1, col2 = st.columns(2)
+    with col1:
+        selected_language = st.selectbox("Select language:", list(processed_voices.keys()))
+    with col2:
+        selected_speaker = st.selectbox("Select speaker:", list(processed_voices[selected_language].keys()))
+
+    selected_voice = processed_voices[selected_language][selected_speaker]
 
     if st.button("Generate Speech"):
         if not text_input:
@@ -39,7 +56,9 @@ async def main():
 
     # List Available Voices
     st.header("Available Voices")
-    st.dataframe(voices)
+    for language, speakers in processed_voices.items():
+        st.subheader(language)
+        st.write(", ".join(speakers.keys()))
 
 if __name__ == '__main__':
     asyncio.run(main())