102 lines
4.2 KiB
Java
102 lines
4.2 KiB
Java
package com.ai.da.controller;
|
|
|
|
import com.ai.da.common.response.Response;
|
|
import com.ai.da.model.dto.GenerateLikeDTO;
|
|
import com.ai.da.model.dto.GenerateModifyDTO;
|
|
import com.ai.da.model.dto.GenerateThroughImageTextDTO;
|
|
import com.ai.da.model.dto.ImageToSketchDTO;
|
|
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 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));
|
|
}
|
|
|
|
// modifySketch
|
|
@ApiOperation(value = "modifySketch")
|
|
@PostMapping("/modifySketch")
|
|
public Response<GenerateResultVO> modifySketch(@Valid @RequestBody GenerateModifyDTO generateModifyDTO) {
|
|
return Response.success(generateService.modifySketch(generateModifyDTO));
|
|
}
|
|
|
|
}
|