From 015112154f47d49660877be7c983d0082b231e98 Mon Sep 17 00:00:00 2001
From: Penny <2500338766@qq.com>
Date: Thu, 27 Apr 2023 21:24:10 +0800
Subject: [PATCH] =?UTF-8?q?feature-img-1.0:=E5=BF=AB=E6=89=8B=E6=B7=BB?=
 =?UTF-8?q?=E5=8A=A0=E6=96=87=E7=94=9F=E5=9B=BE=E7=9B=B8=E5=85=B3=E5=8A=9F?=
 =?UTF-8?q?=E8=83=BD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../img/controller/FhMiniController.java      | 42 +++++++++++++++++--
 1 file changed, 38 insertions(+), 4 deletions(-)

diff --git a/bnyer-services/bnyer-img/src/main/java/com/bnyer/img/controller/FhMiniController.java b/bnyer-services/bnyer-img/src/main/java/com/bnyer/img/controller/FhMiniController.java
index 1a0aa9e..9b7b329 100644
--- a/bnyer-services/bnyer-img/src/main/java/com/bnyer/img/controller/FhMiniController.java
+++ b/bnyer-services/bnyer-img/src/main/java/com/bnyer/img/controller/FhMiniController.java
@@ -12,10 +12,7 @@ import com.bnyer.common.core.web.domain.AjaxResult;
 import com.bnyer.common.core.web.page.TableDataInfo;
 import com.bnyer.img.api.model.LoginFhUser;
 import com.bnyer.img.service.*;
-import com.bnyer.img.vo.CreatorDetailsVo;
-import com.bnyer.img.vo.CreatorFollowVo;
-import com.bnyer.img.vo.SignImgVo;
-import com.bnyer.img.vo.TiktokImgVo;
+import com.bnyer.img.vo.*;
 import com.github.pagehelper.PageHelper;
 import io.swagger.annotations.Api;
 import io.swagger.annotations.ApiOperation;
@@ -88,6 +85,15 @@ public class FhMiniController extends BaseController {
     @Autowired
     private PromptService promptService;
 
+    @Autowired
+    private StableDiffusionService stableDiffusionService;
+
+    @Autowired
+    private AiPaintService aiPaintService;
+
+    @Autowired
+    private PaintCdkService paintCdkService;
+
     @ApiOperation(value="查询banner列表")
     @GetMapping(value = "/listBanner")
     public AjaxResult listBanner(){
@@ -392,4 +398,32 @@ public class FhMiniController extends BaseController {
     public AjaxResult getPrompt(){
         return AjaxResult.success(promptService.queryPromptList());
     }
+
+    @ApiOperation(value="文生图")
+    @PostMapping(value = "/textToImg")
+    public AjaxResult textToImg(@Validated @RequestBody @ApiParam("文生图对象") TextToImgDto param){
+        return AjaxResult.success(stableDiffusionService.textToImg(param));
+    }
+
+    @ApiOperation(value="获取绘画者ai绘画分页")
+    @PostMapping(value = "/getAiPaintPage")
+    public TableDataInfo getAiPaintPage(@Validated @RequestBody @ApiParam("ai绘画对象") AiPaintPageDto dto){
+        PageHelper.startPage(dto.getPageNum(), dto.getPageSize());
+        List<AiPaintVo> aiPaintList = aiPaintService.queryPage(dto);
+        return getDataTable(aiPaintList);
+    }
+
+    @ApiOperation(value="生成cdk")
+    @PostMapping(value = "/generatorCdk/{paintNum}")
+    public AjaxResult generatorCdk(@PathVariable int paintNum){
+        paintCdkService.generatorCdk("2",paintNum);
+        return AjaxResult.success();
+    }
+
+    @ApiOperation(value="使用cdk")
+    @PostMapping(value = "/useCdk")
+    public AjaxResult useCdk(@Validated @RequestBody @ApiParam("ai绘画对象") PaintCdkUseDto dto){
+        paintCdkService.useCdk(dto.getCdk(),dto.getSource(),dto.getUserId(),dto.getAppType());
+        return AjaxResult.success();
+    }
 }