|
|
@@ -90,8 +90,12 @@ public class AiVideoServiceImpl implements AiVideoService {
|
|
|
AiModelDO model = modelService.validateModel(drawReqVO.getModelId());
|
|
|
|
|
|
// 2. 保存数据库
|
|
|
+ String promptImage = null;
|
|
|
+ if (CollUtil.isNotEmpty(drawReqVO.getPromptImage())) {
|
|
|
+ promptImage = StrUtil.join(",", drawReqVO.getPromptImage());
|
|
|
+ }
|
|
|
AiVideoDO video = BeanUtils.toBean(drawReqVO, AiVideoDO.class).setUserId(userId)
|
|
|
- .setPlatform(model.getPlatform()).setPromptImage(drawReqVO.getPromptImage())
|
|
|
+ .setPlatform(model.getPlatform()).setPromptImage(promptImage)
|
|
|
.setModelId(model.getId()).setModel(model.getModel())
|
|
|
.setPublicStatus(false).setStatus(AiVideoStatusEnum.IN_PROGRESS.getStatus());
|
|
|
videoMapper.insert(video);
|
|
|
@@ -123,7 +127,7 @@ public class AiVideoServiceImpl implements AiVideoService {
|
|
|
//文本提示词
|
|
|
String promptText = reqVO.getPrompt();
|
|
|
// 第一帧图片
|
|
|
- String promptImage = reqVO.getPromptImage();
|
|
|
+ List<String> promptImage = reqVO.getPromptImage();
|
|
|
//视频分辨率
|
|
|
String resolution = reqVO.getResolution();
|
|
|
//视频时长
|
|
|
@@ -141,13 +145,15 @@ public class AiVideoServiceImpl implements AiVideoService {
|
|
|
.build());
|
|
|
|
|
|
// 首帧图片
|
|
|
- if (StrUtil.isNotBlank(promptImage)) {
|
|
|
- contents.add(Content.builder()
|
|
|
- .type("image_url")
|
|
|
- .imageUrl(CreateContentGenerationTaskRequest.ImageUrl.builder()
|
|
|
- .url(promptImage) // 请上传可以访问的图片URL
|
|
|
- .build())
|
|
|
- .build());
|
|
|
+ if (CollUtil.isNotEmpty(promptImage)) {
|
|
|
+ promptImage.forEach(url -> {
|
|
|
+ contents.add(Content.builder()
|
|
|
+ .type("image_url")
|
|
|
+ .imageUrl(CreateContentGenerationTaskRequest.ImageUrl.builder()
|
|
|
+ .url(url) // 请上传可以访问的图片URL
|
|
|
+ .build())
|
|
|
+ .build());
|
|
|
+ });
|
|
|
}
|
|
|
|
|
|
// 创建视频生成任务
|