From adde1b007e3a401fd4b6334904ed7a9d29e1232f Mon Sep 17 00:00:00 2001
From: litianxiang <litianxiangxtt@163.com>
Date: Tue, 18 Nov 2025 17:22:08 +0800
Subject: [PATCH] =?UTF-8?q?fix:=E7=94=B1=E4=BA=8Eai=E8=BF=94=E5=9B=9E?=
 =?UTF-8?q?=E7=9A=84=E5=9B=BE=E7=89=87=E4=B8=BApng=EF=BC=8C=E5=9C=A8?=
 =?UTF-8?q?=E6=AD=A4=E6=94=AF=E6=8C=81png=E6=A0=BC=E5=BC=8F?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../service/impl/TryOnEffectServiceImpl.java   | 18 +++++++++++++++---
 1 file changed, 15 insertions(+), 3 deletions(-)

diff --git a/src/main/java/com/aida/lanecarford/service/impl/TryOnEffectServiceImpl.java b/src/main/java/com/aida/lanecarford/service/impl/TryOnEffectServiceImpl.java
index 7b26884..cafc653 100644
--- a/src/main/java/com/aida/lanecarford/service/impl/TryOnEffectServiceImpl.java
+++ b/src/main/java/com/aida/lanecarford/service/impl/TryOnEffectServiceImpl.java
@@ -231,13 +231,25 @@ public class TryOnEffectServiceImpl extends ServiceImpl<TryOnEffectMapper, TryOn
         int startIndex = tryonUrl.indexOf("lanecarford");
         String aiRreultlogicalUrl = "";
         if (startIndex != -1) {
-            // 找到jpg的位置并加上jpg的长度
-            int endIndex = tryonUrl.indexOf(".jpg") + ".jpg".length();
+            // 支持jpg和png格式的图片识别
+            int jpgEndIndex = tryonUrl.indexOf(".jpg");
+            int pngEndIndex = tryonUrl.indexOf(".png");
+            
+            int endIndex = -1;
+
+            if (jpgEndIndex != -1 && (pngEndIndex == -1 || jpgEndIndex < pngEndIndex)) {
+                // 找到jpg格式
+                endIndex = jpgEndIndex + ".jpg".length();
+            } else if (pngEndIndex != -1 && (jpgEndIndex == -1 || pngEndIndex < jpgEndIndex)) {
+                // 找到png格式
+                endIndex = pngEndIndex + ".png".length();
+            }
+            
             if (endIndex != -1) {
                 String resultString = tryonUrl.substring(startIndex, endIndex);
                 aiRreultlogicalUrl = AITryOnEffect(prompt, Arrays.asList(resultString));
             } else {
-                log.error("不是换脸后的url或者换脸结果有误");
+                log.error("未找到jpg或png格式的图片，无法识别图片类型");
             }
         }
         return minioUtil.convertToPresignedUrl(aiRreultlogicalUrl, CommonConstants.MINIO_PATH_TIMEOUT);