Update app.py

app.py

@@ -1,7 +1,8 @@
 import gradio as gr
 import os
 import traceback
-from Zonos_main.tts import ZonosTTS  # Adjust import path as needed
+import base64  # Add this
+from Zonos_main.tts import ZonosTTS
 
 # Initialize TTS with error handling
 try:
@@ -21,8 +22,12 @@ def generate_audio(text):
         
         if not os.path.exists(output_path):
             raise FileNotFoundError("Audio file was not generated")
+        
+        # Return audio as base64 string
+        with open(output_path, "rb") as audio_file:
+            audio_base64 = base64.b64encode(audio_file.read()).decode('utf-8')
+            return f"data:audio/wav;base64,{audio_base64}"
             
-        return output_path
     except Exception as e:
         traceback.print_exc()
         raise gr.Error(f"Audio generation failed: {str(e)}")