up32

app.py

@@ -4,6 +4,7 @@ import outetts
 import io
 import json
 import base64
+import os 
 # Initialize the interface
 interface = outetts.Interface(
     config=outetts.ModelConfig.auto_config(
@@ -47,22 +48,29 @@ async def websocket_tts(websocket: WebSocket):
             # Status: Generating linguistic features
             await websocket.send_text(json.dumps({"generation_status": "Generating linguistic features"}))
             # Save to buffer
-            audio_buffer = io.BytesIO()
-            output.save(audio_buffer)
-            audio_buffer.seek(0)
+            import uuid
+            temp_path = f"temp_{uuid.uuid4().hex}.wav"
+            output.save(temp_path)
             chunk_size = 4096
-            while True:
-                chunk = audio_buffer.read(chunk_size)
-                if not chunk:
-                    break
-                audio_b64 = base64.b64encode(chunk).decode("ascii")
-                await websocket.send_text(json.dumps({
-                    "data": {
-                        "audio_bytes": audio_b64,
-                        "duration": None,
-                        "request_finished": False
-                    }
-                }))
+            try:
+                with open(temp_path, "rb") as f:
+                    while True:
+                        chunk = f.read(chunk_size)
+                        if not chunk:
+                            break
+                        audio_b64 = base64.b64encode(chunk).decode("ascii")
+                        await websocket.send_text(json.dumps({
+                            "data": {
+                                "audio_bytes": audio_b64,
+                                "duration": None,
+                                "request_finished": False
+                            }
+                        }))
+            finally:
+                try:
+                    os.remove(temp_path)
+                except FileNotFoundError:
+                    pass
             # Final event
             await websocket.send_text(json.dumps({
                 "data": {