From ddc089faa40f1528b799cf90d7ba87891b1072aa Mon Sep 17 00:00:00 2001 From: xupei Date: Wed, 3 Sep 2025 15:48:45 +0800 Subject: [PATCH] =?UTF-8?q?BUGFIX:=E5=8D=B0=E8=8A=B1pin=E7=9A=84=E6=95=B0?= =?UTF-8?q?=E9=87=8F=E4=B8=BA9=E6=97=B6=E6=97=A0=E6=B3=95design?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/main/java/com/ai/da/python/PythonService.java | 4 +++- src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java | 2 ++ 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/src/main/java/com/ai/da/python/PythonService.java b/src/main/java/com/ai/da/python/PythonService.java index 9d123b9b..45d8a75a 100644 --- a/src/main/java/com/ai/da/python/PythonService.java +++ b/src/main/java/com/ai/da/python/PythonService.java @@ -555,7 +555,9 @@ public class PythonService { return 0; } else { long pinNum = printBoardElements.stream().filter(f -> f.getHasPin() == 1).count(); - if (designNum - pinNum < designNum/2) { + if (designNum - pinNum < 0){ + return RandomsUtil.randomSysFile(0L, (long) (pinNum/2 + 1)); + } else if (designNum - pinNum < designNum/2) { return RandomsUtil.randomSysFile(0L, designNum - pinNum + 1); } else { return RandomsUtil.randomSysFile(0L, (long) (designNum/2 + 1)); 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 3b08bcc4..a834f987 100644 --- a/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java +++ b/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java @@ -1284,6 +1284,8 @@ public class GenerateServiceImpl extends ServiceImpl i public String imageToSketchAsync(ImageToSketchDTO imageToSketchDTO, String collagePictureUrl, Long projectId) { Long accountId = UserContext.getUserHolder().getId(); log.info("imageToSketch parameter : {}", imageToSketchDTO); + // 目前没有用其他提取模型，暂时写死flux + imageToSketchDTO.setModelName("flux"); // 检查积分是否够本次扣除 // CreditsEventsEnum event = CreditsEventsEnum.IMAGE_TO_SKETCH;