bnyer
/
bnyer-cloud
1
0
Fork 0
Code Issues Pull Requests Projects Releases 2 Wiki Activity
Browse Source

feature:添加flagStudio方法

feature-1.1
Penny 3 years ago
parent
a6a3ba514d
commit
b43b84ee64
3 changed files with 145 additions and 2 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Split View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 31
      bnyer-services/bnyer-img/src/main/java/com/bnyer/img/config/FlagStudioConfig.java
  2. 3
      bnyer-services/bnyer-img/src/main/java/com/bnyer/img/enums/AiPaintButtonEnum.java
  3. 113
      bnyer-services/bnyer-img/src/main/java/com/bnyer/img/service/impl/StableDiffusionServiceImpl.java

31
bnyer-services/bnyer-img/src/main/java/com/bnyer/img/config/FlagStudioConfig.java
View File

@ -0,0 +1,31 @@
package com.bnyer.img.config;
import lombok.Getter;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.cloud.context.config.annotation.RefreshScope;
import org.springframework.context.annotation.Configuration;
/**
* flagStudio配置类
* @author chengkun
* @date 2022/04/27 16:40
*/
@Configuration
@ConfigurationProperties(prefix = "flagstudio")
@Getter
@RefreshScope
public class FlagStudioConfig {
@Value("${flagstudio.txt2ImgUrl}")
public String txt2ImgUrl;
@Value("${flagstudio.tokenUrl}")
public String tokenUrl;
@Value("${flagstudio.apiKey}")
public String apiKey;
}

3
bnyer-services/bnyer-img/src/main/java/com/bnyer/img/enums/AiPaintButtonEnum.java
View File

