From b2e0738045a3f16f0419b37463f941480ba3792d Mon Sep 17 00:00:00 2001
From: Penny <2500338766@qq.com>
Date: Sat, 22 Apr 2023 13:46:05 +0800
Subject: [PATCH] =?UTF-8?q?feature-img-1.0:=E5=BE=85=E5=8A=A0=E5=85=A5?=
 =?UTF-8?q?=E7=BF=BB=E8=AF=91api?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../impl/StableDiffusionServiceImpl.java      | 20 ++++++++++---------
 1 file changed, 11 insertions(+), 9 deletions(-)

diff --git a/bnyer-services/bnyer-img/src/main/java/com/bnyer/img/service/impl/StableDiffusionServiceImpl.java b/bnyer-services/bnyer-img/src/main/java/com/bnyer/img/service/impl/StableDiffusionServiceImpl.java
index d881e5b..9be2f56 100644
--- a/bnyer-services/bnyer-img/src/main/java/com/bnyer/img/service/impl/StableDiffusionServiceImpl.java
+++ b/bnyer-services/bnyer-img/src/main/java/com/bnyer/img/service/impl/StableDiffusionServiceImpl.java
@@ -41,20 +41,22 @@ public class StableDiffusionServiceImpl implements StableDiffusionService {
     @Override
     public TextToImgVo textToImg(TextToImgDto param) {
         try{
-            String prompt = "";
+            //String prompt = "";
             //判断prompt是否包含中文,中文则翻译，否则跳过
-            if(TranslateUtils.isContainChinese(param.getPrompt())){
-                //调用翻译api
-                //prompt =
-            }else{
-                prompt = param.getPrompt();
-            }
+//            if(TranslateUtils.isContainChinese(param.getPrompt())){
+//                //调用翻译api
+//                //prompt =
+//                prompt = param.getPrompt();
+//            }else{
+//                prompt = param.getPrompt();
+//            }
 
             //TODO 根据选择的风格来选择模型
             Map<String, Object> map = new HashMap<>();
             map.put("width",param.getWidth() == null ? 512 : param.getWidth());
             map.put("height",param.getHeight() == null ? 512 : param.getHeight());
-            map.put("prompt", prompt);
+            //map.put("prompt", prompt);
+            map.put("prompt", param.getPrompt());
             map.put("seed",-1);
             map.put("batch_size",1);
             map.put("cfg_scale",7);
@@ -66,7 +68,7 @@ public class StableDiffusionServiceImpl implements StableDiffusionService {
             map.put("steps",25);
             map.put("negative_prompt","nsfw");
             log.info("请求stable_diffusion请求体为:【{}】", JSON.toJSONString(map));
-            JSONObject jsonObject = restTemplate.postForObject("http://localhost:7860/sdapi/v1/txt2img", map, JSONObject.class);
+            JSONObject jsonObject = restTemplate.postForObject("http://124.220.2.237:7860/sdapi/v1/txt2img", map, JSONObject.class);
             log.info("请求stable_diffusion响应体的为:【{}】", JSON.toJSONString(jsonObject));
             TextToImgVo img = new TextToImgVo();
             if(jsonObject != null && jsonObject.getJSONArray("images").size() > 0){