mart-backend/src/main/java/com/pixelai/api/picture/controller/AiPictureController.java

package com.pixelai.api.picture.controller;

import com.base.annotation.SysLog;
import com.base.annotation.Unsecured;
import com.base.helper.Result;
import com.pixelai.api.component.entity.CpAttachment;
import com.pixelai.api.component.entity.CpPhoto;
import com.pixelai.api.component.service.CpAttachmentService;
import com.pixelai.api.component.service.CpPhotoService;
import com.pixelai.api.pa.entity.*;
import com.pixelai.api.pa.service.*;
import com.pixelai.api.picture.config.ExternalResourceConfig;
import com.pixelai.api.picture.service.AiPictureService;
import com.pixelai.api.picture.util.FileSetUtil;
import com.pixelai.api.component.entity.CpArticle;
import com.pixelai.api.picture.vo.AiPictureVo;
import com.pixelai.api.picture.vo.AiViewVo;
import com.pixelai.api.user.service.AcUserService;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiImplicitParam;
import io.swagger.annotations.ApiImplicitParams;
import io.swagger.annotations.ApiOperation;
import org.apache.commons.lang.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.core.io.FileSystemResource;
import org.springframework.http.HttpHeaders;
import org.springframework.http.MediaType;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.multipart.MultipartFile;

import java.io.File;
import java.util.HashMap;
import java.util.Map;

@Api(tags = "AI制图")
@RestController
@RequestMapping("/AiPicture")
public class AiPictureController {
    @Value("${external.resource.dir}")
    private String externalResourceDir;
    @Autowired
    private AiPictureService aiPictureService;
    @Autowired
    private PaVipCurrencyService paVipCurrencyService;
    @Autowired
    private AcUserService acUserService;
    @Autowired
    private PaServicesService paServicesService;
    @Autowired
    private PaCreationService paCreationService;
    @Autowired
    private PaAiTryonService paAiTryonService;
    @Autowired
    private CpPhotoService cpPhotoService;
    @Autowired
    private MbConsumptionService mbConsumptionService;
    @Autowired
    private CpAttachmentService cpAttachmentService;


