return audio

app.py

@@ -1,5 +1,4 @@
 from fastapi import FastAPI
-from pydantic import BaseModel
 from gtts import gTTS
 import aiofiles
 import uvicorn
@@ -8,16 +7,7 @@ import time
 
 app = FastAPI()
 
-# Pydantic model for JSON body
-class TTSRequest(BaseModel):
-    text: str
-    lang: str = "id"
-    slow: bool = False
-
 def generate_timestamps(script: str, wpm: int = 150):
-    """
-    Generate naive timestamps per sentence based on words per minute (WPM).
-    """
     sentences = [s.strip() for s in script.replace("\n", " ").split(".") if s.strip()]
     timestamps = []
     current_time = 0.0
@@ -42,25 +32,38 @@ async def root():
     return {"message": "API is running!"}
 
 @app.post("/tts")
-async def text_to_speech(request: TTSRequest):
-    # Save TTS audio
+async def text_to_speech(payload: dict):
+    text = payload.get("text", "")
+    lang = payload.get("lang", "en")
+    slow = payload.get("slow", False)
+
+    if not text:
+        return {"error": "Text is required"}
+
+    # Save audio file
     file_path = "output.mp3"
-    tts = gTTS(request.text, lang=request.lang, slow=request.slow)
+    tts = gTTS(text=text, lang=lang, slow=slow)
     tts.save(file_path)
 
-    # Read audio file
+    # Read audio for size
     async with aiofiles.open(file_path, mode="rb") as f:
         audio_data = await f.read()
 
-    # Generate timestamps
-    timestamps = generate_timestamps(request.text)
-
+    # Build JSON response with download link
     return {
-        "script": request.text,
-        "timestamps": timestamps,
-        "file": file_path,
+        "script": text,
+        "timestamps": generate_timestamps(text),
+        "file_url": f"/download/{os.path.basename(file_path)}",
         "size": len(audio_data)
     }
 
+@app.get("/download/{filename}")
+async def download_file(filename: str):
+    file_path = os.path.join(os.getcwd(), filename)
+    if os.path.exists(file_path):
+        from fastapi.responses import FileResponse
+        return FileResponse(file_path, media_type="audio/mpeg", filename=filename)
+    return {"error": "File not found"}
+
 if __name__ == "__main__":
     uvicorn.run("app:app", host="0.0.0.0", port=7860)