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