    @SysLog(action = "AiPicture:GenerateImages", value = "AI制图通用接口")
    @ApiOperation(value = "AI制图通用接口",notes = "AI制图通用接口")
    @Unsecured
    @RequestMapping(value="GenerateImages",method = {RequestMethod.GET})
    public Result<CpAttachment> GenerateImages(AiPictureVo aiPictureVo) throws Exception {
        //判断用户是否存在
        Integer userId=acUserService.getUserId();
        if (userId==null)
            return new Result(false, "用户不存在", "用户不存在");

        //判断服务是否存在
        PaServices paServices=paServicesService.lambdaQuery()
                .eq(PaServices::getId,aiPictureVo.getServiceId())
                .eq(PaServices::getPublish,1)
                .eq(PaServices::getState,"t").one();
        if (paServices==null)
            return new Result(false, "该服务不存在", "该服务不存在");

        //进行扣费并记录日志
        Boolean pay=paVipCurrencyService.consume(userId,paServices);
        if (pay){
            MbConsumption mbConsumption=new MbConsumption(userId,paServices.getName(),"sub",paServices.getPrice(),paServices.consume());
            mbConsumptionService.save(mbConsumption);
        }else{
            return new Result(false, "余额不足", "余额不足");
        }
        //预记录生成图片位置
        CpAttachment cpAttachment=new CpAttachment(userId,"","https://guojunjie.oss-cn-hangzhou.aliyuncs.com/%E5%8A%A0%E8%BD%BD%E7%AD%89%E5%BE%85_%E7%88%B1%E7%BB%99%E7%BD%91_aigei_com.gif",aiPictureVo.getServiceId());
        cpAttachmentService.save(cpAttachment);
        //预保存用户作品
        PaCreation paCreation=new PaCreation(userId,"",cpAttachment.getId(),aiPictureVo.getServiceId(),"create");
        paCreation.setManufacturers(paServices.getManufacturers());
        paCreation.setParm(aiPictureVo.toString());
        paCreationService.save(paCreation);
        //进行图片生成服务
        String serviceApi=paServices.getApi();
        switch(serviceApi){
            //百度：黑白图片上色
            case "ImageColoring" :
                aiPictureService.ImageColoring(aiPictureVo,paCreation);
                break;
            //百度：人物动漫化
            case "CharacterAnime" :
                aiPictureService.CharacterAnime(aiPictureVo,paCreation);
                break;
            //百度：更换图片风格
            case "styleTrans" :
                aiPictureService.styleTrans(aiPictureVo,paCreation);
                break;
            //阿里云：文字生成图片
            case "textToImage" :
                aiPictureService.textToImage(aiPictureVo,paCreation);
                break;
            //阿里云：AI换衣
            case "aiTryon" :
                if(!(StringUtils.isNotEmpty(aiPictureVo.getImageUrl()) && StringUtils.isNotEmpty(aiPictureVo.getTopImageUrl()))){
                    return new Result(false, "","人物形象和上衣必须选择");
                }
//                //判断是否为预览模式，不走阿里云，读取服务器资源
//                if(aiPictureVo.getImageUrl().contains("upload/picture") &&
//                   aiPictureVo.getTopImageUrl().contains("upload/picture") &&
//                   (StringUtils.isEmpty(aiPictureVo.getBottomImageUrl()) || aiPictureVo.getBottomImageUrl().contains("upload/picture") ) )
//                {
//                    aiPictureService.aiTryonPreview(aiPictureVo,paCreation);
//                    break;
//                }
                //换衣自定义，走阿里云
                aiPictureService.aiTryon(aiPictureVo,paCreation);
                break;
            //阿里云：人像风格重绘
            case "portraitStyleRepainting" :
                aiPictureService.portraitStyleRepainting(aiPictureVo,paCreation);
                break;
            //阿里云：艺术字
            case "texture" :
                aiPictureService.texture(aiPictureVo,paCreation);
                break;
            //火山引擎：文字生成图片
            case "textToImageByVolcengine" :
                aiPictureService.textToImageByVolcengine(aiPictureVo,paCreation);
                break;
            //火山引擎：表情编辑
            case "emotionPortrait" :
                aiPictureService.emotionPortrait(aiPictureVo,paCreation);
                break;
            //火山引擎：发型编辑
            case "hairStyle" :
                aiPictureService.hairStyle(aiPictureVo,paCreation);
                break;
            //火山引擎：年龄变换
            case "ageChange" :
                aiPictureService.ageChange(aiPictureVo,paCreation);
                break;
////            //火山引擎：3D游戏特效
//            case "threeDGameCartoon" :
//                aiPictureService.threeDGameCartoon(aiPictureVo,paCreation);
//                break;
            //火山引擎：人像漫画风
            case "jPCartoon" :
                aiPictureService.jPCartoon(aiPictureVo,paCreation);
                break;
            //火山引擎：单张pv写真
            case "HighAesSmartDrawing" :
                aiPictureService.HighAesSmartDrawing(aiPictureVo,paCreation);
                break;
            //火山引擎：人像抠图
            case "HumanSegmentDemo" :
                aiPictureService.HumanSegmentDemo(aiPictureVo,paCreation);
                break;

        }

        return new Result(true, "调用成功","",cpAttachment);
    }



//    @SysLog(action = "AiPicture:styleTrans", value = "ai图像增强：更换图片风格")
//    @ApiOperation(value = "更换图片风格",notes = "ai图像增强")
//    @RequestMapping(value="styleTrans",method = {RequestMethod.POST})
//    @ApiImplicitParams({
//            @ApiImplicitParam(name = "userid", value = "id", paramType = "query"),//用户id，用于生成base64的txt文件，用以转储
//            @ApiImplicitParam(name = "image", value = "图片文件，必须是图片才行，pngjpg皆可，后缀名不限", paramType = "query"),
//            @ApiImplicitParam(name = "options", value = "图片转换的风格,1:卡通画,2:铅笔风格,3:彩色铅笔画,4:彩色糖块油画,5:神奈川冲浪里油画,6:薰衣草油画,7:奇异油画,8:呐喊油画,9:哥特油画。", paramType = "query")})
//    public ResponseEntity<FileSystemResource> styleTrans(@RequestParam("userid")Integer id, @RequestParam("image") MultipartFile image, @RequestParam("options") Integer options) throws Exception {
//        String fileName= FileSetUtil.resetFileName(id,externalResourceDir,image);//生成的filename无后缀
//        File file=FileSetUtil.saveFile(image,externalResourceDir,fileName);//生成的file为png文件
//        System.out.println(fileName+"//////////"+externalResourceDir);
//        String filePath=aiPictureService.styleTrans(fileName,externalResourceDir,options);//转码传输到baiduapi后写入fileName+.txt,所返回的filePath直接指向对应txt文件
//        aiPictureService.Base64ToImage(fileName,filePath,externalResourceDir);
//        HttpHeaders headers = new HttpHeaders();
//        headers.add(HttpHeaders.CONTENT_DISPOSITION, "attachment; filename=" + file.getName());
//        return ResponseEntity.ok()
//                .headers(headers)
//                .contentLength(file.length())
//                .contentType(MediaType.APPLICATION_OCTET_STREAM)
//                .body(new FileSystemResource(file));
//    }


}