From be2d1db165fba481d0fa92afdda13b6a38690226 Mon Sep 17 00:00:00 2001
From: zchengrong <124802516+zchengrong@users.noreply.github.com>
Date: Fri, 6 Jun 2025 16:50:55 +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:=20fix=EF=BC=88=E4=BF=AE=E5=A4=8Dbug=EF=BC=89:=20=20b?=
 =?UTF-8?q?atch=20generate=20product=20/=20relight=20=E6=8E=A5=E5=8F=A3?=
 =?UTF-8?q?=E5=93=8D=E5=BA=94=E5=BC=82=E5=B8=B8=E4=BF=AE=E5=A4=8D=20docs?=
 =?UTF-8?q?=EF=BC=88=E6=96=87=E6=A1=A3=E5=8F=98=E6=9B=B4=EF=BC=89:=20refac?=
 =?UTF-8?q?tor=EF=BC=88=E9=87=8D=E6=9E=84=EF=BC=89:=20test(=E5=A2=9E?=
 =?UTF-8?q?=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/service.py   |  8 +++----
 .../service_batch_generate_relight_image.py   | 22 ++++++++++++++++++-
 2 files changed, 25 insertions(+), 5 deletions(-)

diff --git a/app/service/generate_batch_image/service.py b/app/service/generate_batch_image/service.py
index 2279382..6d007c8 100644
--- a/app/service/generate_batch_image/service.py
+++ b/app/service/generate_batch_image/service.py
@@ -6,15 +6,15 @@ from app.service.generate_batch_image.service_batch_pose_transform import batch_
 async def start_product_batch_generate(data):
     generate_clothes_task = batch_generate_product.delay(data.dict())
     print(generate_clothes_task)
-    product_publish_status(data.tasks_id, f"0/{data.batch_size}", "")
-    return {"task_id": data.tasks_id, "state": generate_clothes_task.state}
+    product_publish_status(data.batch_tasks_id, f"0/{len(data.batch_data_list)}", "")
+    return {"task_id": data.batch_tasks_id, "state": generate_clothes_task.state}
 
 
 async def start_relight_batch_generate(data):
     generate_clothes_task = batch_generate_relight.delay(data.dict())
     print(generate_clothes_task)
-    relight_publish_status(data.tasks_id, f"0/{data.batch_size}", "")
-    return {"task_id": data.tasks_id, "state": generate_clothes_task.state}
+    relight_publish_status(data.batch_tasks_id, f"0/{len(data.batch_data_list)}", "")
+    return {"task_id": data.batch_tasks_id, "state": generate_clothes_task.state}
 
 
 async def start_pose_transform_batch_generate(data):
diff --git a/app/service/generate_batch_image/service_batch_generate_relight_image.py b/app/service/generate_batch_image/service_batch_generate_relight_image.py
index d75b0a7..0a90646 100644
--- a/app/service/generate_batch_image/service_batch_generate_relight_image.py
+++ b/app/service/generate_batch_image/service_batch_generate_relight_image.py
@@ -184,5 +184,25 @@ if __name__ == '__main__':
             )
         ]
     )
-
     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",
+    #         }
+    #
+    #     ]
+    # }