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

feat: configuration

Browse Source
This commit is contained in:
Dan Chen
2024-04-26 17:46:49 +08:00
committed by superobk
parent d493bd221f
commit ee17313dbc
1 changed files with 26 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

54
src/blackbox/blackbox_factory.py
View File

@ -12,42 +12,36 @@ from .text_and_image import TextAndImage
from injector import Injector
class BlackboxFactory:
models = {}
def __init__(self) -> None:
injector = Injector()
self.tts = TTS()
self.asr = ASR(".env.yaml")
self.sentiment = Sentiment()
self.audio_to_text = AudioToText()
self.text_to_audio = TextToAudio()
self.tesou = injector.get(Tesou)
self.fastchat = Fastchat()
self.audio_chat = AudioChat(self.asr, self.tesou, self.tts)
self.g2e = G2E()
self.text_and_image = TextAndImage()
self.models["audio_to_text"] = AudioToText()
self.models["text_to_audio"] = TextToAudio()
self.models["asr"] = ASR(".env.yaml")
self.models["tts"] = TTS()
self.models["sentiment_engine"] = Sentiment()
self.models["tesou"] = injector.get(Tesou)
self.models["fastchat"] = Fastchat()
self.models["audio_chat"] = AudioChat(self.models["asr"], self.models["tesou"], self.models["tts"])
self.models["g2e"] = G2E()
self.models["text_and_image"] = TextAndImage()
# self.tts = TTS()
# self.asr = ASR(".env.yaml")
# self.sentiment = Sentiment()
# self.audio_to_text = AudioToText()
# self.text_to_audio = TextToAudio()
# self.tesou = injector.get(Tesou)
# self.fastchat = Fastchat()
# self.audio_chat = AudioChat(self.asr, self.tesou, self.tts)
# self.g2e = G2E()
# self.text_and_image = TextAndImage()
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 self.audio_to_text
if blackbox_name == "text_to_audio":
return self.text_to_audio
if blackbox_name == "asr":
return self.asr
if blackbox_name == "tts":
return self.tts
if blackbox_name == "sentiment_engine":
return self.sentiment
if blackbox_name == "tesou":
return self.tesou
if blackbox_name == "fastchat":
return self.fastchat
if blackbox_name == "audio_chat":
return self.audio_chat
if blackbox_name == "g2e":
return self.g2e
if blackbox_name == 'text_and_image':
return self.text_and_image
model = self.models.get(blackbox_name)
if model is None:
raise ValueError("Invalid blockbox type")
return model