aida_back/src/main/java/com/ai/da/controller/GenerateController.java

package com.ai.da.controller;

import com.ai.da.common.response.Response;
import com.ai.da.model.dto.*;
import com.ai.da.model.vo.*;
import com.ai.da.service.GenerateService;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import io.swagger.annotations.ApiParam;
import lombok.extern.slf4j.Slf4j;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

import javax.annotation.Resource;
import javax.validation.Valid;
import java.util.List;
/**
 * @author XP
 */
@Api(tags = "Generate模块")
@Slf4j
@RestController
@RequestMapping("/api/generate")
public class GenerateController {

    @Resource
    private GenerateService generateService;

     /*@ApiOperation("自动识别sketch的caption  暂时未上")
    @PostMapping("/caption")
    public Response<GenerateCaptionVO> generateCaption(@RequestParam Long sketchElementId) {
        return Response.success(generateService.generateCaption(sketchElementId));
    }*/

    /*@ApiOperation("通过文字、图片生成图片")
    @PostMapping("/sketchAndPrint")
    public void generateThroughImageText(@Valid @RequestBody GenerateThroughImageTextDTO generateThroughImageTextDTO) {
//        return Response.success(generateService.generateThroughImageText(generateThroughImageTextDTO));
        generateService.generateThroughImageText(generateThroughImageTextDTO);
    }*/

    @ApiOperation("喜欢生成的图片")
    @PostMapping("/like")
    public Response<GenerateLikeVO> like(@Valid @RequestBody GenerateLikeDTO generateLikeDTO) {
        return Response.success(generateService.generateLike(generateLikeDTO));
    }

    @ApiOperation(value = "取消喜欢")
    @GetMapping("/dislike")
    public Response<Boolean> dislike(@ApiParam("generateDetailId") @RequestParam Long generateDetailId,
                                     @ApiParam("timeZone") @RequestParam String timeZone) {
        return Response.success(generateService.generateDislike(generateDetailId, timeZone));
    }

    @ApiOperation(value = "发起生成请求，异步获取结果")
    @PostMapping("/prepare")
    public Response<PrepareForGenerateVO> prepareForGenerate(@Valid @RequestBody GenerateThroughImageTextDTO generateThroughImageTextDTO) {
        return Response.success(generateService.prepareForGenerate(generateThroughImageTextDTO));
    }

    @ApiOperation(value = "取消继续生成")
    @GetMapping("/stopWaiting")
    public Response<String> stopWaiting(@RequestParam("userId") Long userId,
                                        @RequestParam("uniqueId") List<String> uniqueId,
                                        @RequestParam("timeZone") String timeZone,
                                        @RequestParam("type") String type) {
        generateService.cancelGenerate(userId, uniqueId, timeZone, type);
        return Response.success("stop waiting successfully");
    }

    /*@ApiOperation(value = "获取生成结果")
    @GetMapping("/result")
    public Response<GenerateCollectionVO> getGenerateResult(@RequestParam("uniqueId") String uniqueId) {
        GenerateCollectionVO generateResult = generateService.getGenerateResult(uniqueId);
        return Response.success(generateResult);
    }*/

    @ApiOperation(value = "获取生成结果")
    @PostMapping("/result")
    public Response<List<GenerateResultVO>> getGenerateResults(@Valid @RequestBody List<String> taskIdList) {
        List<GenerateResultVO> generateResult = generateService.getGenerateResultList(taskIdList);
        return Response.success(generateResult);
    }

    @ApiOperation(value = "imageToSketch")
    @PostMapping("/imageToSketch")
    public Response<GenerateResultVO> imageToSketch(@Valid @RequestBody ImageToSketchDTO imageToSketchDTO) {
        return Response.success(generateService.imageToSketch(imageToSketchDTO, null, null));
    }

    // modifySketch
    @ApiOperation(value = "modifySketch")
    @PostMapping("/modifySketch")
    public Response<GenerateResultVO> modifySketch(@Valid @RequestBody GenerateModifyDTO generateModifyDTO) {
        return Response.success(generateService.modifySketch(generateModifyDTO));
    }

    @ApiOperation(value = "请求进行姿势变换")
    @GetMapping("/poseTransform")
    public Response<String> poseTransform(@ApiParam("projectId") @RequestParam Long projectId,
                                          @ApiParam("productImage") @RequestParam String productImage,
                                          @ApiParam("poseId") @RequestParam int poseId) {
        return Response.success(generateService.poseTransform(projectId, productImage, poseId));
    }

    @ApiOperation(value = "获取姿势变换生成结果")
    @GetMapping("/poseTransformResult")
    public Response<PoseTransformationVO> getPoseTransformationResults(@ApiParam("taskId") @RequestParam String taskId) {
        PoseTransformationVO generateResult = generateService.getPoseTransformationResult(taskId);
        return Response.success(generateResult);
    }

    @ApiOperation("喜欢或取消喜欢姿势变换生成的图片")
    @PostMapping("/likeOrDislike")
    public Response<Boolean> likeOrDislike(@ApiParam("id") @RequestParam Long transformedId, @ApiParam("like || dislike") @RequestParam String likeOrDislike) {
        return Response.success(generateService.disOrLikePose(transformedId, likeOrDislike));
    }

    @ApiOperation(value = "修改模特比例")
    @PostMapping("/modifyProportion")
    public Response<String> modifyModelProportion(@Valid @RequestBody ModifyModelProportionDTO proportionDTO){
        String path = generateService.modifyModelProportion(proportionDTO);
        return Response.success(path);
    }

    @ApiOperation(value = "拼贴图生成线稿")
    @PostMapping("/genSketchRecon")
    public Response<GenerateResultVO> sketchReconstructionGenerate(@Valid @RequestBody SketchReconstructionDTO sketchReconstructionDTO){
        GenerateResultVO generateResultVO = generateService.sketchReconstructionGenerate(sketchReconstructionDTO);
        return Response.success(generateResultVO);
    }

    @ApiOperation(value = "获取拼贴图最后一次生成结果")
    @GetMapping("/getReconLastResult")
    public Response<SketchReconstructionVO> getSketchReconstruction(@RequestParam("projectId") Long projectId){
        SketchReconstructionVO sketchReconstruction = generateService.getSketchReconstruction(projectId);
        return Response.success(sketchReconstruction);
    }




}