From 19d170c1bfe908053ecaa1a55a144750fd361a03 Mon Sep 17 00:00:00 2001 From: xupei Date: Mon, 22 Apr 2024 15:32:59 +0800 Subject: [PATCH] =?UTF-8?q?=E8=BE=93=E5=85=A5prompt=E4=BF=AE=E9=A5=B0?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../da/service/impl/GenerateServiceImpl.java | 21 +++++++++++++++++-- 1 file changed, 19 insertions(+), 2 deletions(-) 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 6b26bea4..d0613304 100644 --- a/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java +++ b/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java @@ -232,20 +232,37 @@ public class GenerateServiceImpl extends ServiceImpl i if (StringUtil.isNullOrEmpty(text)) { throw new BusinessException("please.input.the.caption"); } - generate.setText(text); +// generate.setText(text); + modifyPrompt(text, generate); break; case "image": if (Objects.isNull(elementId)) { throw new BusinessException("please.choose.an.image"); } generate.setElementId(elementId); + modifyPrompt(text, generate); break; case "text-image": if (StringUtil.isNullOrEmpty(text) || Objects.isNull(elementId)) { throw new BusinessException("please.input.the.caption.and.choose.an.image"); } - generate.setText(text); +// generate.setText(text); generate.setElementId(elementId); + modifyPrompt(text, generate); + default: + } + } + + private void modifyPrompt(String userInput, Generate generate){ + switch (generate.getLevel1Type()) { + case "Moodboard": + generate.setText(userInput + ",high quality"); + break; + case "Printboard": + generate.setText(userInput + ", fabric print, high quality"); + break; + case "Sketchboard": + generate.setText("a single item of " + userInput + "with clean background, hand-drawing sketch style, high quality"); default: } }