sij/sijapi
1
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Auto-update: Thu Aug 8 06:28:18 PDT 2024

Browse source
This commit is contained in:
sanj 2024-08-08 06:28:18 -07:00
parent da80656c36
commit 5a03b3af0e
1 changed files with 17 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

30
sijapi/routers/tts.py
View file

@ -147,7 +147,6 @@ async def generate_speech(
if model == "eleven_turbo_v2": if model == "eleven_turbo_v2":
info("Using ElevenLabs.") info("Using ElevenLabs.")
voice = await determine_voice_id(voice)
audio_file_path = await elevenlabs_tts(model, text, voice, title, output_dir) audio_file_path = await elevenlabs_tts(model, text, voice, title, output_dir)
else: # if model == "xtts": else: # if model == "xtts":
info("Using XTTS2") info("Using XTTS2")
@ -233,18 +232,23 @@ async def elevenlabs_tts(model: str, input_text: str, voice: str, title: str = N
"model_id": model "model_id": model
} }
headers = {"Content-Type": "application/json", "xi-api-key": ELEVENLABS_API_KEY} headers = {"Content-Type": "application/json", "xi-api-key": ELEVENLABS_API_KEY}
async with httpx.AsyncClient(timeout=httpx.Timeout(300.0)) as client: # 5 minutes timeout try:
response = await client.post(url, json=payload, headers=headers) async with httpx.AsyncClient(timeout=httpx.Timeout(300.0)) as client: # 5 minutes timeout
output_dir = output_dir if output_dir else TTS_OUTPUT_DIR response = await client.post(url, json=payload, headers=headers)
title = title if title else dt_datetime.now().strftime("%Y%m%d%H%M%S") output_dir = output_dir if output_dir else TTS_OUTPUT_DIR
filename = f"{sanitize_filename(title)}.mp3" title = title if title else dt_datetime.now().strftime("%Y%m%d%H%M%S")
file_path = Path(output_dir) / filename filename = f"{sanitize_filename(title)}.mp3"
if response.status_code == 200: file_path = Path(output_dir) / filename
with open(file_path, "wb") as audio_file: if response.status_code == 200:
audio_file.write(response.content) with open(file_path, "wb") as audio_file:
return file_path audio_file.write(response.content)
else: info(f"file_path: {file_path}")
raise HTTPException(status_code=response.status_code, detail="Error from ElevenLabs API") return file_path
else:
raise HTTPException(status_code=response.status_code, detail="Error from ElevenLabs API")
except Exception as e:
err(f"Error from Elevenlabs API: {e}")
raise HTTPException(status_code=response.status_code, detail="Error from ElevenLabs API")