mirror of
https://github.com/ijaric/voice_assistant.git
synced 2025-05-24 14:33:26 +00:00
36 lines
1.4 KiB
Python
36 lines
1.4 KiB
Python
import lib.app.settings as app_settings
|
|
import lib.models as models
|
|
import lib.tts.models as tts_models
|
|
|
|
|
|
class TTSService:
|
|
def __init__(
|
|
self,
|
|
settings: app_settings.Settings,
|
|
repositories: dict[models.VoiceModelProvidersEnum, tts_models.TTSRepositoryProtocol],
|
|
):
|
|
self.settings = settings
|
|
self.repositories = repositories
|
|
|
|
def get_audio_as_bytes(self, request: models.TTSCreateRequestModel) -> models.TTSCreateResponseModel:
|
|
model = request.voice_model
|
|
repository = self.repositories[model.provider]
|
|
audio_response = repository.get_audio_as_bytes(request)
|
|
return audio_response
|
|
|
|
def get_voice_model_by_name(self, voice_model_name: str) -> models.BaseVoiceModel | None:
|
|
for repository in self.repositories.values():
|
|
voice_model = repository.get_voice_model_by_name(voice_model_name)
|
|
if voice_model:
|
|
return voice_model
|
|
|
|
def get_list_voice_models_by_fields(
|
|
self, fields: models.TTSSearchVoiceRequestModel
|
|
) -> list[models.AVAILABLE_MODELS_TYPE]:
|
|
response_models: list[models.AVAILABLE_MODELS_TYPE] = []
|
|
for repository in self.repositories.values():
|
|
voice_models = repository.get_voice_models_by_fields(fields)
|
|
if voice_models.models:
|
|
response_models.extend(voice_models.models)
|
|
return response_models
|