fix 代码整理

app/api/api_outfit_matcher.py

@@ -16,9 +16,9 @@ logger = logging.getLogger()
 router = APIRouter()
 
 
-@RunTime
 @router.post("outfit_matcher")
 def outfit_matcher(request_item: OutfitMatcher):
+    start_time = time.time()
     request_item = dict(request_item)
     for i in range(len(request_item['query'])):
         request_item['query'][i] = dict(request_item['query'][i])
@@ -69,7 +69,6 @@ def outfit_matcher(request_item: OutfitMatcher):
         prepared_feature[hfd['item_name']] = hfd['features']
 
     result = []
-    start_time = time.time()
     for item in request_item['query']:
         # try:
         outfits = fashion_dataset.generate_outfit(item, request_item["topk"], request_item["max_outfits"])

app/api/api_similar_match.py

@@ -1,8 +1,12 @@
+import io
 import logging
 import time
 
+from PIL import Image
 from fastapi import APIRouter
+from matplotlib import pyplot as plt
 
+from app.core.config import SIMILAR_MATCH_DRAW
 from app.schemas.similar_match import SimilarMatchMItem
 from app.service.similar_match.service import SimilarMatch
 from app.service.utils.decorator import RunTime
@@ -22,6 +26,13 @@ def similar_match(request_item: SimilarMatchMItem):
         response_data = []
         for response in search_response[0]:
             response_data.append(response['entity'])
+
+        if SIMILAR_MATCH_DRAW:
+            resource_image = service.load_image(request_item.image_path)
+            similar_diagram_list = [service.load_image(image_url['image_path']) for image_url in response_data]
+            resource_image.save("similar_match/3/resource.png")
+            for i, image in enumerate(similar_diagram_list):
+                image.save(f"similar_match/3/{i}.png")
         return {"message": "ok", "data": response_data}
     except KeyError as e:
         logger.warning(str(e))