From 66fa349e9986bf40cee9c7360bfed3c0b05e9a1b Mon Sep 17 00:00:00 2001
From: zchengrong <124802516+zchengrong@users.noreply.github.com>
Date: Tue, 22 Apr 2025 14:09:40 +0800
Subject: [PATCH] =?UTF-8?q?feat=EF=BC=88=E6=96=B0=E5=8A=9F=E8=83=BD?=
 =?UTF-8?q?=EF=BC=89:=20=20batch=20generate=20=E5=A2=9E=E5=8A=A0=E6=8E=A5?=
 =?UTF-8?q?=E5=8F=A3=E8=AF=B4=E6=98=8E=EF=BC=8Cdesign=20batch=E9=98=9F?=
 =?UTF-8?q?=E5=88=97=E4=BF=AE=E6=94=B9=20fix=EF=BC=88=E4=BF=AE=E5=A4=8Dbug?=
 =?UTF-8?q?=EF=BC=89:=20docs=EF=BC=88=E6=96=87=E6=A1=A3=E5=8F=98=E6=9B=B4?=
 =?UTF-8?q?=EF=BC=89:=20refactor=EF=BC=88=E9=87=8D=E6=9E=84=EF=BC=89:=20te?=
 =?UTF-8?q?st(=E5=A2=9E=E5=8A=A0=E6=B5=8B=E8=AF=95):?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 app/service/generate_batch_image/tasks.py | 28 ++++++++++++++++++
 app/service/generate_batch_image/test.py  | 36 +++++++++++++++++++++++
 2 files changed, 64 insertions(+)
 create mode 100644 app/service/generate_batch_image/tasks.py
 create mode 100644 app/service/generate_batch_image/test.py

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())