This commit is contained in:
Dan Chen
2024-04-30 16:21:17 +08:00
committed by superobk
parent a586d97c12
commit f8e51135cb
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 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 from src.blackbox.blackbox_factory import BlackboxFactory
import uvicorn
from fastapi.middleware.cors import CORSMiddleware from fastapi.middleware.cors import CORSMiddleware
from injector import Injector from injector import Injector

View File

@ -4,15 +4,19 @@ from typing import Any, Coroutine
from fastapi import Request, Response, status from fastapi import Request, Response, status
from fastapi.responses import JSONResponse from fastapi.responses import JSONResponse
from ..log.log import Logger
from ..asr.rapid_paraformer.utils import read_yaml from ..asr.rapid_paraformer.utils import read_yaml
from ..asr.rapid_paraformer import RapidParaformer from ..asr.rapid_paraformer import RapidParaformer
from .blackbox import Blackbox from .blackbox import Blackbox
from injector import singleton from injector import singleton, inject
@singleton @singleton
class ASR(Blackbox): 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) config = read_yaml(path)
self.paraformer = RapidParaformer(config) self.paraformer = RapidParaformer(config)

View File

@ -53,3 +53,10 @@ class MeloConf():
@inject @inject
def __init__(self, config: Configuration) -> None: def __init__(self, config: Configuration) -> None:
self.melotts = config.get("melotts.url") 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'))