diff --git a/src/main/java/com/ai/da/controller/GenerateController.java b/src/main/java/com/ai/da/controller/GenerateController.java index e98c4465..0d149a9f 100644 --- a/src/main/java/com/ai/da/controller/GenerateController.java +++ b/src/main/java/com/ai/da/controller/GenerateController.java @@ -137,6 +137,11 @@ public class GenerateController { return Response.success(sketchReconstruction); } + @ApiOperation(value = "获取pose transfer的所有pose") + @GetMapping("/getAllPose") + public Response> getAllPose(){ + return Response.success(generateService.getAllPose()); + } diff --git a/src/main/java/com/ai/da/service/GenerateService.java b/src/main/java/com/ai/da/service/GenerateService.java index 445a8eb7..140cbdb6 100644 --- a/src/main/java/com/ai/da/service/GenerateService.java +++ b/src/main/java/com/ai/da/service/GenerateService.java @@ -61,4 +61,6 @@ public interface GenerateService extends IService { GenerateResultVO sketchReconstructionGenerate(SketchReconstructionDTO sketchReconstructionDTO); SketchReconstructionVO getSketchReconstruction(Long projectId); + + List getAllPose(); } diff --git a/src/main/java/com/ai/da/service/impl/CreditsServiceImpl.java b/src/main/java/com/ai/da/service/impl/CreditsServiceImpl.java index 959461c6..4bf27209 100644 --- a/src/main/java/com/ai/da/service/impl/CreditsServiceImpl.java +++ b/src/main/java/com/ai/da/service/impl/CreditsServiceImpl.java @@ -303,6 +303,8 @@ public class CreditsServiceImpl extends ServiceImpl i if (!StringUtil.isNullOrEmpty(resultJson)){ PoseTransformationVO poseTransformationVO = new Gson().fromJson(redisUtil.getFromString(key), PoseTransformationVO.class); if (poseTransformationVO.getStatus().equals("Success")){ - poseTransformationVO.setGifUrl(minioUtil.getPreSignedUrl(poseTransformationVO.getGifUrl(), CommonConstant.MINIO_IMAGE_EXPIRE_TIME)); - poseTransformationVO.setVideoUrl(minioUtil.getPreSignedUrl(poseTransformationVO.getVideoUrl(), CommonConstant.MINIO_IMAGE_EXPIRE_TIME)); - poseTransformationVO.setFirstFrameUrl(minioUtil.getPreSignedUrl(poseTransformationVO.getFirstFrameUrl(), CommonConstant.MINIO_IMAGE_EXPIRE_TIME)); + if (!poseTransformationVO.getGifUrl().equals("None")){ + poseTransformationVO.setGifUrl(minioUtil.getPreSignedUrl(poseTransformationVO.getGifUrl(), CommonConstant.MINIO_IMAGE_EXPIRE_TIME)); + } + if (!poseTransformationVO.getVideoUrl().equals("None")){ + poseTransformationVO.setVideoUrl(minioUtil.getPreSignedUrl(poseTransformationVO.getVideoUrl(), CommonConstant.MINIO_IMAGE_EXPIRE_TIME)); + } + if (!poseTransformationVO.getFirstFrameUrl().equals("None")){ + poseTransformationVO.setFirstFrameUrl(minioUtil.getPreSignedUrl(poseTransformationVO.getFirstFrameUrl(), CommonConstant.MINIO_IMAGE_EXPIRE_TIME)); + } } return poseTransformationVO; }else { @@ -1187,4 +1193,10 @@ public class GenerateServiceImpl extends ServiceImpl i return vo; } + public List getAllPose(){ + String posePath = "aida-sys-image/pose/pose-1.gif"; + String preSignedUrl = minioUtil.getPreSignedUrl(posePath, CommonConstant.MINIO_IMAGE_EXPIRE_TIME); + return Arrays.asList(preSignedUrl); + } + }