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