|
|
from fastapi import FastAPI, HTTPException |
|
|
from fastapi.responses import FileResponse |
|
|
from pydantic import BaseModel |
|
|
from talk import say |
|
|
|
|
|
app = FastAPI() |
|
|
|
|
|
|
|
|
class TTSRequest(BaseModel): |
|
|
text: str |
|
|
voice: str = "af_heart" |
|
|
|
|
|
|
|
|
@app.get("/") |
|
|
def root(): |
|
|
return { |
|
|
"message": "Kokoro TTS API", |
|
|
"endpoints": { |
|
|
"/tts": "POST - Generate speech from text", |
|
|
"/voices": "GET - List available voices" |
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|
@app.post("/tts") |
|
|
def generate_speech(request: TTSRequest): |
|
|
"""Generate speech from text using specified voice.""" |
|
|
if not request.text or not request.text.strip(): |
|
|
raise HTTPException(status_code=400, detail="Text cannot be empty") |
|
|
|
|
|
audio_file = say(request.text, voice=request.voice, play=False) |
|
|
|
|
|
if audio_file: |
|
|
return FileResponse( |
|
|
audio_file, |
|
|
media_type="audio/wav", |
|
|
filename=f"speech_{request.voice}.wav" |
|
|
) |
|
|
|
|
|
raise HTTPException(status_code=500, detail="Failed to generate audio") |
|
|
|
|
|
|
|
|
@app.get("/voices") |
|
|
def list_voices(): |
|
|
"""List available voices.""" |
|
|
return { |
|
|
"voices": [ |
|
|
{"id": "af_heart", "name": "Female Heart", "language": "English"}, |
|
|
{"id": "af_sky", "name": "Female Sky", "language": "English"}, |
|
|
{"id": "af_bella", "name": "Female Bella", "language": "English"}, |
|
|
{"id": "af_nicole", "name": "Female Nicole", "language": "English"}, |
|
|
{"id": "af_sarah", "name": "Female Sarah", "language": "English"}, |
|
|
{"id": "am_adam", "name": "Male Adam", "language": "English"}, |
|
|
{"id": "am_michael", "name": "Male Michael", "language": "English"}, |
|
|
{"id": "bf_emma", "name": "British Female Emma", "language": "English (British)"}, |
|
|
{"id": "bf_isabella", "name": "British Female Isabella", "language": "English (British)"}, |
|
|
{"id": "bm_george", "name": "British Male George", "language": "English (British)"}, |
|
|
{"id": "bm_lewis", "name": "British Male Lewis", "language": "English (British)"} |
|
|
] |
|
|
} |
|
|
|
