feat 模特预处理接口

fix
This commit is contained in:
zhouchengrong
2024-06-24 16:35:34 +08:00
parent d827147e8a
commit 26a56b4d9f
4 changed files with 49 additions and 3 deletions

View File

@@ -3,8 +3,9 @@ import logging
from fastapi import APIRouter, HTTPException
from app.schemas.design import DesignModel, DesignProgressModel
from app.schemas.design import DesignModel, DesignProgressModel, ModelProgressModel
from app.schemas.response_template import ResponseModel
from app.service.design.model_process_service import model_transpose
from app.service.design.service import generate
from app.service.design.utils.redis_utils import Redis
@@ -38,3 +39,16 @@ def get_progress(request_data: DesignProgressModel):
logger.warning(f"get_progress Run Exception @@@@@@:{e}")
raise HTTPException(status_code=404, detail=str(e))
return ResponseModel(data=data)
@router.post('/model_process')
def model_process(request_data: ModelProgressModel):
try:
logger.info(f"model_process request item is : @@@@@@:{request_data.dict()}")
data = model_transpose(image_path=request_data.model_path)
logger.info(f"model_process response @@@@@@:{json.dumps(data, indent=4)}")
except Exception as e:
logger.warning(f"model_process Run Exception @@@@@@:{e}")
raise HTTPException(status_code=404, detail=str(e))
return ResponseModel(data=data)