Update app.py

app.py

@@ -9,7 +9,7 @@ from typing import Optional
 
 import requests
 import torch
-from fastapi import Body, FastAPI, Header, HTTPException
+from fastapi import BackgroundTasks, Body, FastAPI, Header, HTTPException
 from fastapi.responses import FileResponse, JSONResponse
 from pydantic import BaseModel, Field, HttpUrl
 
@@ -104,9 +104,20 @@ def health(x_api_key: Optional[str] = Header(default=None)):
     return {"status": "ok", "model": "xtts_v2", "device": DEVICE}
 
 
+def _cleanup_files(*files: str):
+    """Background task to clean up temporary files after response is sent."""
+    for file_path in files:
+        if file_path and Path(file_path).exists():
+            try:
+                Path(file_path).unlink(missing_ok=True)
+            except Exception:
+                pass  # Ignore cleanup errors
+
+
 @app.post("/generate")
 def generate(
     payload: GenerateRequest = Body(...),
+    background_tasks: BackgroundTasks = BackgroundTasks(),
     x_api_key: Optional[str] = Header(default=None),
 ):
     _require_api_key(x_api_key)
@@ -126,19 +137,29 @@ def generate(
             split_sentences=True,
         )
 
+        # Verify the output file was created
+        if not Path(output_file).exists():
+            raise RuntimeError(f"TTS generation failed: output file was not created at {output_file}")
+
+        # Schedule cleanup after response is sent
+        background_tasks.add_task(_cleanup_files, speaker_file, output_file)
+
         return FileResponse(output_file, media_type="audio/wav", filename="output.wav")
 
     except HTTPException:
+        # Clean up on HTTPException
+        if speaker_file and Path(speaker_file).exists():
+            Path(speaker_file).unlink(missing_ok=True)
         raise
     except Exception as exc:  # pragma: no cover
-        return JSONResponse(status_code=500, content={"error": str(exc)})
-    finally:
+        # Clean up on error
         if speaker_file and Path(speaker_file).exists():
             Path(speaker_file).unlink(missing_ok=True)
         if output_file and Path(output_file).exists():
             Path(output_file).unlink(missing_ok=True)
+        return JSONResponse(status_code=500, content={"error": str(exc)})
 
 
 @app.get("/")
 def root():
-    return {"name": "xtts-v2-api", "endpoints": ["/health", "/generate"]}
+    return {"name": "xtts-v2-api", "endpoints": ["/health", "/generate"]}