fix 代码整理

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))