AiDA_Python/app/api/api_design_pre_processing.py

import json
import logging
from fastapi import APIRouter, HTTPException
from app.schemas.pre_processing import DesignPreProcessingModel
from app.schemas.response_template import ResponseModel
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}")
        server = DesignPreprocessing()
        data = server.pipeline(image_list=request_data.sketches)
        logger.info(f"design response @@@@@@:{json.dumps(data, indent=4)}")
    except Exception as e:
        logger.warning(f"design Run Exception @@@@@@:{e}")
        raise HTTPException(status_code=404, detail=str(e))
    return ResponseModel(data=data)