File size: 2,860 Bytes
a4b70d9 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 |
from __future__ import annotations
import os
try:
from gtts import gTTS as gTTS_Service
has_gtts = True
except ImportError:
has_gtts = False
from ...typing import AsyncResult, Messages
from ...providers.response import AudioResponse
from ...image.copy_images import get_filename, get_media_dir, ensure_media_dir
from ..base_provider import AsyncGeneratorProvider, ProviderModelMixin
from ..helper import get_last_message
locals = {
"en-AU": ["English (Australia)", "en", "com.au"],
"en-GB": ["English (United Kingdom)", "en", "co.uk"],
"en-US": ["English (United States)", "en", "us"],
"en-CA": ["English (Canada)", "en", "ca"],
"en-IN": ["English (India)", "en", "co.in"],
"en-IE": ["English (Ireland)", "en", "ie"],
"en-ZA": ["English (South Africa)", "en", "co.za"],
"en-NG": ["English (Nigeria)", "en", "com.ng"],
"fr-CA": ["French (Canada)", "fr", "ca"],
"fr-FR": ["French (France)", "fr", "fr"],
"de-DE": ["German (Germany)", "de", "de"],
"zh-CN": ["Mandarin (China Mainland)", "zh-CN", "com"],
"zh-TW": ["Mandarin (Taiwan)", "zh-TW", "com"],
"pt-BR": ["Portuguese (Brazil)", "pt", "com.br"],
"pt-PT": ["Portuguese (Portugal)", "pt", "pt"],
"es-MX": ["Spanish (Mexico)", "es", "com.mx"],
"es-ES": ["Spanish (Spain)", "es", "es"],
"es-US": ["Spanish (United States)", "es", "us"],
}
models = {locale[0]: {"lang": locale[1], "tld": locale[2]} for locale in locals.values()}
class gTTS(AsyncGeneratorProvider, ProviderModelMixin):
label = "gTTS (Google Text-to-Speech)"
working = has_gtts
model_id = "google-tts"
default_language = "en"
default_tld = "com"
default_format = "mp3"
default_model = "en-US"
models = list(models.keys())
audio_models = models
@classmethod
async def create_async_generator(
cls,
model: str,
messages: Messages,
prompt: str = None,
audio: dict = {},
**kwargs
) -> AsyncResult:
prompt = get_last_message(messages, prompt)
if not prompt:
raise ValueError("Prompt is empty.")
format = audio.get("format", cls.default_format)
filename = get_filename([cls.model_id], prompt, f".{format}", prompt)
target_path = os.path.join(get_media_dir(), filename)
ensure_media_dir()
if "language" in audio:
model = locals[audio["language"]][0] if audio["language"] in locals else model
gTTS_Service(
prompt,
**{
"lang": audio.get("language", cls.default_language),
"tld": audio.get("tld", cls.default_tld),
"slow": audio.get("slow", False),
**models.get(model, {})
}
).save(target_path)
yield AudioResponse(f"/media/{filename}", audio=audio, text=prompt) |