AiDA_Python/app/api/api_design.py

import json
import logging
import os

from fastapi import APIRouter, HTTPException, UploadFile, File, Form, BackgroundTasks

from app.schemas.design import DesignModel, DesignProgressModel, ModelProgressModel, DBGConfigModel
from app.schemas.response_template import ResponseModel
from app.service.design.model_process_service import model_transpose
from app.service.design_batch.service import start_design_batch_generate
from app.service.design_fast.design_generate import design_generate, design_generate_v2
from app.service.design_fast.utils.redis_utils import Redis

router = APIRouter()
logger = logging.getLogger()


@router.post("/design")
def design(request_data: DesignModel, background_tasks: BackgroundTasks):
    """
        创建一个具有以下参数的请求体:
        示例参数：
        {
            "objects": [
                {
                    "basic": {
                        "body_point_test": {
                            "waistband_right": [
                                200,
                                241
                            ],
                            "hand_point_right": [
                                223,
                                297
                            ],
                            "waistband_left": [
                                112,
                                241
                            ],
                            "hand_point_left": [
                                92,
                                305
                            ],
                            "shoulder_left": [
                                99,
                                116
                            ],
                            "shoulder_right": [
                                215,
                                116
                            ]
                        },
                        "layer_order": true,
                        "scale_bag": 0.7,
                        "scale_earrings": 0.16,
                        "self_template": true,
                        "single_overall": "overall",
                        "switch_category": ""
                    },
                    "items": [
                        {
                            "businessId": 270372,
                            "color": "30 28 28",
                            "image_id": 69780,
                            "offset": [
                                0,
                                0
                            ],
                            "path": "aida-sys-image/images/female/trousers/0825000630.jpg",
                            "print": {
                                "element": {
                                    "element_angle_list": [],
                                    "element_path_list": [],
                                    "element_scale_list": [],
                                    "location": []
                                },
                                "overall": {
                                    "location": [],
                                    "print_angle_list": [],
                                    "print_path_list": [],
                                    "print_scale_list": []
                                },
                                "single": {
                                    "location": [],
                                    "print_angle_list": [],
                                    "print_path_list": [],
                                    "print_scale_list": []
                                }
                            },
                            "priority": 10,
                            "resize_scale": [
                                1.0,
                                1.0
                            ],
                            "type": "Trousers"
                        },
                        {
                            "businessId": 270373,
                            "color": "30 28 28",
                            "image_id": 98243,
                            "offset": [
                                0,
                                0
                            ],
                            "path": "aida-sys-image/images/female/blouse/0902003811.jpg",
                            "print": {
                                "element": {
                                    "element_angle_list": [],
                                    "element_path_list": [],
                                    "element_scale_list": [],
                                    "location": []
                                },
                                "overall": {
                                    "location": [],
                                    "print_angle_list": [],
                                    "print_path_list": [],
                                    "print_scale_list": []
                                },
                                "single": {
                                    "location": [],
                                    "print_angle_list": [],
                                    "print_path_list": [],
                                    "print_scale_list": []
                                }
                            },
                            "priority": 11,
                            "resize_scale": [
                                1.0,
                                1.0
                            ],
                            "type": "Blouse"
                        },
                        {
                            "businessId": 270374,
                            "color": "172 68 68",
                            "image_id": 98244,
                            "offset": [
                                0,
                                0
                            ],
                            "path": "aida-sys-image/images/female/outwear/0825000410.jpg",
                            "print": {
                                "element": {
                                    "element_angle_list": [],
                                    "element_path_list": [],
                                    "element_scale_list": [],
                                    "location": []
                                },
                                "overall": {
                                    "location": [],
                                    "print_angle_list": [],
                                    "print_path_list": [],
                                    "print_scale_list": []
                                },
                                "single": {
                                    "location": [],
                                    "print_angle_list": [],
                                    "print_path_list": [],
                                    "print_scale_list": []
                                }
                            },
                            "priority": 12,
                            "resize_scale": [
                                1.0,
                                1.0
                            ],
                            "transparent":{
                                "mask_url":"test/transparent_test/transparent_mask.png",
                                "scale":0.1
                            },
                            "type": "Outwear"
                        },
                        {
                            "body_path": "aida-sys-image/models/female/5bdfe7ca-64eb-44e4-b03d-8e517520c795.png",
                            "image_id": 96090,
                            "type": "Body"
                        }
                    ]
                }
            ],
            "process_id": "83"
        }
    """
    # logger.info(f"design request item is : @@@@@@:{json.dumps(request_data.dict())}")
    # data = generate(request_data=request_data)
    # logger.info(f"design response @@@@@@:{json.dumps(data)}")
    #

    try:
        logger.info(f"design request item is : @@@@@@:{json.dumps(request_data.dict())}")
        data = design_generate(request_data=request_data)
        logger.info(f"design response @@@@@@:{json.dumps(data)}")
    except Exception as e:
        logger.warning(f"design Run Exception @@@@@@:{e}")
        raise HTTPException(status_code=404, detail=str(e))
    return ResponseModel(data=data)


