From 0ef28afa227e78afceff4d4568375a89ca0eb7bf Mon Sep 17 00:00:00 2001
From: ksieuk <iebebd3@gmail.com>
Date: Tue, 10 Oct 2023 18:34:05 +0300
Subject: [PATCH] feat: [#28] add base service

---
 src/assistant/lib/tts/services.py | 35 +++++++++++++++++++++++++------
 1 file changed, 29 insertions(+), 6 deletions(-)

diff --git a/src/assistant/lib/tts/services.py b/src/assistant/lib/tts/services.py
index 904157a..df8a00b 100644
--- a/src/assistant/lib/tts/services.py
+++ b/src/assistant/lib/tts/services.py
@@ -1,16 +1,39 @@
-import abc
-
 import lib.app.settings as app_settings
+import lib.models as models
 import lib.tts.models as tts_models
 
 
-class BaseTTSService(abc.ABC):
+class TTSService:
     def __init__(
         self,
         settings: app_settings.Settings,
+        tts_repositories: list[tts_models.TTSRepositoryProtocol],
     ):
         self.settings = settings
+        self.tts_repositories = tts_repositories
 
-    @abc.abstractmethod
-    def get_tts(self, tts_request: tts_models.TTSRequestModel) -> tts_models.TTSResponseModel:
-        raise NotImplementedError()
+    def get_audio_as_bytes_from_text(self, tts_request: models.TTSCreateRequestModel) -> models.TTSCreateResponseModel:
+        for repository in self.tts_repositories:
+            voice_model = repository.get_voice_model_by_name(tts_request.voice_model_name)
+            if voice_model:
+                audio_response = repository.get_audio_as_bytes_from_text(tts_request.text)
+                break
+        else:
+            raise ValueError(f"Voice model {tts_request.voice_model_name} not found")
+        return audio_response
+
+    def get_voice_model_by_name(self, voice_model_name: str) -> models.BaseVoiceModel | None:
+        for repository in self.tts_repositories:
+            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[tts_models.VOICE_MODELS_TYPE]:
+        response_models: list[tts_models.VOICE_MODELS_TYPE] = []
+        for repository in self.tts_repositories:
+            voice_models = repository.get_voice_models_by_fields(fields)
+            if voice_models.models:
+                response_models.extend(voice_models.models)
+        return response_models