Update app.py

app.py

@@ -1,39 +1,57 @@
 import gradio as gr
 import pyttsx3
 import os
+import warnings
 
-# Initialize the TTS engine
-engine = pyttsx3.init()
+# Suppress pygame welcome message
+with warnings.catch_warnings():
+    warnings.simplefilter("ignore")
+    import pygame
 
-# Get available voices
-voices = engine.getProperty('voices')
+def get_voices():
+    """Initialize engine and get available voices with error handling"""
+    try:
+        engine = pyttsx3.init()
+        voices = engine.getProperty('voices')
+        return {f"{voice.name} ({voice.languages[0] if voice.languages else 'Unknown'})": voice.id 
+               for voice in voices}
+    except Exception as e:
+        print(f"Error initializing TTS engine: {e}")
+        # Return a default voice if engine fails
+        return {"Default Voice": None}
 
-# Create a dictionary to map voice names to IDs
-voice_map = {f"{voice.name} ({voice.languages[0] if voice.languages else 'Unknown'})": voice.id for voice in voices}
+# Get available voices
+voice_map = get_voices()
 
 def text_to_speech(text, voice_name, rate=200):
     """
     Convert text to speech with selected voice and rate.
     Saves output as an audio file and returns the file path.
     """
-    # Initialize engine for each call to avoid threading issues with Gradio
-    engine = pyttsx3.init()
-    
-    # Set voice
-    voice_id = voice_map.get(voice_name)
-    if not voice_id:
-        return "Error: Selected voice not found."
-    engine.setProperty('voice', voice_id)
-    
-    # Set speech rate
-    engine.setProperty('rate', rate)
-    
-    # Save audio to a file
-    output_file = "output.wav"
-    engine.save_to_file(text, output_file)
-    engine.runAndWait()
-    
-    return output_file
+    try:
+        # Initialize engine for each call
+        engine = pyttsx3.init()
+        
+        # Set voice if available
+        voice_id = voice_map.get(voice_name)
+        if voice_id:
+            engine.setProperty('voice', voice_id)
+        
+        # Set speech rate
+        engine.setProperty('rate', rate)
+        
+        # Save audio to a file
+        output_file = "output.wav"
+        engine.save_to_file(text, output_file)
+        engine.runAndWait()
+        
+        # Ensure file exists before returning
+        if os.path.exists(output_file):
+            return output_file
+        return None
+    except Exception as e:
+        print(f"Error in text_to_speech: {e}")
+        return None
 
 # Gradio interface
 with gr.Blocks(title="Text-to-Speech with Different Voices") as demo:
@@ -41,7 +59,7 @@ with gr.Blocks(title="Text-to-Speech with Different Voices") as demo:
     gr.Markdown("Enter text and select a voice to convert it to speech with different voices and accents.")
     
     text_input = gr.Textbox(label="Enter Text", placeholder="Type your text here...")
-    voice_dropdown = gr.Dropdown(choices=list(voice_map.keys()), label="Select Voice/Accent")
+    voice_dropdown = gr.Dropdown(choices=list(voice_map.keys()), label="Select Voice/Accent", value=list(voice_map.keys())[0] if voice_map else None)
     rate_slider = gr.Slider(minimum=100, maximum=300, value=200, step=10, label="Speech Rate")
     
     convert_button = gr.Button("Convert to Speech")
@@ -53,8 +71,5 @@ with gr.Blocks(title="Text-to-Speech with Different Voices") as demo:
         outputs=audio_output
     )
 
-# Launch the app (commented out for Hugging Face Spaces deployment)
-# demo.launch()
-
 if __name__ == "__main__":
     demo.launch()