Update app.py

app.py

@@ -3,6 +3,7 @@ import edge_tts
 import gradio as gr
 import asyncio
 
+# Updated language dictionary with clear voice labeling
 language_dict = {
     "Amharic": {
         "Ameha": "am-ET-AmehaNeural",
@@ -13,18 +14,24 @@ language_dict = {
         "Clara": "en-CA-ClaraNeural"
     },
     "Tigrinya": {
-        "Ameha": "am-ET-AmehaNeural",
-        "Mekdes": "am-ET-MekdesNeural"
+        "Ameha (Amharic Voice)": "am-ET-AmehaNeural",
+        "Mekdes (Amharic Voice)": "am-ET-MekdesNeural"
     }
 }
 
 async def text_to_speech_edge(text, language, speaker):
-    # If Tigrinya is selected, override to use Amharic voices.
+    # Clean speaker name if it's a Tigrinya selection
     if language == "Tigrinya":
-        language = "Amharic"
-    
-    voice = language_dict[language][speaker]
+        speaker = speaker.replace(" (Amharic Voice)", "")
     
+    try:
+        voice = language_dict[language][speaker]
+    except KeyError:
+        error_msg = (f"ስህተት: ድምፅ '{speaker}' ለቋንቋ '{language}' አልተገኘም።"
+                    if language in ["Amharic", "Tigrinya"]
+                    else f"Error: Voice '{speaker}' not available for '{language}'")
+        raise gr.Error(error_msg)
+
     try:
         communicate = edge_tts.Communicate(text, voice)
         with tempfile.NamedTemporaryFile(delete=False, suffix=".mp3") as tmp_file:
@@ -33,25 +40,25 @@ async def text_to_speech_edge(text, language, speaker):
         return tmp_path
         
     except asyncio.TimeoutError:
-        error_msg = ("ስህተት: ጊዜ አልቋል። እባክዎ እንደገና ይሞክሩ። (Timeout)"
-                     if language == "Amharic"
-                     else "Error: Timeout. Please try again.")
+        error_msg = ("ስህተት: ጊዜ አልቋል። እባክዎ እንደገና ይሞክሩ።"
+                    if language in ["Amharic", "Tigrinya"]
+                    else "Error: Timeout. Please try again.")
         raise gr.Error(error_msg)
     except Exception as e:
-        error_msg = (f"ስህተት: ድምፅ መፍጠር አልተቻለም።\nError: {str(e)}"
-                     if language == "Amharic"
-                     else f"Error: Failed to generate audio.\nDetails: {str(e)}")
+        error_msg = (f"ስህተት: {str(e)}"
+                    if language in ["Amharic", "Tigrinya"]
+                    else f"Error: {str(e)}")
         raise gr.Error(error_msg)
 
 def update_speakers(language):
-    # For Tigrinya, return Amharic speakers.
-    if language == "Tigrinya":
-         speakers = list(language_dict["Amharic"].keys())
-    else:
-         speakers = list(language_dict[language].keys())
-    return gr.update(choices=speakers, value=speakers[0])
+    speakers = list(language_dict[language].keys())
+    return gr.Dropdown(
+        choices=speakers,
+        value=speakers[0],
+        label=f"Select Speaker {'(Amharic Voices)' if language == 'Tigrinya' else ''}"
+    )
 
-with gr.Blocks(title="Amharic, English & Tigrinya TTS") as demo:
+with gr.Blocks(title="Amharic, English & Tigrinya TTS", theme=gr.themes.Soft()) as demo:
     gr.HTML("""
     <style>
         h1 { 
@@ -60,16 +67,26 @@ with gr.Blocks(title="Amharic, English & Tigrinya TTS") as demo:
             background: linear-gradient(45deg, #FF007F, #2E86C1);
             -webkit-background-clip: text;
             -webkit-text-fill-color: transparent;
+            margin-bottom: 20px;
         }
-        .gradio-button { 
-            background: linear-gradient(45deg, #FF007F, #2E86C1) !important; 
-            color: white !important; 
+        .notice {
+            font-size: 0.9em;
+            color: #666;
+            text-align: center;
+            margin: 10px 0;
+            font-style: italic;
         }
-        .gradio-textbox, .gradio-dropdown { 
-            border-color: #2E86C1 !important; 
+        .gradio-button {
+            background: linear-gradient(45deg, #FF007F, #2E86C1) !important;
+            color: white !important;
         }
     </style>
-    <center><h1>Amharic, English & Tigrinya Text-to-Speech</h1></center>
+    <center>
+        <h1>Amharic, English & Tigrinya Text-to-Speech</h1>
+        <div class="notice">
+            Note: Tigrinya uses Amharic-accented voices until dedicated models become available
+        </div>
+    </center>
     """)
 
     with gr.Row():
@@ -80,25 +97,33 @@ with gr.Blocks(title="Amharic, English & Tigrinya TTS") as demo:
                 label="Select Language / ቋንቋ ይምረጡ"
             )
             input_text = gr.Textbox(
-                lines=5, 
+                lines=5,
                 label="Enter Text / ጽሑፍ ያስገቡ",
                 placeholder="Type your text here... / ጽሑፍዎን ይጻፉ..."
             )
             speaker = gr.Dropdown(
-                # For Tigrinya, this will be updated to Amharic speakers.
-                choices=["Ameha", "Mekdes"],
-                value="Ameha",
-                label="Select Speaker / አርቲስት ይምረጡ"
+                label="Select Speaker / አርቲስት ይምረጡ",
+                interactive=True
+            )
+            run_btn = gr.Button(
+                value="Generate Audio / ድምፅ ፍጠር",
+                variant="primary"
             )
-            run_btn = gr.Button(value="Generate Audio / ድምፅ ፍጠር", variant="primary")
 
         with gr.Column():
             output_audio = gr.Audio(
                 type="filepath",
-                label="Generated Audio / የተፈጠረ ድምፅ"
+                label="Generated Audio / የተፈጠረ ድምፅ",
+                interactive=False
             )
 
-    # Update the speaker dropdown when language changes.
+    # Initialize speakers dropdown
+    demo.load(
+        fn=lambda: gr.Dropdown(choices=list(language_dict["Amharic"].keys()),
+        outputs=speaker
+    )
+
+    # Update speakers when language changes
     language.change(
         update_speakers,
         inputs=language,
@@ -112,4 +137,4 @@ with gr.Blocks(title="Amharic, English & Tigrinya TTS") as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch(server_port=7860, share=False)
+    demo.launch(server_port=7860, share=False)