feat 新增 process lookbooks 接口

fix
This commit is contained in:
zhouchengrong
2024-10-22 15:11:08 +08:00
parent 61af80541b
commit ed017fdf9d
8 changed files with 24 additions and 24 deletions

View File

@@ -11,7 +11,7 @@ logger = logging.getLogger()
router = APIRouter()
@router.post("")
@router.post("/attribute")
def attribute(request_data: AttributeModel):
logger.info(f"attribute requests is @@@@@@@@@@@:{request_data}")
service = AttributeRecognition()

View File

@@ -16,7 +16,7 @@ logger = logging.getLogger()
router = APIRouter()
@router.post("outfit_matcher")
@router.post("/outfit_matcher")
def outfit_matcher(request_item: OutfitMatcher):
start_time = time.time()
request_item = dict(request_item)

View File

@@ -1,18 +1,17 @@
import logging
import os
import shutil
from typing import List
import aiofiles
import tqdm
from fastapi import UploadFile, File, APIRouter, BackgroundTasks, Form
from app.service.lookbooks.service import create_image_batch_requests, process_lookbook_task # 引入服务逻辑
from app.service.lookbooks.service import process_lookbook_task # 引入服务逻辑
logger = logging.getLogger()
router = APIRouter()
@router.post("/process-lookbooks/")
@router.post("/process-lookbooks")
async def process_lookbooks(
background_tasks: BackgroundTasks,
files: List[UploadFile] = File(...),

View File

@@ -5,7 +5,7 @@ from app.api import api_test, api_outfit_matcher, api_attribute, api_similar_mat
router = APIRouter()
router.include_router(api_test.router, tags=["test"], prefix="/test")
router.include_router(api_outfit_matcher.router, tags=["outfit_matcher"], prefix="/api/outfit_matcher")
router.include_router(api_attribute.router, tags=["attribute"], prefix="/api/attribute")
router.include_router(api_similar_match.router, tags=["similar_match"], prefix="/api/similar_match")
router.include_router(api_process_lookbooks.router, tags=["process_lookbooks"], prefix="/api/process_lookbooks")
router.include_router(api_outfit_matcher.router, tags=["outfit_matcher"], prefix="/api")
router.include_router(api_attribute.router, tags=["attribute"], prefix="/api")
router.include_router(api_similar_match.router, tags=["similar_match"], prefix="/api")
router.include_router(api_process_lookbooks.router, tags=["process-lookbooks"], prefix="/api")

View File

@@ -16,11 +16,11 @@ router = APIRouter()
@RunTime
@router.post("similar_match")
@router.post("/similar_match")
def similar_match(request_item: SimilarMatchMItem):
try:
if request_item.result_number <= 0:
raise KeyError("result number can't be less than 0")
raise KeyError("results number can't be less than 0")
service = SimilarMatch(request_item)
search_response = service.match_features()
response_data = []
@@ -36,4 +36,4 @@ def similar_match(request_item: SimilarMatchMItem):
return {"message": "ok", "data": response_data}
except KeyError as e:
logger.warning(str(e))
return {"message": "result number can't be less than 0", "data": []}
return {"message": "results number can't be less than 0", "data": []}