package com.ai.da.service;

import com.ai.da.mapper.primary.entity.Generate;
import com.ai.da.mapper.primary.entity.GenerateDetail;
import com.ai.da.model.dto.*;
import com.ai.da.model.vo.*;
import com.baomidou.mybatisplus.extension.service.IService;
import org.springframework.web.multipart.MultipartFile;

import java.util.List;
import java.util.Map;

public interface GenerateService extends IService<Generate> {

    GenerateCaptionVO generateCaption(Long sketchElementId);

    void generateThroughImageText(GenerateThroughImageTextDTO generateThroughImageTextDTO);

    void processGenerateResult(String taskId, String url, String category);

    void processToProductImageResult(String taskId, String url, String category);

    GenerateLikeVO generateLike(GenerateLikeDTO generateLikeDTO);

    Boolean generateDislike(Long generateDetailId, String timeZone);

    void updateLikeStatusBatch(List<Long> generateDetailIdList, Byte hasLike, Long libraryId, String timeZone);

    List<GenerateDetail> selectBatchByLibraryId(List<Long> libraryId);

//    GenerateCollectionVO getGenerateResult(String uniqueId);

    List<GenerateResultVO> getGenerateResultList(List<String> taskIdList);

    PrepareForGenerateVO prepareForGenerate(GenerateThroughImageTextDTO generateThroughImageTextDTO);

    Long getRankPosition(String uniqueId);

    void cancelGenerate(Long userId, List<String> uniqueId, String timeZone, String type);

    void processRelightResult(String taskId, String url, String category);

    List<Map<String, Object>> getCountByUserAndTime(String startTime, String endTime, List<Long> accountIdList);

    GenerateResultVO imageToSketch(ImageToSketchDTO imageToSketchDTO, String collagePictureUrl, Long projectId);

    GenerateResultVO modifySketch(GenerateModifyDTO generateModifyDTO);

    String poseTransform(Long projectId, String productImage, int poseId);

    void processPoseTransformResult(String taskId, String gifUrl, String videoUrl, String imageUrl);

    PoseTransformationVO getPoseTransformationResult(String taskId);

    List<PoseTransformationVO> getPoseTransformationResultList(Long projectId);

    Boolean disOrLikePose(Long transformedId, String likeOrDislike);

    String modifyModelProportion(ModifyModelProportionDTO proportionDTO);

    GenerateResultVO sketchReconstructionGenerate(SketchReconstructionDTO sketchReconstructionDTO);

    SketchReconstructionVO getSketchReconstruction(Long projectId);

    List<Map<String, String>> getAllPose();

    void processPoseTransformResultBatch(String taskId, String gifUrl, String videoUrl, String imageUrl, String progress);
}