diff --git a/app/service/design_pre_processing/service.py b/app/service/design_pre_processing/service.py index f6f239a..272dc7e 100644 --- a/app/service/design_pre_processing/service.py +++ b/app/service/design_pre_processing/service.py @@ -9,6 +9,7 @@ from urllib3.exceptions import ResponseError from app.core.config import * from app.service.design.utils.design_ensemble import get_keypoint_result +from app.service.utils.generate_uuid import generate_uuid from app.service.utils.oss_client import oss_get_image, oss_upload_image @@ -121,8 +122,11 @@ class DesignPreprocessing: image_bytes = cv2.imencode(".jpg", item['obj'])[1].tobytes() # self.minio_client.put_object(item['image_url'].split("/", 1)[0], item['image_url'].split("/", 1)[1], io.BytesIO(image_bytes), len(image_bytes), content_type="image/jpeg", ) bucket_name = item['image_url'].split("/", 1)[0] + # 由于延迟,bounding box后的sketch与前端缓存的sketch有误差 object_name = item['image_url'].split("/", 1)[1] - oss_upload_image(bucket=bucket_name, object_name=object_name, image_bytes=image_bytes) + new_object = f"{object_name[:object_name.rfind('/') + 1]}{generate_uuid()}.{object_name.split('.', 1)[1]}" + oss_upload_image(bucket=bucket_name, object_name=new_object, image_bytes=image_bytes) + item['new_image_url'] = f"{bucket_name}/{new_object}" print(f"Object '{item['image_url'].split('/', 1)[1]}' overwritten successfully.") except ResponseError as err: print(f"Error: {err}")