34 lines
1.1 KiB
Python
34 lines
1.1 KiB
Python
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": "海边派对主题的系列设计"
|
|
}
|
|
"""
|
|
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)
|