aidlab/AiDA_Python
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat（新功能）:

Browse Source 
fix（修复bug）:  batch generate product / relight 接口响应异常修复
docs（文档变更）:
refactor（重构）:
test(增加测试):
This commit is contained in:
zchengrong
2025-06-06 16:50:55 +08:00
parent 12bb128351
commit be2d1db165
2 changed files with 25 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
app/service/generate_batch_image/service.py
View File

@@ -6,15 +6,15 @@ from app.service.generate_batch_image.service_batch_pose_transform import batch_
async def start_product_batch_generate(data): async def start_product_batch_generate(data):
generate_clothes_task = batch_generate_product.delay(data.dict()) generate_clothes_task = batch_generate_product.delay(data.dict())
print(generate_clothes_task) print(generate_clothes_task)
product_publish_status(data.tasks_id, f"0/{data.batch_size}", "") product_publish_status(data.batch_tasks_id, f"0/{len(data.batch_data_list)}", "")
return {"task_id": data.tasks_id, "state": generate_clothes_task.state} return {"task_id": data.batch_tasks_id, "state": generate_clothes_task.state}
async def start_relight_batch_generate(data): async def start_relight_batch_generate(data):
generate_clothes_task = batch_generate_relight.delay(data.dict()) generate_clothes_task = batch_generate_relight.delay(data.dict())
print(generate_clothes_task) print(generate_clothes_task)
relight_publish_status(data.tasks_id, f"0/{data.batch_size}", "") relight_publish_status(data.batch_tasks_id, f"0/{len(data.batch_data_list)}", "")
return {"task_id": data.tasks_id, "state": generate_clothes_task.state} return {"task_id": data.batch_tasks_id, "state": generate_clothes_task.state}
async def start_pose_transform_batch_generate(data): async def start_pose_transform_batch_generate(data):

22
app/service/generate_batch_image/service_batch_generate_relight_image.py
View File

@@ -184,5 +184,25 @@ if __name__ == '__main__':
) )
] ]
) )
batch_generate_relight(rd.dict()) batch_generate_relight(rd.dict())
# X = {
# "batch_tasks_id": "abcd",
# "user_id": "89",
# "batch_data_list": [
# {
# "tasks_id": "123-5464",
# "product_type": "overall",
# "image_url": "aida-users/89/product_image/02894523-19b5-46eb-a9c6-2f512f5fec84-0-89.png",
# "prompt": "Colorful black",
# "direction": "Right Light",
# },
# {
# "tasks_id": "123-5464",
# "product_type": "overall",
# "image_url": "aida-users/89/product_image/02894523-19b5-46eb-a9c6-2f512f5fec84-0-89.png",
# "prompt": "Colorful black",
# "direction": "Right Light",
# }
#
# ]
# }