From c24862507f49ed29ac35c8bd3d6bf68350694cc8 Mon Sep 17 00:00:00 2001
From: zcr <cgzhou@aidlab.hk>
Date: Wed, 28 Jan 2026 15:37:03 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20fix:=20=20slogan=20=E6=9C=8D=E5=8A=A1?=
 =?UTF-8?q?=E8=BF=81=E7=A7=BB?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 app/api/api_generate_image.py | 28 ++++++++++++++++++++++++++--
 app/schemas/generate_image.py |  7 +++++++
 2 files changed, 33 insertions(+), 2 deletions(-)

diff --git a/app/api/api_generate_image.py b/app/api/api_generate_image.py
index 7eb5bdf..c98ef12 100644
--- a/app/api/api_generate_image.py
+++ b/app/api/api_generate_image.py
@@ -5,7 +5,7 @@ import requests
 from fastapi import APIRouter, BackgroundTasks, HTTPException
 
 from app.core.config import settings
-from app.schemas.generate_image import GenerateImageModel, GenerateProductImageModel, GenerateSingleLogoImageModel, GenerateRelightImageModel, GenerateMultiViewModel, BatchGenerateProductImageModel, BatchGenerateRelightImageModel, AgentTollGenerateImageModel, Flux2ToProductImgModel
+from app.schemas.generate_image import GenerateImageModel, GenerateProductImageModel, GenerateSingleLogoImageModel, GenerateRelightImageModel, GenerateMultiViewModel, BatchGenerateProductImageModel, BatchGenerateRelightImageModel, AgentTollGenerateImageModel, Flux2ToProductImgModel, GenerateSloganImageModel
 from app.schemas.pose_transform import BatchPoseTransformModel
 from app.schemas.response_template import ResponseModel
 from app.service.generate_batch_image.service import start_product_batch_generate, start_relight_batch_generate, start_pose_transform_batch_generate
@@ -156,6 +156,31 @@ def generate_single_logo_image(tasks_id: str):
     return ResponseModel(data=data['data'])
 
 
+"""slogan """
+
+
+@router.post("/generate_slogan")
+async def generate_slogan(request_data: GenerateSloganImageModel):
+    """
+        ### 请求体示例:
+        ```json
+        {
+          "num_point": 16,
+          "image_url": "aida-slogan/6886785f-0aac-4052-b6fd-7ae20a841d8d.png",
+          "prompt": "123",
+          "tasks_id": "string-89"
+        }
+        ```
+    """
+    try:
+        logger.info(f"generate_slogan request item is : @@@@@@:{json.dumps(request_data.dict(), indent=4)}")
+        data = requests.post(f"http://{settings.A6000_SERVICE_HOST}:10020/api/slogan", json=request_data.dict())
+        logger.info(f"generate_slogan response @@@@@@:{json.dumps(json.loads(data.content), indent=4)}")
+        return ResponseModel(data=json.loads(data.content))
+    except Exception as e:
+        logger.warning(f"generate_slogan Run Exception @@@@@@:{e}")
+
+
 """product image flux2.0"""
 
 
@@ -190,7 +215,6 @@ async def img_to_product(request_data: Flux2ToProductImgModel):
 
 '''product image 停用'''
 
-
 # @router.post("/generate_product_image")
 # def generate_product_image(request_item: GenerateProductImageModel, background_tasks: BackgroundTasks):
 #     """
diff --git a/app/schemas/generate_image.py b/app/schemas/generate_image.py
index a26d5d6..d4c19a7 100644
--- a/app/schemas/generate_image.py
+++ b/app/schemas/generate_image.py
@@ -24,6 +24,13 @@ class GenerateSingleLogoImageModel(BaseModel):
     seed: str
 
 
+class GenerateSloganImageModel(BaseModel):
+    num_point: int
+    tasks_id: str
+    prompt: str
+    image_url: str
+
+
 class GenerateProductImageModel(BaseModel):
     tasks_id: str
     prompt: str