ivanwu/jarvis-models
1
0
Fork 0
mirror of https://github.com/BoardWare-Genius/jarvis-models.git synced 2025-12-14 00:53:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: blackbox call

Browse Source
This commit is contained in:
Dan Chen
2024-03-21 15:59:35 +08:00
parent 726b926d98
commit 11e1659e22
12 changed files with 49 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
src/blackbox/blackbox_factory.py
View File

@ -14,14 +14,21 @@ class BlackboxFactory:
self.tts = TTS()
self.asr = ASR("./.env.yaml")
self.sentiment = Sentiment()
self.sum = SUM()
self.calculator = Calculator()
self.audio_to_text = AudioToText()
self.text_to_audio = TextToAudio()
def create_blackbox(self, blackbox_name: str, blackbox_config: dict) -> Blackbox:
def __call__(self, *args, **kwargs):
return self.processing(*args, **kwargs)
def create_blackbox(self, blackbox_name: str) -> Blackbox:
if blackbox_name == "audio_to_text":
return AudioToText(blackbox_config)
return self.audio_to_text
if blackbox_name == "text_to_audio":
return TextToAudio(blackbox_config)
return self.text_to_audio
if blackbox_name == "calculator":
return Calculator(blackbox_config)
return self.calculator
if blackbox_name == "asr":
return self.asr
if blackbox_name == "tts":
@ -29,5 +36,5 @@ class BlackboxFactory:
if blackbox_name == "sentiment_engine":
return self.sentiment
if blackbox_name == "sum":
return SUM()
return self.sum
raise ValueError("Invalid blockbox type")