From b43f9baead0e46575bb4f4839bc02c875f8d2e57 Mon Sep 17 00:00:00 2001 From: xupei Date: Wed, 25 Sep 2024 11:30:51 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=20imageToSketch?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java b/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java index 54b5878e..3b83ca8d 100644 --- a/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java +++ b/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java @@ -780,7 +780,7 @@ public class GenerateServiceImpl extends ServiceImpl i String imagePath = collectionElement.getUrl(); log.info(minioUtil.getPreSignedUrl(imagePath, CommonConstant.MINIO_IMAGE_EXPIRE_TIME)); - String imageName = imagePath.substring(imagePath.lastIndexOf("/") + 1); + String imageName = UUID.randomUUID().toString(); String objectName = accountId + "/imageToSketch/" + imageName; String style = imageToSketchDTO.getStyle(); String styleCode = style.equals(SketchStyle.THICK.getValue()) ? "1" : @@ -819,6 +819,7 @@ public class GenerateServiceImpl extends ServiceImpl i generateDetailMapper.insert(generateDetail); String clothCategory = pythonService.getClothCategory(sketchPath, imageToSketchDTO.getGender()); + clothCategory = BusinessException.getMessageFromResource(clothCategory.toUpperCase()); return new GenerateResultVO(generateDetail.getId(), minioUtil.getPreSignedUrl(sketchPath, CommonConstant.MINIO_IMAGE_EXPIRE_TIME), "Success", clothCategory); }