From 4ddc33e75bb20504d93a0103dd55af1a9082a1b0 Mon Sep 17 00:00:00 2001
From: superobk <superobk@hotmail.com>
Date: Mon, 29 Apr 2024 09:31:37 +0800
Subject: [PATCH] fix: inject fail

---
 main.py                          | 5 +++--
 src/blackbox/audio_chat.py       | 6 +++---
 src/blackbox/blackbox_factory.py | 2 +-
 3 files changed, 7 insertions(+), 6 deletions(-)

diff --git a/main.py b/main.py
index 2cb5207..10ff932 100644
--- a/main.py
+++ b/main.py
@@ -9,6 +9,7 @@ from src.dotchain.runtime.runtime import Runtime
 from src.blackbox.blackbox_factory import BlackboxFactory
 import uvicorn
 from fastapi.middleware.cors import CORSMiddleware
+from injector import Injector
 
 app = FastAPI()
 
@@ -20,8 +21,8 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-
-blackbox_factory = BlackboxFactory()
+injector = Injector()
+blackbox_factory = injector.get(BlackboxFactory)
 
 @app.post("/")
 async def blackbox(blackbox_name: Union[str, None] = None, request: Request = None):
diff --git a/src/blackbox/audio_chat.py b/src/blackbox/audio_chat.py
index 1131c22..091c78e 100644
--- a/src/blackbox/audio_chat.py
+++ b/src/blackbox/audio_chat.py
@@ -2,9 +2,9 @@ from fastapi import Request, Response,status
 from fastapi.responses import JSONResponse
 from injector import inject
 
-from blackbox.asr import ASR
-from blackbox.tesou import Tesou
-from blackbox.tts import TTS
+from .asr import ASR
+from .tesou import Tesou
+from .tts import TTS
 
 from .blackbox import Blackbox
 
diff --git a/src/blackbox/blackbox_factory.py b/src/blackbox/blackbox_factory.py
index d6ac563..79378d1 100644
--- a/src/blackbox/blackbox_factory.py
+++ b/src/blackbox/blackbox_factory.py
@@ -9,7 +9,7 @@ from .tesou import Tesou
 from .fastchat import Fastchat
 from .g2e import G2E
 from .text_and_image import TextAndImage
-from injector import Injector, inject
+from injector import inject
 
 class BlackboxFactory:
     models = {}