@router.post("/design_v2")
async def design_v2(request_data: DesignModel, background_tasks: BackgroundTasks):
    """
        创建一个具有以下参数的请求体:
        示例参数：
        {
            "objects": [
                {
                    "basic": {
                        "body_point_test": {
                            "waistband_right": [
                                200,
                                241
                            ],
                            "hand_point_right": [
                                223,
                                297
                            ],
                            "waistband_left": [
                                112,
                                241
                            ],
                            "hand_point_left": [
                                92,
                                305
                            ],
                            "shoulder_left": [
                                99,
                                116
                            ],
                            "shoulder_right": [
                                215,
                                116
                            ]
                        },
                        "layer_order": true,
                        "scale_bag": 0.7,
                        "scale_earrings": 0.16,
                        "self_template": true,
                        "single_overall": "overall",
                        "switch_category": ""
                    },
                    "items": [
                        {
                            "businessId": 270372,
                            "color": "30 28 28",
                            "image_id": 69780,
                            "offset": [
                                0,
                                0
                            ],
                            "path": "aida-sys-image/images/female/trousers/0825000630.jpg",
                            "print": {
                                "element": {
                                    "element_angle_list": [],
                                    "element_path_list": [],
                                    "element_scale_list": [],
                                    "location": []
                                },
                                "overall": {
                                    "location": [],
                                    "print_angle_list": [],
                                    "print_path_list": [],
                                    "print_scale_list": []
                                },
                                "single": {
                                    "location": [],
                                    "print_angle_list": [],
                                    "print_path_list": [],
                                    "print_scale_list": []
                                }
                            },
                            "priority": 10,
                            "resize_scale": [
                                1.0,
                                1.0
                            ],
                            "type": "Trousers"
                        },
                        {
                            "businessId": 270373,
                            "color": "30 28 28",
                            "image_id": 98243,
                            "offset": [
                                0,
                                0
                            ],
                            "path": "aida-sys-image/images/female/blouse/0902003811.jpg",
                            "print": {
                                "element": {
                                    "element_angle_list": [],
                                    "element_path_list": [],
                                    "element_scale_list": [],
                                    "location": []
                                },
                                "overall": {
                                    "location": [],
                                    "print_angle_list": [],
                                    "print_path_list": [],
                                    "print_scale_list": []
                                },
                                "single": {
                                    "location": [],
                                    "print_angle_list": [],
                                    "print_path_list": [],
                                    "print_scale_list": []
                                }
                            },
                            "priority": 11,
                            "resize_scale": [
                                1.0,
                                1.0
                            ],
                            "type": "Blouse"
                        },
                        {
                            "businessId": 270374,
                            "color": "172 68 68",
                            "image_id": 98244,
                            "offset": [
                                0,
                                0
                            ],
                            "path": "aida-sys-image/images/female/outwear/0825000410.jpg",
                            "print": {
                                "element": {
                                    "element_angle_list": [],
                                    "element_path_list": [],
                                    "element_scale_list": [],
                                    "location": []
                                },
                                "overall": {
                                    "location": [],
                                    "print_angle_list": [],
                                    "print_path_list": [],
                                    "print_scale_list": []
                                },
                                "single": {
                                    "location": [],
                                    "print_angle_list": [],
                                    "print_path_list": [],
                                    "print_scale_list": []
                                }
                            },
                            "priority": 12,
                            "resize_scale": [
                                1.0,
                                1.0
                            ],
                            "transparent":{
                                "mask_url":"test/transparent_test/transparent_mask.png",
                                "scale":0.1
                            },
                            "type": "Outwear"
                        },
                        {
                            "body_path": "aida-sys-image/models/female/5bdfe7ca-64eb-44e4-b03d-8e517520c795.png",
                            "image_id": 96090,
                            "type": "Body"
                        }
                    ]
                }
            ],
            "process_id": "83"
        }
    """
    try:
        # 异步
        logger.info(f"generate_image request item is : @@@@@@:{json.dumps(request_data.dict())}")
        background_tasks.add_task(design_generate_v2, request_data)
    except Exception as e:
        logger.warning(f"design Run Exception @@@@@@:{e}")
        raise HTTPException(status_code=404, detail=str(e))
    return ResponseModel()


