feat(新功能): batch generate 增加接口说明,design batch队列修改

fix(修复bug):
docs(文档变更):
refactor(重构):
test(增加测试):
This commit is contained in:
zchengrong
2025-04-22 14:09:40 +08:00
parent bb3ef39e10
commit 66fa349e99
2 changed files with 64 additions and 0 deletions

View File

@@ -0,0 +1,28 @@
# import logging
#
# from celery import Celery
#
# from app.service.generate_batch_image.service_batch_generate_product_image import batch_generate_product
# from app.service.generate_batch_image.service_batch_generate_relight_image import batch_generate_relight
# from app.service.generate_batch_image.service_batch_pose_transform import batch_generate_pose_transform
#
# logger = logging.getLogger()
# celery_app = Celery('tasks', broker=f'amqp://rabbit:123456@18.167.251.121:5672//', backend='rpc://', BROKER_CONNECTION_RETRY_ON_STARTUP=True)
# celery_app.conf.worker_log_format = '%(asctime)s %(filename)s [line:%(lineno)d] %(levelname)s %(message)s'
# celery_app.conf.worker_hijack_root_logger = False
# logging.getLogger('pika').setLevel(logging.WARNING)
#
#
# @celery_app.task
# def batch_pose_transform_tasks(batch_request_data):
# batch_generate_pose_transform(batch_request_data)
#
#
# @celery_app.task
# def batch_generate_relight_tasks(batch_request_data):
# batch_generate_relight(batch_request_data)
#
#
# @celery_app.task
# def batch_generate_product_tasks(batch_request_data):
# batch_generate_product(batch_request_data)

View File

@@ -0,0 +1,36 @@
from app.schemas.generate_image import BatchGenerateRelightImageModel, BatchGenerateProductImageModel
from app.service.generate_batch_image.service_batch_generate_product_image import batch_generate_product
from app.service.generate_batch_image.service_batch_generate_relight_image import batch_generate_relight
if __name__ == '__main__':
rd = BatchGenerateProductImageModel(
tasks_id="test1-89",
image_strength=0.7,
prompt=" The best quality, masterpiece, real image.Outwear,high quality clothing details,8K realistic,HDR",
image_url="aida-results/result_40b1a2fe-e220-11ef-9bfa-0242ac150003.png",
product_type="single",
batch_size=2
)
x = batch_generate_product.delay(rd.dict())
print(x)
"""relight"""
# rd = BatchGenerateRelightImageModel(
# tasks_id="123-89",
# # prompt="beautiful woman, detailed face, sunshine, outdoor, warm atmosphere",
# prompt="Colorful black",
# image_url='aida-results/result_0000b606-1902-11ef-9424-0242ac180002.png',
# direction="Right Light",
# product_type="single",
# batch_size=2
# )
# batch_generate_relight.delay(rd.dict())
"""pose transform"""
# rd = BatchPoseTransformModel(
# tasks_id="123-89",
# image_url='aida-results/result_0000b606-1902-11ef-9424-0242ac180002.png',
# pose_id="1",
# batch_size=10
# )
# batch_pose_transform_tasks.delay(rd.dict())