From f0fccaaf510c48198970056439af7222b6d0050a Mon Sep 17 00:00:00 2001 From: xupei Date: Mon, 21 Jul 2025 17:11:10 +0800 Subject: [PATCH] =?UTF-8?q?TASK:pose=20transfer=E6=B7=BB=E5=8A=A0=E8=BF=94?= =?UTF-8?q?=E5=9B=9E=E5=8F=82=E6=95=B0?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../java/com/ai/da/model/vo/ToProductImageResultVO.java | 2 ++ .../java/com/ai/da/service/impl/GenerateServiceImpl.java | 7 ++++++- 2 files changed, 8 insertions(+), 1 deletion(-) diff --git a/src/main/java/com/ai/da/model/vo/ToProductImageResultVO.java b/src/main/java/com/ai/da/model/vo/ToProductImageResultVO.java index b9d7f90c..1ce7ebc8 100644 --- a/src/main/java/com/ai/da/model/vo/ToProductImageResultVO.java +++ b/src/main/java/com/ai/da/model/vo/ToProductImageResultVO.java @@ -14,4 +14,6 @@ public class ToProductImageResultVO extends ToProductImageResult implements AllC private Long parentId; @ApiModelProperty("用户输入的提示词") private String prompt; + @ApiModelProperty("姿势序号") + private Integer poseId; } 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 113288ff..838458b5 100644 --- a/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java +++ b/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java @@ -1246,6 +1246,12 @@ public class GenerateServiceImpl extends ServiceImpl i ToProductImageResultVO toProductImageResultVO = new ToProductImageResultVO(); toProductImageResultVO.setParentId(poseTransformDTO.getParentId()); toProductImageResultVO.setResultType(Module.poseTransfer.getValue()); + toProductImageResultVO.setTaskId(taskId); + toProductImageResultVO.setTaskStatus(taskStatus); + toProductImageResultVO.setSourceUrl(minioUtil.getPreSignedUrl(productImage, CommonConstant.MINIO_IMAGE_EXPIRE_TIME)); + toProductImageResultVO.setPoseId(poseId); + toProductImageResultVO.setModelName(poseTransformDTO.getModelName()); + if (Objects.nonNull(poseTransformDTO.getIsDefaultLike()) && poseTransformDTO.getIsDefaultLike()) { // 满足条件下添加到like poseTransformation.setIsLiked((byte) 1); @@ -1258,7 +1264,6 @@ public class GenerateServiceImpl extends ServiceImpl i } else if (Objects.nonNull(poseTransformDTO.getIsDefaultLike()) && Objects.nonNull(poseTransformDTO.getParentId())) { toProductImageResultVO.setParentId(poseTransformDTO.getParentId()); } - toProductImageResultVO.setTaskId(taskId); if (flag) {