jsdio/voice_assistant
Template
1
0
Fork 0
mirror of https://github.com/ijaric/voice_assistant.git synced 2025-05-24 22:43:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: [#47] add tts

Browse Source
This commit is contained in:
ksieuk 2023-10-13 17:04:36 +03:00
parent 1170532c93
commit cd96624135
1 changed files with 31 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
src/assistant/lib/app/app.py
View File

@ -11,7 +11,9 @@ import lib.app.errors as app_errors
import lib.app.settings as app_settings import lib.app.settings as app_settings
import lib.app.split_settings as app_split_settings import lib.app.split_settings as app_split_settings
import lib.clients as clients import lib.clients as clients
import lib.models as models
import lib.stt as stt import lib.stt as stt
import lib.tts as tts
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
@ -59,21 +61,43 @@ class Application:
logger.info("Initializing clients") logger.info("Initializing clients")
http_yandex_tts_client = clients.AsyncHttpClient( http_yandex_tts_client = clients.AsyncHttpClient(
base_url="yandex", # todo add yandex api url from settings
proxy_settings=settings.proxy, proxy_settings=settings.proxy,
base_url="https://tts.api.cloud.yandex.net/speech/v1/",
headers=settings.tts_yandex.base_headers,
) )
http_eleven_labs_tts_client = clients.AsyncHttpClient(
base_url="https://api.elevenlabs.io/v1/",
headers=settings.tts_eleven_labs.base_headers,
)
disposable_resources.append( disposable_resources.append(
DisposableResource( DisposableResource(
name="http_client yandex", name="http_client yandex",
dispose_callback=http_yandex_tts_client.close(), dispose_callback=http_yandex_tts_client.close(),
) )
) )
disposable_resources.append(
DisposableResource(
name="http_client eleven labs",
dispose_callback=http_eleven_labs_tts_client.close(),
)
)
# Repositories # Repositories
logger.info("Initializing repositories") logger.info("Initializing repositories")
stt_repository: stt.STTProtocol = stt.OpenaiSpeechRepository(settings=settings) stt_repository: stt.STTProtocol = stt.OpenaiSpeechRepository(settings=settings)
tts_yandex_repository = tts.TTSYandexRepository(
tts_settings=app_split_settings.TTSYandexSettings(),
client=http_yandex_tts_client,
)
tts_eleven_labs_repository = tts.TTSElevenLabsRepository(
tts_settings=app_split_settings.TTSElevenLabsSettings(),
client=http_eleven_labs_tts_client,
is_models_from_api=True,
)
# Caches # Caches
logger.info("Initializing caches") logger.info("Initializing caches")
@ -83,6 +107,12 @@ class Application:
logger.info("Initializing services") logger.info("Initializing services")
stt_service: stt.SpeechService = stt.SpeechService(repository=stt_repository) # type: ignore stt_service: stt.SpeechService = stt.SpeechService(repository=stt_repository) # type: ignore
tts_service: tts.TTSService = tts.TTSService( # type: ignore
repositories={
models.VoiceModelProvidersEnum.YANDEX: tts_yandex_repository,
models.VoiceModelProvidersEnum.ELEVEN_LABS: tts_eleven_labs_repository,
},
)
# Handlers # Handlers
logger.info("Initializing handlers") logger.info("Initializing handlers")