Merge remote-tracking branch 'origin/develop' into develop

This commit is contained in:
shahaibo
2025-06-10 10:55:02 +08:00
10 changed files with 369 additions and 129 deletions

View File

@@ -0,0 +1,39 @@
import logging
from fastapi import APIRouter, HTTPException
from app.schemas.project_info_extraction import ProjectInfoExtractionModel
from app.schemas.response_template import ResponseModel
from app.service.project_info_extraction.service import ProjectInfoExtraction
router = APIRouter()
logger = logging.getLogger()
@router.post("/extraction_project_info")
def extraction_project_info(request_data: ProjectInfoExtractionModel):
"""
通过prompt 提取project_name,role ,gender ,style。
创建一个具有以下参数的请求体:
- **prompt**:
示例参数:
{
"prompt": "海边派对主题的系列设计",
"image_list": [
"https://www.minio-api.aida.com.hk/test/test123.jpg?X-Amz-Algorithm=AWS4-HMAC-SHA256&X-Amz-Credential=vXKFLSJkYeEq2DrSZvkB%2F20250519%2Fus-east-1%2Fs3%2Faws4_request&X-Amz-Date=20250519T050808Z&X-Amz-Expires=7200&X-Amz-SignedHeaders=host&X-Amz-Signature=296ff07cc4692d0a26ddffac582064f036494af343389fe60193dc2c5dc883ff"
],
"file_list": [
""
]
}
"""
try:
logger.info(f"extraction_project_info request item is : @@@@@@:{request_data}")
service = ProjectInfoExtraction(request_data)
data = service.get_result()
logger.info(f"extraction_project_info response @@@@@@:{data}")
except Exception as e:
logger.warning(f"extraction_project_info Run Exception @@@@@@:{e}")
raise HTTPException(status_code=404, detail=str(e))
return ResponseModel(data=data)

View File

@@ -5,16 +5,17 @@ from app.api import api_attribute_retrieve, api_query_image
from app.api import api_brand_dna
from app.api import api_brighten
from app.api import api_chat_robot
from app.api import api_clothing_seg
from app.api import api_design
from app.api import api_design_pre_processing
from app.api import api_extraction_project_info
from app.api import api_generate_image
from app.api import api_image2sketch
from app.api import api_mannequins_edit
from app.api import api_pose_transform
from app.api import api_prompt_generation
from app.api import api_clothing_seg
from app.api import api_super_resolution
from app.api import api_recommendation
from app.api import api_super_resolution
from app.api import api_test
router = APIRouter()
@@ -36,3 +37,4 @@ router.include_router(api_mannequins_edit.router, tags=['api_mannequins_edit'],
router.include_router(api_agent_generate_image.router, tags=['api_agent_generate_image'], prefix="/api")
router.include_router(api_pose_transform.router, tags=['api_pose_transform'], prefix="/api")
router.include_router(api_clothing_seg.router, tags=['api_clothing_seg'], prefix="/api")
router.include_router(api_extraction_project_info.router, tags=['api_extraction_project_info'], prefix="/api")