package com.ai.da.controller;

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

import javax.annotation.Resource;
import javax.validation.Valid;

/**
 * @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("/sketch")
    public Response<GenerateCollectionVO> generateThroughImageText(@Valid @RequestBody GenerateThroughImageTextDTO generateThroughImageTextDTO){
        return Response.success(generateService.generateSketchThroughImageText(generateThroughImageTextDTO));
    }







}