Update app.py

app.py

@@ -1,32 +1,39 @@
 import gradio as gr
 import os
-import sys
+import traceback
 from Zonos_main.tts import ZonosTTS  # Adjust import path as needed
 
-# Initialize TTS model
-tts = ZonosTTS()
+# Initialize TTS with error handling
+try:
+    tts = ZonosTTS()
+    print("✅ TTS model loaded successfully")
+except Exception as e:
+    print(f"❌ Model loading failed: {str(e)}")
+    raise
 
 def generate_audio(text):
     try:
-        # Generate audio file path
-        output_path = "output.wav"
-        
-        # Run TTS
+        if not text or len(text.strip()) == 0:
+            raise ValueError("Input text cannot be empty")
+            
+        output_path = "/tmp/output.wav"
         tts.generate(text, output_path)
         
-        # Return audio file
+        if not os.path.exists(output_path):
+            raise FileNotFoundError("Audio file was not generated")
+            
         return output_path
     except Exception as e:
+        traceback.print_exc()
         raise gr.Error(f"Audio generation failed: {str(e)}")
 
-# Create Gradio interface
+# Gradio interface
 demo = gr.Interface(
     fn=generate_audio,
-    inputs=gr.Textbox(label="Input Text"),
+    inputs=gr.Textbox(label="Input Text", placeholder="Enter text here..."),
     outputs=gr.Audio(label="Generated Speech"),
-    title="Zonos TTS Service",
-    description="Convert text to speech using Zonos TTS model"
+    title="Zonos TTS",
+    examples=[["Hello world"], ["This is a test"]]
 )
 
-# Launch the app
-demo.launch()
+demo.launch(show_error=True)