feature-img-1.0:待加入翻译api

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())){
+//            if(TranslateUtils.isContainChinese(param.getPrompt())){
-                //调用翻译api
+//                //调用翻译api
-                //prompt =
+//                //prompt =
-            }else{
+//                prompt = param.getPrompt();
-                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){