diff --git a/app/api/api_generate_image.py b/app/api/api_generate_image.py index 077fea9..c3332b6 100644 --- a/app/api/api_generate_image.py +++ b/app/api/api_generate_image.py @@ -1,11 +1,12 @@ import json import logging +import httpx 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, GenerateSloganImageModel +from app.schemas.generate_image import GenerateImageModel, GenerateProductImageModel, GenerateSingleLogoImageModel, GenerateRelightImageModel, GenerateMultiViewModel, BatchGenerateProductImageModel, BatchGenerateRelightImageModel, AgentTollGenerateImageModel, Flux2ToProductImgModel, GenerateSloganImageModel, GenerateImageFlux2KleinModel 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 @@ -22,6 +23,57 @@ logger = logging.getLogger() '''generate image''' +# flux2 klein +@router.post("/generate_image_flux2_klein") +async def generate_image_flux2_klein(request_item: GenerateImageFlux2KleinModel): + """ + 创建一个具有以下参数的请求体: + - **bucket_name**: OSS桶名 (必填) + - **object_name**: OSS对象名(文件路径)(必填) + + - **width**: 图片宽度,默认1024像素 (非必填,1024) + - **height**: 图片高度,默认1024像素 (非必填,默认1024) + - **prompt**: 文本提示词,用于模型推理等场景 (非必填,默认"") + - **steps**: 推理步数,控制模型生成过程的迭代次数 (非必填,默认4) + - **guidance**: 引导系数,调节提示词对生成结果的影响程度 (非必填,默认 4.0 ) + + ### 示例参数: + ``` + { + "bucket_name": "aida-users", + "object_name": "89/moodboard/5fdc698c-cb9b-4b36-afa9ce4-1-89.png", + "prompt": "a single item of sketch of dress, 4k, white background" + } + ``` + ### 输出示例: + ``` + { + "code": 200, + "msg": "OK!", + "data": { + "output_path": "aida-users/89/moodboard/5fdc698c-cb9b-4b36-afa9ce4-1-89.png" + } + } + ``` + """ + try: + logger.info(f"generate_image_flux2_gen_img request: {json.dumps(request_item.model_dump(), indent=4)}") + async with httpx.AsyncClient(timeout=120) as client: + resp = await client.post( + f"http://{settings.FLUX2_GEN_IMG_MODEL_URL}/predict", + json=request_item.model_dump(), + ) + result = resp.json() + + logger.info(f"generate_image_flux2_gen_img response: {json.dumps(result, indent=4)}") + return ResponseModel(data=result) + + except Exception as e: + logger.warning(f"generate_image_flux2_gen_img Run Exception @@@@@@:{e}") + raise HTTPException(status_code=404, detail=str(e)) + + +# sdxl @router.post("/generate_image") def generate_image(request_item: GenerateImageModel, background_tasks: BackgroundTasks): """ @@ -183,7 +235,6 @@ async def generate_slogan(request_data: GenerateSloganImageModel): """product image flux2.0""" - # @router.post("/img_to_product") # async def img_to_product(request_data: Flux2ToProductImgModel): # """ diff --git a/app/core/config.py b/app/core/config.py index 9e6813d..f3c1f69 100644 --- a/app/core/config.py +++ b/app/core/config.py @@ -64,6 +64,9 @@ class Settings(BaseSettings): # --- Design Callback Java 接口 --- JAVA_STREAM_API_URL: str = Field(default='', description="") + # --- flux2 klein model url --- + FLUX2_GEN_IMG_MODEL_URL: str = Field(default='', description="") + # --- 服务器IP --- A6000_SERVICE_HOST: str = Field(default='', description="") B_4_X_4090_SERVICE_HOST: str = Field(default='', description="") diff --git a/app/schemas/generate_image.py b/app/schemas/generate_image.py index d4c19a7..6fe2e70 100644 --- a/app/schemas/generate_image.py +++ b/app/schemas/generate_image.py @@ -1,6 +1,6 @@ -from typing import List +from typing import List, Optional -from pydantic import BaseModel +from pydantic import BaseModel, Field class GenerateMultiViewModel(BaseModel): @@ -8,8 +8,20 @@ class GenerateMultiViewModel(BaseModel): image_url: str +class GenerateImageFlux2KleinModel(BaseModel): + bucket_name: str = Field(..., description="OSS桶名,不传则为None") + object_name: str = Field(..., description="OSS对象名(文件路径),不传则为None") + # input_image_paths: Optional[List[str]] = Field(default=[], description="输入图片路径列表") + width: Optional[int] = Field(default=1024, description="图片宽度,默认512像素") + height: Optional[int] = Field(default=1024, description="图片高度,默认512像素") + prompt: Optional[str] = Field(default="", description="文本提示词,用于模型推理等场景") + steps: Optional[int] = Field(default=4, description="推理步数,控制模型生成过程的迭代次数") + guidance: Optional[float] = Field(default=4.0, description="引导系数,调节提示词对生成结果的影响程度") + + class GenerateImageModel(BaseModel): - tasks_id: str + bucket_name: str + object_name: str prompt: str image_url: str mode: str