feature:情侣头像模式添加

bnyer-common/bnyer-common-core/src/main/java/com/bnyer/common/core/dto/TextToImgLoveDto.java

@@ -0,0 +1,37 @@
+package com.bnyer.common.core.dto;
+
+import io.swagger.annotations.ApiModel;
+import io.swagger.annotations.ApiModelProperty;
+import lombok.Getter;
+import lombok.Setter;
+
+import java.io.Serializable;
+
+
+@Getter
+@Setter
+@ApiModel("情侣头像接收类")
+public class TextToImgLoveDto implements Serializable {
+
+
+    @ApiModelProperty(value="内容提示词")
+    private String prompt;
+
+    @ApiModelProperty(value="场景提示词")
+    private String scenePrompt;
+
+    @ApiModelProperty(value="风格名称")
+    private String styleName;
+
+    @ApiModelProperty(value="绘图者id")
+    private Long painterId;
+
+    @ApiModelProperty(value="绘图者昵称")
+    private String painterName;
+
+    @ApiModelProperty(value="应用")
+    private String appType;
+
+    @ApiModelProperty(value="平台")
+    private String platform;
+}

bnyer-services/bnyer-img/src/main/java/com/bnyer/img/service/StableDiffusionService.java

@@ -1,6 +1,7 @@
 package com.bnyer.img.service;
 
 import com.bnyer.common.core.dto.TextToImgDto;
+import com.bnyer.common.core.dto.TextToImgLoveDto;
 import com.bnyer.common.core.vo.TextToImgVo;
 
 public interface StableDiffusionService {
@@ -13,6 +14,13 @@ public interface StableDiffusionService {
      */
     TextToImgVo textToImg(TextToImgDto param);
 
+    /**
+     * 情侣头像
+     * @param param
+     * @return
+     */
+    TextToImgVo textToImgLove(TextToImgLoveDto param);
+
     void imgToImg();
 
     /**

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

@@ -6,6 +6,7 @@ import com.alibaba.fastjson.JSONObject;
 import com.bnyer.common.core.constant.RedisKeyConstant;
 import com.bnyer.common.core.domain.AiPaint;
 import com.bnyer.common.core.dto.TextToImgDto;
+import com.bnyer.common.core.dto.TextToImgLoveDto;
 import com.bnyer.common.core.exception.ServiceException;
 import com.bnyer.common.core.utils.StringUtils;
 import com.bnyer.common.core.utils.TranslateUtils;
@@ -89,7 +90,6 @@ public class StableDiffusionServiceImpl implements StableDiffusionService {
 
     @Override
     public TextToImgVo textToImg(TextToImgDto param) {
-
         //配置控制采用sd还是tx文生图
         int button = redisService.getCacheObject(RedisKeyConstant.AI_PAINT_BUTTON);
         if(button == AiPaintButtonEnum.TX.getCode()){
@@ -286,6 +286,11 @@ public class StableDiffusionServiceImpl implements StableDiffusionService {
         return null;
     }
 
+    @Override
+    public TextToImgVo textToImgLove(TextToImgLoveDto param) {
+        return null;
+    }
+
     /**
      * 写入ai绘画次数
      * @param appType 应用