From 48607529f84911d76d652e5544034915413c092b Mon Sep 17 00:00:00 2001
From: Penny <2500338766@qq.com>
Date: Wed, 10 May 2023 23:01:29 +0800
Subject: [PATCH] =?UTF-8?q?feature:=E6=96=B0=E5=A2=9E=E5=88=9B=E6=84=8F?=
 =?UTF-8?q?=E5=B7=A5=E5=9D=8A=E8=AF=A6=E6=83=85=E6=8E=A5=E5=8F=A3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../img/controller/TiktokMiniController.java  |  6 ++++++
 .../main/java/com/bnyer/img/vo/AiPaintVo.java | 21 +++++++++++++++++++
 .../com/bnyer/img/mapper/AiPaintMapper.xml    | 15 ++++++++++---
 3 files changed, 39 insertions(+), 3 deletions(-)

diff --git a/bnyer-services/bnyer-img/src/main/java/com/bnyer/img/controller/TiktokMiniController.java b/bnyer-services/bnyer-img/src/main/java/com/bnyer/img/controller/TiktokMiniController.java
index 491f5b2..308162f 100644
--- a/bnyer-services/bnyer-img/src/main/java/com/bnyer/img/controller/TiktokMiniController.java
+++ b/bnyer-services/bnyer-img/src/main/java/com/bnyer/img/controller/TiktokMiniController.java
@@ -444,6 +444,12 @@ public class TiktokMiniController extends BaseController {
         return getDataTable(aiPaintList);
     }
 
+    @ApiOperation(value="获取绘画者ai绘画详情")
+    @GetMapping(value = "/getAiPaintDetials/{id}")
+    public AjaxResult getAiPaintDetials(@PathVariable @ApiParam("ai绘画对象id") Long id){
+        return AjaxResult.success(aiPaintService.queryDetails(id));
+    }
+
     @ApiOperation(value="生成cdk")
     @PostMapping(value = "/generatorCdk/{paintNum}")
     public AjaxResult generatorCdk(@PathVariable int paintNum){
diff --git a/bnyer-services/bnyer-img/src/main/java/com/bnyer/img/vo/AiPaintVo.java b/bnyer-services/bnyer-img/src/main/java/com/bnyer/img/vo/AiPaintVo.java
index 0262bff..412cdf1 100644
--- a/bnyer-services/bnyer-img/src/main/java/com/bnyer/img/vo/AiPaintVo.java
+++ b/bnyer-services/bnyer-img/src/main/java/com/bnyer/img/vo/AiPaintVo.java
@@ -32,6 +32,27 @@ public class AiPaintVo implements Serializable {
     @ApiModelProperty(value="关键词")
     private String prompt;
 
+    @ApiModelProperty(value="反向提示词")
+    private String negativePrompt;
+
+    @ApiModelProperty(value="采样步数")
+    private Integer steps;
+
+    @ApiModelProperty(value="种子值")
+    private Integer seed;
+
+    @ApiModelProperty(value="精细度")
+    private Double cfgScale;
+
+    @ApiModelProperty(value="批量数量")
+    private Integer batchSize;
+
+    @ApiModelProperty(value="eta值")
+    private Double eta;
+
+    @ApiModelProperty(value="采样器")
+    private String samplerIndex;
+
     @ApiModelProperty(value="风格名称")
     private String styleName;
 
diff --git a/bnyer-services/bnyer-img/src/main/resources/com/bnyer/img/mapper/AiPaintMapper.xml b/bnyer-services/bnyer-img/src/main/resources/com/bnyer/img/mapper/AiPaintMapper.xml
index e5011ba..04b003e 100644
--- a/bnyer-services/bnyer-img/src/main/resources/com/bnyer/img/mapper/AiPaintMapper.xml
+++ b/bnyer-services/bnyer-img/src/main/resources/com/bnyer/img/mapper/AiPaintMapper.xml
@@ -17,16 +17,24 @@
     <result column="is_show" jdbcType="CHAR" property="isShow" />
     <result column="source" jdbcType="CHAR" property="source" />
     <result column="create_time" jdbcType="TIMESTAMP" property="createTime" />
+    <result column="negative_prompt" jdbcType="VARCHAR" property="negativePrompt" />
+    <result column="steps" jdbcType="INTEGER" property="steps" />
+    <result column="seed" jdbcType="INTEGER" property="seed" />
+    <result column="batch_size" jdbcType="INTEGER" property="batchSize" />
+    <result column="cfg_scale" jdbcType="DOUBLE" property="cfgScale" />
+    <result column="eta" jdbcType="DOUBLE" property="eta" />
+    <result column="sampler_index" jdbcType="VARCHAR" property="samplerIndex" />
   </resultMap>
   <sql id="Base_Column_List">
     <!--@mbg.generated-->
     id, paint_id, painter_id, painter_name, img_url, prompt, model,style_name, height, width, is_show,
-    `source`, create_time
+    `source`, create_time,negative_prompt,steps,seed,batch_size,cfg_scale,eta,sampler_index
   </sql>
 
   <select id="queryPage" resultType="com.bnyer.img.vo.AiPaintVo">
     select
-        id, paint_id,painter_name,img_url, prompt, style_name,height, width,create_time,source
+        id, paint_id,painter_name,img_url, prompt, style_name,height, width,create_time,source,
+        negative_prompt,steps,seed,batch_size,cfg_scale,eta,sampler_index
     from img_ai_paint
     where is_show = '1' and source = #{source} and painter_id = #{painterId}
     order by create_time desc
@@ -34,7 +42,8 @@
 
   <select id="queryDetails" resultType="com.bnyer.img.vo.AiPaintVo">
     select
-        id, paint_id,painter_name,img_url, prompt, style_name,height, width,create_time,source
+        id, paint_id,painter_name,img_url, prompt, style_name,height, width,create_time,source,
+        negative_prompt,steps,seed,batch_size,cfg_scale,eta,sampler_index
     from img_ai_paint where id = #{id}
   </select>
 </mapper>
\ No newline at end of file