| from fastapi import FastAPI, UploadFile, Form |
| from fastapi.responses import FileResponse |
| import uuid |
| import os |
| import soundfile as sf |
| from TTS.api import TTS |
|
|
| app = FastAPI() |
|
|
| |
| |
| tts = TTS("tts_models/multilingual/multi-dataset/xtts_v2").to("cuda") |
|
|
|
|
| @app.post("/generate") |
| async def generate(text: str = Form(...), speaker_wav: UploadFile = None): |
| output_path = f"/tmp/{uuid.uuid4().hex}.wav" |
|
|
| if speaker_wav: |
| wav_path = f"/tmp/{speaker_wav.filename}" |
| with open(wav_path, "wb") as f: |
| f.write(await speaker_wav.read()) |
| tts.tts_to_file(text=text, speaker_wav=wav_path, file_path=output_path) |
| else: |
| tts.tts_to_file(text=text, speaker="random", file_path=output_path) |
|
|
| return FileResponse(output_path, media_type="audio/wav") |
|
|
|
|
| @app.get("/health") |
| def health(): |
| return {"status": "ok"} |
|
|
