File size: 1,581 Bytes
1ef9436 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 |
from fastapi import FastAPI
from fastapi.exceptions import RequestValidationError
from fastapi.responses import PlainTextResponse
from loguru import logger
from pydantic import BaseModel
from ..web_configs import WEB_CONFIGS
from .asr_worker import load_asr_model, process_asr
app = FastAPI()
if WEB_CONFIGS.ENABLE_ASR:
ASR_HANDLER = load_asr_model()
else:
ASR_HANDLER = None
class ASRItem(BaseModel):
user_id: int # User 识别号，用于区分不用的用户调用
request_id: str # 请求 ID，用于生成 TTS & 数字人
wav_path: str # wav 文件路径
@app.post("/asr")
async def get_asr(asr_item: ASRItem):
# 语音转文字
result = ""
status = "success"
if ASR_HANDLER is None:
result = "ASR not enable in sever"
status = "fail"
logger.error(f"ASR not enable...")
else:
result = process_asr(ASR_HANDLER, asr_item.wav_path)
logger.info(f"ASR res for id {asr_item.request_id}, res = {result}")
return {"user_id": asr_item.user_id, "request_id": asr_item.request_id, "status": status, "result": result}
@app.get("/asr/check")
async def check_server():
return {"message": "server enabled"}
@app.exception_handler(RequestValidationError)
async def validation_exception_handler(request, exc):
"""调 API 入参错误的回调接口
Args:
request (_type_): _description_
exc (_type_): _description_
Returns:
_type_: _description_
"""
logger.info(request)
logger.info(exc)
return PlainTextResponse(str(exc), status_code=400)
|