aidlab/aida_back
7
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

sketch拼贴 添加返回服装分类

Browse Source
This commit is contained in:
xupei
2025-03-26 13:39:39 +08:00
parent a4365abd4d
commit 20b15c0b78
3 changed files with 11 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

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

4
src/main/java/com/ai/da/model/vo/SketchReconstructionVO.java
View File

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

5
src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java
View File

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