feat:

fix:  slogan 服务迁移

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):
 #     """

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