This commit is contained in:
Dan Chen
2024-04-30 16:21:17 +08:00
committed by superobk
parent 032f6cb35a
commit 884a835cae
4 changed files with 30 additions and 8 deletions

View File

@ -1,13 +1,9 @@
from typing import Annotated, Union
from typing import Union
from fastapi import FastAPI, Request, status, Form
from fastapi import FastAPI, Request, status
from fastapi.responses import JSONResponse
from src.dotchain.runtime.interpreter import program_parser
from src.dotchain.runtime.tokenizer import Tokenizer
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

View File

@ -4,15 +4,19 @@ from typing import Any, Coroutine
from fastapi import Request, Response, status
from fastapi.responses import JSONResponse
from ..log.log import Logger
from ..asr.rapid_paraformer.utils import read_yaml
from ..asr.rapid_paraformer import RapidParaformer
from .blackbox import Blackbox
from injector import singleton
from injector import singleton, inject
@singleton
class ASR(Blackbox):
def __init__(self, path = ".env.yaml") -> None:
@inject
def __init__(self,logger: Logger,path = ".env.yaml") -> None:
logger.info("ASR init")
config = read_yaml(path)
self.paraformer = RapidParaformer(config)

View File

@ -53,3 +53,10 @@ class MeloConf():
@inject
def __init__(self, config: Configuration) -> None:
self.melotts = config.get("melotts.url")
class LogConf():
level: str
@inject
def __init__(self,config: Configuration) -> None:
self.level = config.get("log.level")

15
src/log/log.py Normal file
View File

@ -0,0 +1,15 @@
import logging
from injector import singleton, inject
from ..configuration import LogConf
@singleton
class Logger(logging.Logger):
@inject
def __init__(self, config: LogConf) -> None:
super().__init__(__name__)
self.setLevel(config.level)
self.addHandler(logging.StreamHandler())
self.addHandler(logging.FileHandler('log.log'))