diff --git a/app/service/generate_batch_image/tasks.py b/app/service/generate_batch_image/tasks.py new file mode 100644 index 0000000..a79d402 --- /dev/null +++ b/app/service/generate_batch_image/tasks.py @@ -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) \ No newline at end of file diff --git a/app/service/generate_batch_image/test.py b/app/service/generate_batch_image/test.py new file mode 100644 index 0000000..ece4b39 --- /dev/null +++ b/app/service/generate_batch_image/test.py @@ -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())