Update app.py

app.py

@@ -1,24 +1,58 @@
+import os
 import gradio as gr
 import base64
 from Zonos_main.tts import ZonosTTS
 
-# Initialize model ONCE when space starts
-tts = ZonosTTS()  # <-- MUST WORK WITHOUT ERRORS
+# Force CPU mode to avoid CUDA issues
+os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
+
+def load_model():
+    try:
+        tts = ZonosTTS()
+        print("✅ TTS model loaded successfully")
+        return tts
+    except Exception as e:
+        print(f"❌ Model loading failed: {str(e)}")
+        raise RuntimeError("Could not initialize TTS model") from e
+
+tts = load_model()
 
 def generate_audio(text):
     try:
-        # Generate raw audio bytes (MODIFY THIS TO MATCH YOUR MODEL)
-        audio_bytes = tts.generate(text)  # Should return bytes directly
+        if not text.strip():
+            raise ValueError("Input text cannot be empty")
+        
+        # Generate audio (must return raw bytes)
+        audio_bytes = tts.generate(text)
         
-        # Convert to base64
-        return f"data:audio/wav;base64,{base64.b64encode(audio_bytes).decode('utf-8')}"
+        if not isinstance(audio_bytes, bytes):
+            raise TypeError("Model must return bytes")
+            
+        if len(audio_bytes) < 100:  # Minimum valid audio size
+            raise ValueError("Generated audio too small")
+            
+        return {
+            "name": "output.wav",
+            "data": f"data:audio/wav;base64,{base64.b64encode(audio_bytes).decode()}",
+            "size": len(audio_bytes)
+        }
         
     except Exception as e:
-        raise gr.Error(f"TTS Failed: {str(e)}")
+        print(f"❌ Generation error: {str(e)}")
+        raise gr.Error(f"Audio generation failed: {str(e)}") from e
 
-# Minimal interface
-gr.Interface(
+# Robust Gradio interface
+demo = gr.Interface(
     fn=generate_audio,
-    inputs=gr.Textbox(label="Input"),
-    outputs=gr.Audio(type="filepath"),
-).launch()
+    inputs=gr.Textbox(label="Input Text", placeholder="Type something..."),
+    outputs=gr.Audio(label="Output", type="filepath"),
+    title="Zonos TTS",
+    allow_flagging="never"
+)
+
+# Launch with error handling
+try:
+    demo.launch(show_error=True)
+except Exception as e:
+    print(f"❌ Gradio launch failed: {str(e)}")
+    raise