feat 新增 process lookbooks 接口

fix

app/api/api_process_lookbooks.py

@@ -0,0 +1,55 @@
+import logging
+import os
+import shutil
+from typing import List
+
+import tqdm
+from fastapi import UploadFile, File, APIRouter
+
+from app.service.lookbooks.service import create_image_batch_requests
+
+logger = logging.getLogger()
+router = APIRouter()
+
+
+@router.post("/process_lookbooks/")
+async def process_lookbooks(files: List[UploadFile] = File(...)):
+    lookbook_dir = "service/lookbooks/temp_lookbooks"
+    os.makedirs(lookbook_dir, exist_ok=True)
+
+    lookbook_list = []
+    for file in files:
+        file_path = os.path.join(lookbook_dir, file.filename)
+        with open(file_path, "wb") as f:
+            shutil.copyfileobj(file.file, f)
+        lookbook_list.append(file_path)
+
+    image_list = []
+    for look_book_path in tqdm.tqdm(lookbook_list):
+        lookbook_name = os.path.splitext(os.path.basename(look_book_path))[0]
+        output_dir = os.path.join("app/service/lookbooks/fashion_documents/lookbook/images", lookbook_name)
+        os.makedirs(output_dir, exist_ok=True)
+        if not os.listdir(output_dir):
+            from unstructured.partition.pdf import partition_pdf
+            partition_pdf(
+                filename=look_book_path,
+                extract_images_in_pdf=True,
+                infer_table_structure=False,
+                chunking_strategy="by_title",
+                max_characters=4000,
+                new_after_n_chars=3800,
+                combine_text_under_n_chars=2000,
+                extract_image_block_output_dir=output_dir,
+            )
+        else:
+            current_images = os.listdir(output_dir)
+            image_list.extend([os.path.join(output_dir, x) for x in current_images])
+
+    image_description_results_file = create_image_batch_requests(image_list, "app/service/lookbooks/fashion_documents/lookbook/results")
+
+    shutil.rmtree(lookbook_dir)
+
+    if image_description_results_file:
+        return {"message": "Lookbooks processed successfully", "result_file": image_description_results_file}
+    else:
+        return {"message": "No new images to process"}

app/api/api_route.py

@@ -1,9 +1,6 @@
 from fastapi import APIRouter
 
-from app.api import api_test
-from app.api import api_outfit_matcher
-from app.api import api_attribute
-from app.api import api_similar_match
+from app.api import api_test, api_outfit_matcher, api_attribute, api_similar_match, api_process_lookbooks
 
 router = APIRouter()
 
@@ -11,3 +8,4 @@ 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")