sketch拼贴 添加返回服装分类

src/main/java/com/ai/da/mapper/primary/entity/SketchReconstruction.java

@@ -15,4 +15,6 @@ public class SketchReconstruction extends BaseEntity{
 
     private Long generateDetailId;
 
+    private String gender;
+
 }

src/main/java/com/ai/da/model/vo/SketchReconstructionVO.java

@@ -10,4 +10,8 @@ public class SketchReconstructionVO {
 
     private boolean isLiked;
 
+    private String category;
+
+    private String categoryValue;
+
 }

src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java

@@ -1141,6 +1141,7 @@ public class GenerateServiceImpl extends ServiceImpl<GenerateMapper, Generate> i
             sketchReconstruction.setProjectId(projectId);
             sketchReconstruction.setCollageImgSketchUrl(targetPath);
             sketchReconstruction.setGenerateDetailId(generateResultVO.getId());
+            sketchReconstruction.setGender(sketchReconstructionDTO.getGender());
             sketchReconstruction.setCreateTime(LocalDateTime.now());
             sketchReconstructionMapper.insert(sketchReconstruction);
         }else {
@@ -1165,6 +1166,10 @@ public class GenerateServiceImpl extends ServiceImpl<GenerateMapper, Generate> i
             GenerateDetail generateDetail = generateDetailMapper.selectById(sketchReconstruction.getGenerateDetailId());
             vo.setCollageSketchUrl(minioUtil.getPreSignedUrl(generateDetail.getUrl(), CommonConstant.MINIO_IMAGE_EXPIRE_TIME));
             vo.setLiked(generateDetail.getIsLike().equals((byte)1));
+            String clothCategory = pythonService.getClothCategory(generateDetail.getUrl(), sketchReconstruction.getGender());
+            String messageFromResource = BusinessException.getMessageFromResource(clothCategory.toUpperCase());
+            vo.setCategory(clothCategory);
+            vo.setCategoryValue(messageFromResource);
         }
         return vo;
     }