@router.post('/get_progress')
def get_progress(request_data: DesignProgressModel):
    """
        获取design 进度
        创建一个具有以下参数的请求体:
        - **process_id**: 进度id

        示例参数：
        {
          "process_id": "6878547032381675"
        }
    """
    try:
        logger.info(f"get_progress request item is : @@@@@@:{json.dumps(request_data.dict())}")
        process_id = request_data.process_id
        r = Redis()
        data = r.read(key=process_id)
        if data is None:
            raise ValueError(f"No progress ID: {process_id}")
        logging.info(f"get_progress process_id @@@@@@ : {process_id} , progress : {json.dumps(data)}")
    except Exception as e:
        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):
    """
        获取模特图片预处理
        创建一个具有以下参数的请求体:
        - **model_path**: 模特图片的minio或s3 url地址

        示例参数：
        {
          "model_path": "aida-users/10/models/female/9c788f5b-b8c7-424c-b149-025aeb0bda51model.jpg"
        }
    """
    try:
        logger.info(f"model_process request item is : @@@@@@:{json.dumps(request_data.dict())}")

        data = model_transpose(image_path=request_data.model_path)
        logger.info(f"model_process response @@@@@@:{json.dumps(data)}")
    except Exception as e:
        logger.warning(f"model_process Run Exception @@@@@@:{e}")
        raise HTTPException(status_code=404, detail=str(e))
    return ResponseModel(data=data)


# ##############################################################


@router.post("/design_batch_generate")
async def design(file: UploadFile = File(...),
                 tasks_id: str = Form(...),
                 user_id: str = Form(...),
                 file_name: str = Form(...),
                 total: int = Form(...)
                 ):
    dbg_config = DBGConfigModel(
        tasks_id=tasks_id,
        user_id=user_id,
        file_name=file_name,
        total=total
    )
    contents = await file.read()
    file_name = file.filename
    await save_request_file(contents, file_name)
    return await start_design_batch_generate(dbg_config, contents)


async def save_request_file(contents, file_name):
    # 创建保存文件的目录（如果不存在）
    save_dir = os.path.join(os.getcwd(), "service/design_batch", "request_data")
    if not os.path.exists(save_dir):
        os.makedirs(save_dir)
    # 处理文件
    file_path = os.path.join(save_dir, file_name)
    with open(file_path, "wb") as f:
        f.write(contents)