From 6250e763c3d7ff7488367d7c9d902d80e4d0a1b0 Mon Sep 17 00:00:00 2001
From: xupei <xupei3360@163.com>
Date: Tue, 8 Apr 2025 18:05:49 +0800
Subject: [PATCH] =?UTF-8?q?=E8=8E=B7=E5=8F=96=E6=89=80=E6=9C=89=E5=A7=BF?=
 =?UTF-8?q?=E5=8A=BF=E5=8F=98=E6=8D=A2=E7=9A=84pose?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../ai/da/controller/GenerateController.java   |  5 +++++
 .../com/ai/da/service/GenerateService.java     |  2 ++
 .../ai/da/service/impl/CreditsServiceImpl.java |  2 ++
 .../da/service/impl/GenerateServiceImpl.java   | 18 +++++++++++++++---
 4 files changed, 24 insertions(+), 3 deletions(-)

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<List<String>> 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<Generate> {
     GenerateResultVO sketchReconstructionGenerate(SketchReconstructionDTO sketchReconstructionDTO);
 
     SketchReconstructionVO getSketchReconstruction(Long projectId);
+
+    List<String> 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<CreditsDetailMapper, Credits
             credits = CreditsEventsEnum.LOGO.getValue();
         }else if (changeEvent.equals("Slogan")){
             credits =  CreditsEventsEnum.SLOGAN.getValue();
+        }else if (changeEvent.equals("PoseTransformation")){
+            credits =  CreditsEventsEnum.POSE_TRANSFORMATION.getValue();
         }
 
 //        BigDecimal finalCredits = currentCredits.subtract(new BigDecimal(credits));
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 752b90d1..840cbd05 100644
--- a/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java
+++ b/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java
@@ -1009,9 +1009,15 @@ public class GenerateServiceImpl extends ServiceImpl<GenerateMapper, Generate> 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<GenerateMapper, Generate> i
         return vo;
     }
 
+    public List<String> getAllPose(){
+        String posePath = "aida-sys-image/pose/pose-1.gif";
+        String preSignedUrl = minioUtil.getPreSignedUrl(posePath, CommonConstant.MINIO_IMAGE_EXPIRE_TIME);
+        return Arrays.asList(preSignedUrl);
+    }
+
 }