feat design 预处理接口迁移

This commit is contained in:
zhouchengrong
2024-05-30 09:48:13 +08:00
parent 13fec64125
commit 1f47d94431
7 changed files with 361 additions and 2 deletions

View File

@@ -0,0 +1,29 @@
import logging
import time
from fastapi import APIRouter
from app.schemas.pre_processing import DesignPreProcessingModel
from app.service.design_pre_processing.service import DesignPreprocessing
router = APIRouter()
logger = logging.getLogger()
@router.post("/design_pre_processing")
def design_pre_processing(request_data: DesignPreProcessingModel):
try:
logger.info(f"design_pre_processing request item is : @@@@@@:{request_data}")
code = 200
message = "access"
start_time = time.time()
server = DesignPreprocessing()
data = server.pipeline(image_list=request_data.sketches)
logger.info(f"design_pre_processing Run time is @@@@@@:{time.time() - start_time}")
except Exception as e:
code = 400
message = str(e)
data = str(e)
logger.warning(f"design Run Exception @@@@@@:{e}")
logger.info({"code": code, "message": message, "data": data})
return {"code": code, "message": message, "data": data}