zcr
c792106f02
All checks were successful
git commit AiDA python develop 分支构建部署 / scheduled_deploy (push) Has been skipped
490 lines
20 KiB
Python
490 lines
20 KiB
Python
import json
|
|
import logging
|
|
|
|
import requests
|
|
from fastapi import APIRouter, HTTPException, BackgroundTasks
|
|
|
|
from app.schemas.design import DesignModel, ModelProgressModel, DesignStreamModel, SAMRequestModel
|
|
from app.schemas.response_template import ResponseModel
|
|
from app.service.design_fast.design_generate import design_generate, design_generate_v2
|
|
from app.service.design_fast.model_process_service import model_transpose
|
|
|
|
router = APIRouter()
|
|
logger = logging.getLogger()
|
|
|
|
|
|
@router.post("/design")
|
|
def design(request_data: DesignModel):
|
|
"""
|
|
- **objects.items.transparent**:
|
|
```json
|
|
"transparent":{
|
|
"mask_url":"test/transparent_test/transparent_mask.png",
|
|
"scale":0.1
|
|
},
|
|
```
|
|
- **mask_url** 为空"" -> 单件衣服透明
|
|
- **mask_url** 非空"mask_url" -> 区域透明
|
|
- **transpose** 镜像模式 ,:"top_bottom"或"left_right"
|
|
- **rotate** 45,
|
|
- 创建一个具有以下参数的请求体:
|
|
示例参数：
|
|
```json
|
|
{
|
|
"objects": [
|
|
{
|
|
"basic": {
|
|
"body_point_test": {
|
|
"waistband_right": [
|
|
203,
|
|
249
|
|
],
|
|
"hand_point_right": [
|
|
229,
|
|
343
|
|
],
|
|
"waistband_left": [
|
|
119,
|
|
248
|
|
],
|
|
"hand_point_left": [
|
|
97,
|
|
343
|
|
],
|
|
"shoulder_left": [
|
|
108,
|
|
107
|
|
],
|
|
"shoulder_right": [
|
|
212,
|
|
107
|
|
]
|
|
},
|
|
"layer_order": true,
|
|
"preview_submit": "preview",
|
|
"scale_bag": 0.7,
|
|
"scale_earrings": 0.16,
|
|
"self_template": true,
|
|
"single_overall": "overall",
|
|
"switch_category": ""
|
|
},
|
|
"items": [
|
|
{
|
|
"businessId": 2115382,
|
|
"color": "",
|
|
"image_id": 61686,
|
|
"offset": [
|
|
0,
|
|
0
|
|
],
|
|
"path": "aida-sys-image/images/female/dress/0628000564.jpg",
|
|
"transpose": [
|
|
1,
|
|
1
|
|
],
|
|
"rotate": 45,
|
|
"print": {
|
|
"element": {
|
|
"element_angle_list": [],
|
|
"element_path_list": [],
|
|
"element_scale_list": [],
|
|
"location": []
|
|
},
|
|
"overall": {
|
|
"location": [
|
|
[
|
|
53.0,
|
|
118.5
|
|
]
|
|
],
|
|
"print_angle_list": [
|
|
0.0
|
|
],
|
|
"print_path_list": [
|
|
"aida-users/89/print/02d57aa8-f342-4e1d-b02c-b278f94dcfe6-3-89.png"
|
|
],
|
|
"print_scale_list": [
|
|
[
|
|
0.5,
|
|
0.5
|
|
]
|
|
],
|
|
"gap": [
|
|
[
|
|
10,
|
|
10
|
|
]
|
|
]
|
|
},
|
|
"single": {
|
|
"location": [],
|
|
"print_angle_list": [],
|
|
"print_path_list": [],
|
|
"print_scale_list": []
|
|
}
|
|
},
|
|
"priority": 10,
|
|
"resize_scale": [
|
|
1.0,
|
|
1.0
|
|
],
|
|
"seg_mask_url": "aida-clothing/mask/mask_9698b428-eb93-11f0-9327-0242c0a80003.png",
|
|
"type": "Dress"
|
|
},
|
|
{
|
|
"body_path": "aida-sys-image/models/female/2e4815b9-1191-419d-94ed-5771239ca4a5.png",
|
|
"image_id": 67277,
|
|
"type": "Body"
|
|
}
|
|
]
|
|
}
|
|
],
|
|
"process_id": "89"
|
|
}
|
|
```
|
|
"""
|
|
# logger.info(f"design request item is : @@@@@@:{json.dumps(request_data.dict(),indent=4)}")
|
|
# data = generate(request_data=request_data)
|
|
# logger.info(f"design response @@@@@@:{json.dumps(data, indent=4)}")
|
|
#
|
|
|
|
try:
|
|
logger.info(f"design request item is : @@@@@@:{json.dumps(request_data.dict(), indent=4)}")
|
|
data = design_generate(request_data=request_data)
|
|
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)
|
|
|
|
|
|
@router.post("/design_v2")
|
|
async def design_v2(request_data: DesignStreamModel, background_tasks: BackgroundTasks):
|
|
"""
|
|
创建一个具有以下参数的请求体:
|
|
示例参数：
|
|
{
|
|
"objects": [
|
|
{
|
|
"basic": {
|
|
"body_point_test": {
|
|
"waistband_right": [
|
|
203,
|
|
249
|
|
],
|
|
"hand_point_right": [
|
|
229,
|
|
343
|
|
],
|
|
"waistband_left": [
|
|
119,
|
|
248
|
|
],
|
|
"hand_point_left": [
|
|
97,
|
|
343
|
|
],
|
|
"shoulder_left": [
|
|
108,
|
|
107
|
|
],
|
|
"relation_type": "System",
|
|
"shoulder_right": [
|
|
212,
|
|
107
|
|
],
|
|
"relation_id": 1020356
|
|
},
|
|
"layer_order": false,
|
|
"scale_bag": 0.7,
|
|
"scale_earrings": 0.16,
|
|
"self_template": false,
|
|
"single_overall": "overall",
|
|
"switch_category": ""
|
|
},
|
|
"items": [
|
|
{
|
|
"color": "209 196 171",
|
|
"image_id": 84093,
|
|
"offset": [
|
|
1,
|
|
1
|
|
],
|
|
"path": "aida-users/89/sketchboard/female/Outwear/0943d209-7ce0-408c-bc61-83f15da94138.png",
|
|
"print": {
|
|
"element": {
|
|
"element_angle_list": [],
|
|
"element_path_list": [],
|
|
"element_scale_list": [],
|
|
"location": []
|
|
},
|
|
"overall": {
|
|
"location": [
|
|
[
|
|
0.0,
|
|
0.0
|
|
]
|
|
],
|
|
"print_angle_list": [
|
|
0.0,
|
|
0.0
|
|
],
|
|
"print_path_list": [],
|
|
"print_scale_list": [
|
|
[
|
|
0.0,
|
|
0.0
|
|
]
|
|
]
|
|
},
|
|
"single": {
|
|
"location": [],
|
|
"print_angle_list": [],
|
|
"print_path_list": [],
|
|
"print_scale_list": []
|
|
}
|
|
},
|
|
"resize_scale": [
|
|
1.0,
|
|
1.0
|
|
],
|
|
"type": "Outwear"
|
|
},
|
|
{
|
|
"color": "63 71 73",
|
|
"image_id": 100496,
|
|
"offset": [
|
|
1,
|
|
1
|
|
],
|
|
"path": "aida-sys-image/images/female/blouse/0628001684.jpg",
|
|
"print": {
|
|
"element": {
|
|
"element_angle_list": [],
|
|
"element_path_list": [],
|
|
"element_scale_list": [],
|
|
"location": []
|
|
},
|
|
"overall": {
|
|
"location": [
|
|
[
|
|
0.0,
|
|
0.0
|
|
]
|
|
],
|
|
"print_angle_list": [
|
|
0.0,
|
|
0.0
|
|
],
|
|
"print_path_list": [],
|
|
"print_scale_list": [
|
|
[
|
|
0.0,
|
|
0.0
|
|
]
|
|
]
|
|
},
|
|
"single": {
|
|
"location": [],
|
|
"print_angle_list": [],
|
|
"print_path_list": [],
|
|
"print_scale_list": []
|
|
}
|
|
},
|
|
"resize_scale": [
|
|
1.0,
|
|
1.0
|
|
],
|
|
"type": "Blouse"
|
|
},
|
|
{
|
|
"color": "111 78 63",
|
|
"gradient": "aida-gradient/f69b98e8-4248-4f7a-98a2-21bac41bf3e0.png",
|
|
"image_id": 92193,
|
|
"offset": [
|
|
1,
|
|
1
|
|
],
|
|
"path": "aida-sys-image/images/female/trousers/0825001160.jpg",
|
|
"print": {
|
|
"element": {
|
|
"element_angle_list": [],
|
|
"element_path_list": [],
|
|
"element_scale_list": [],
|
|
"location": []
|
|
},
|
|
"overall": {
|
|
"location": [
|
|
[
|
|
0.0,
|
|
0.0
|
|
]
|
|
],
|
|
"print_angle_list": [
|
|
0.0,
|
|
0.0
|
|
],
|
|
"print_path_list": [],
|
|
"print_scale_list": [
|
|
[
|
|
0.0,
|
|
0.0
|
|
]
|
|
]
|
|
},
|
|
"single": {
|
|
"location": [],
|
|
"print_angle_list": [],
|
|
"print_path_list": [],
|
|
"print_scale_list": []
|
|
}
|
|
},
|
|
"resize_scale": [
|
|
1.0,
|
|
1.0
|
|
],
|
|
"type": "Trousers"
|
|
},
|
|
{
|
|
"body_path": "aida-sys-image/models/female/2e4815b9-1191-419d-94ed-5771239ca4a5.png",
|
|
"image_id": 67277,
|
|
"offset": [
|
|
1,
|
|
1
|
|
],
|
|
"resize_scale": [
|
|
1.0,
|
|
1.0
|
|
],
|
|
"type": "Body"
|
|
}
|
|
],
|
|
"objectSign": "65830966"
|
|
}
|
|
],
|
|
"process_id": "4802946666428422",
|
|
"requestId": "1d1e7641-0d62-4da2-adc0-b4404910723c",
|
|
"callback_url": "https://api.aida.com.hk/api/third/party/receiveDesignResults"
|
|
}
|
|
"""
|
|
try:
|
|
# 异步
|
|
logger.info(f"generate_image request item is : @@@@@@:{json.dumps(request_data.dict(), indent=4)}")
|
|
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("/seg_anything")
|
|
async def seg_anything(request_data: SAMRequestModel):
|
|
"""
|
|
**Segment Anything 交互式分割接口**
|
|
|
|
通过传入图片路径和点击的点坐标，返回分割后的掩码数据。
|
|
|
|
### 参数说明:
|
|
- **image_path**: 图片在服务器或云端的相对路径。
|
|
- **points**: 交互点的坐标列表。每个点为 [x, y] 像素格式。
|
|
- **labels**: 坐标点的属性标签，必须与 points 长度一致：
|
|
- 1: **前景点** (代表想要分割出的区域)
|
|
- 0: **背景点** (代表想要排除的区域)
|
|
|
|
### 请求体示例:
|
|
```json
|
|
{
|
|
"image_path": "aida-users/89/sketch/4e8fe37d-7068-400a-ac94-c01647fa5f6f.png",
|
|
"points": [[310, 403], [493, 375], [261, 266], [404, 484]],
|
|
"labels": [1, 1, 0, 1]
|
|
}
|
|
```
|
|
"""
|
|
try:
|
|
logger.info(f"seg_anything request item is : @@@@@@:{json.dumps(request_data.dict(), indent=4)}")
|
|
data = requests.post("http://10.1.1.240:10075/predict", json=request_data.dict())
|
|
logger.info(f"seg_anything response @@@@@@:{json.dumps(json.loads(data.content), indent=4)}")
|
|
return ResponseModel(data=json.loads(data.content))
|
|
except Exception as e:
|
|
logger.warning(f"seg_anything Run Exception @@@@@@:{e}")
|
|
|
|
|
|
# @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(), indent=4)}")
|
|
# 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, indent=4)}")
|
|
# 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(), indent=4)}")
|
|
|
|
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)
|
|
|
|
|
|
"""design 批量处理 停用"""
|
|
# @router.post("/design_batch_generate")
|
|
# async def design_batch(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)
|