@ -12,7 +12,8 @@ import lombok.Getter;
@AllArgsConstructor
public enum AiPaintButtonEnum {
SD(1,"stable diffusion"),
TX(0,"tx文生图");
TX(0,"tx文生图"),
FS(2,"FlagStudio");
private int code;

113
bnyer-services/bnyer-img/src/main/java/com/bnyer/img/service/impl/StableDiffusionServiceImpl.java
View File

@ -16,6 +16,7 @@ import com.bnyer.common.core.utils.file.Base64ToMultipartFileUtils;
import com.bnyer.common.core.vo.TextToImgVo;
import com.bnyer.common.redis.service.RedisService;
import com.bnyer.file.api.RemoteFileService;
import com.bnyer.img.config.FlagStudioConfig;
import com.bnyer.img.config.StableDiffusionConfig;
import com.bnyer.img.config.TencentTranslateConfig;
import com.bnyer.img.enums.AiPaintButtonEnum;
@ -33,6 +34,8 @@ import com.tencentcloudapi.tmt.v20180321.models.TextTranslateRequest;
import com.tencentcloudapi.tmt.v20180321.models.TextTranslateResponse;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.http.HttpEntity;
import org.springframework.http.HttpHeaders;
import org.springframework.stereotype.Service;
import org.springframework.web.client.RestTemplate;
import org.springframework.web.multipart.MultipartFile;
@ -61,6 +64,9 @@ public class StableDiffusionServiceImpl implements StableDiffusionService {
@Autowired
private StableDiffusionConfig stableDiffusionConfig;
@Autowired
private FlagStudioConfig flagStudioConfig;
@Autowired
private TiktokUserService tiktokUserService;
@ -230,7 +236,7 @@ public class StableDiffusionServiceImpl implements StableDiffusionService {
log.error("腾讯文生图调用错误!"+e.getMessage());
throw new ServiceException(e.getMessage(),500);
}
}else{
}else if(button == AiPaintButtonEnum.SD.getCode()){
//采用sd
try{
//内容提示词
@ -343,6 +349,111 @@ public class StableDiffusionServiceImpl implements StableDiffusionService {
}catch (Exception e){
log.error("文本翻译错误!"+e);
}
}else{
//FlagStudio
try{
//获取token
JSONObject tokenObject = restTemplate.getForObject(flagStudioConfig.getTokenUrl() + "?apikey=" + flagStudioConfig.getApiKey(), JSONObject.class);
String token = "";
if(tokenObject != null && tokenObject.getString("code").equals("200")){
token = tokenObject.getJSONObject("data").getString("token");
}
//描述词处理
//负面提示词
String negaPrompt = "lowres,bad anatomy,bad hands,text,error,missing fingers,extra digit,fewer digits,cropped,worst quality,low quality,normal quality,jpeg artifacts,signature,watermark,username,blurry,lowres,text, cropped,worst quality,low quality,normal quality,jpeg artifacts,signature, watermark,username,blurry,text,signature,watermark,simple background,toony,dated,low res,line art,flat colors,nsfw,easynegative,naked,nsfw";
//内容提示词
String contentPrompt = param.getPrompt();
//积极预设词
String activePrompt = "8k,masterpiece,best quality,ultra high res,";
String resultPrompt = activePrompt + contentPrompt;
//画布大小
Integer width = 512;
Integer height = 512;
//1:1正方形
if(param.getHeight() == 512 && param.getWidth() == 512){
width = 512;
height = 512;
//16:9宽屏
}else if(param.getHeight() == 512 && param.getWidth() == 1024){
width = 768;
height = 432;
//9:16竖屏
}else{
width = 432;
height = 768;
}
// 返回的resp是一个TextToImageResponse的实例，与请求对象对应
TextToImgVo img = new TextToImgVo();
//组装参数并发送
Map<String, Object> map = new HashMap<>();
map.put("prompt", resultPrompt); //存预设词+用户输入词
map.put("guidance_scale",7.5); //精细度可调
map.put("height",height);
map.put("width",width);
map.put("negative_prompts",negaPrompt);
map.put("sampler","ddim"); //采样风格可调
map.put("seed",0); //种子
map.put("steps",50); //采样步数可调
map.put("style",param.getStyleName()); //风格
map.put("upsample",1);
log.info("请求flagStudio请求体为:【{}】", JSON.toJSONString(map));
// 构建你的请求头
HttpHeaders headers = new HttpHeaders();
headers.set("Content-Type", "application/json");
headers.set("Accept", "application/json");
headers.set("token",token);
// 组合请求头与请求体参数
HttpEntity<String> requestEntity = new HttpEntity<>(JSONObject.toJSONString(map), headers);
JSONObject jsonObject = restTemplate.postForObject(flagStudioConfig.getTxt2ImgUrl(), requestEntity, JSONObject.class);
if(jsonObject != null && jsonObject.getString("data") != null && jsonObject.getString("code").equals("200") && jsonObject.getString("nsfw").equals("0")) {
String imgUrl = jsonObject.getString("data");
List<String> list = new ArrayList<>();
list.add(imgUrl);
img.setImages(list);
String paintId = null;
Date paintTime = null;
for (String image : list) {
//base64转file
MultipartFile file = new Base64ToMultipartFileUtils(image, "data:image/png;base64", "file", "tempSDImg");
//上传图片到七牛云/minio
//String imgStr = remoteFileService.uploadBanner(file).getData();
//上传图片到七牛云并存入sd文件夹
String imgStr = remoteFileService.uploadQiNiu(file,7).getData();
//保存生辰该图片到ai绘画表
AiPaint paint = new AiPaint();
paintId = IdUtil.getSnowflakeNextIdStr();
paintTime = new Date();
paint.setPaintId(paintId);
paint.setCreateTime(paintTime);
paint.setImgUrl(imgStr);
paint.setPrompt(contentPrompt); //只存储用户输入的内容
paint.setNegativePrompt(negaPrompt);
paint.setModel(param.getModelName());
paint.setStyleName(param.getStyleName());
paint.setHeight(param.getHeight() == null ? "512" : String.valueOf(param.getHeight()));
paint.setWidth(param.getWidth() == null ? "512" : String.valueOf(param.getWidth()));
paint.setIsShow("1");
paint.setSource(param.getPlatform());
paint.setPainterId(param.getPainterId());
paint.setPainterName(param.getPainterName());
aiPaintService.insert(paint);
//写入画意值消耗记录
GoldLog goldLog = new GoldLog();
goldLog.setGoldNum(param.getGoldNum());
goldLog.setUserId(param.getPainterId());
goldLog.setSource(param.getPlatform());
goldLog.setReason(GoldEnum.PAINT.getValue());
goldLogService.insert(goldLog);
}
img.setPaintId(paintId);
img.setPaintTime(paintTime);
}
return img;
} catch (Exception e) {
log.error("FlagStudio文生图调用错误!"+e.getMessage());
throw new ServiceException(e.getMessage(),500);
}
}
return null;
}

Write Preview
Loading…
Cancel
Save