From 72c864bb82c8fec9d607927806d9cd7c072574b8 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E2=80=98=E2=80=98?= <kun.cheng@funi.com>
Date: Thu, 11 May 2023 21:23:26 +0800
Subject: [PATCH] =?UTF-8?q?feature:=E6=83=85=E4=BE=A3=E5=A4=B4=E5=83=8F?=
 =?UTF-8?q?=E6=A8=A1=E5=BC=8F=E6=B7=BB=E5=8A=A0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../common/core/dto/TextToImgLoveDto.java     | 37 +++++++++++++++++++
 .../img/service/StableDiffusionService.java   |  8 ++++
 .../impl/StableDiffusionServiceImpl.java      |  7 +++-
 3 files changed, 51 insertions(+), 1 deletion(-)
 create mode 100644 bnyer-common/bnyer-common-core/src/main/java/com/bnyer/common/core/dto/TextToImgLoveDto.java

diff --git a/bnyer-common/bnyer-common-core/src/main/java/com/bnyer/common/core/dto/TextToImgLoveDto.java b/bnyer-common/bnyer-common-core/src/main/java/com/bnyer/common/core/dto/TextToImgLoveDto.java
new file mode 100644
index 0000000..db318ba
--- /dev/null
+++ b/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;
+}
diff --git a/bnyer-services/bnyer-img/src/main/java/com/bnyer/img/service/StableDiffusionService.java b/bnyer-services/bnyer-img/src/main/java/com/bnyer/img/service/StableDiffusionService.java
index d8d9025..85e6a6a 100644
--- a/bnyer-services/bnyer-img/src/main/java/com/bnyer/img/service/StableDiffusionService.java
+++ b/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();
 
     /**
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 b4c2383..0913795 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
@@ -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 应用