|
|
|
@ -268,20 +268,32 @@ public class PortalVideoController extends BaseController {
|
|
|
|
public AjaxResult imageReference(@RequestBody PortalVideoGenRequest request) {
|
|
|
|
public AjaxResult imageReference(@RequestBody PortalVideoGenRequest request) {
|
|
|
|
List<ContentItem> contentList;
|
|
|
|
List<ContentItem> contentList;
|
|
|
|
if (request.getContent() != null && !request.getContent().isEmpty()) {
|
|
|
|
if (request.getContent() != null && !request.getContent().isEmpty()) {
|
|
|
|
contentList = request.getContent();
|
|
|
|
contentList = new ArrayList<>(request.getContent());
|
|
|
|
ContentItem head = contentList.get(0);
|
|
|
|
ContentItem head = contentList.get(0);
|
|
|
|
if (head == null || !"text".equals(head.getType()) || StringUtils.isEmpty(head.getText())) {
|
|
|
|
if (head == null || !"text".equals(head.getType()) || StringUtils.isEmpty(head.getText())) {
|
|
|
|
return AjaxResult.error("请输入视频描述文本(首条须为 type=text,可含 [图n] 占位)");
|
|
|
|
return AjaxResult.error("请输入视频描述文本(首条须为 type=text,可含 [图片n]/[视频n]/[音频n] 占位)");
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// 保留 text + 合法 reference_*(图/音/视频);允许“只有 text 没有参考素材”
|
|
|
|
|
|
|
|
List<ContentItem> filtered = new ArrayList<>();
|
|
|
|
|
|
|
|
filtered.add(head);
|
|
|
|
|
|
|
|
for (int i = 1; i < contentList.size(); i++) {
|
|
|
|
|
|
|
|
ContentItem it = contentList.get(i);
|
|
|
|
|
|
|
|
if (isReferenceImageContentItem(it)
|
|
|
|
|
|
|
|
|| isReferenceAudioContentItem(it)
|
|
|
|
|
|
|
|
|| isReferenceVideoContentItem(it)) {
|
|
|
|
|
|
|
|
filtered.add(it);
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
contentList = filtered;
|
|
|
|
|
|
|
|
|
|
|
|
String firstRef = contentList.stream()
|
|
|
|
String firstRef = contentList.stream()
|
|
|
|
.filter(PortalVideoController::isReferenceImageContentItem)
|
|
|
|
.skip(1)
|
|
|
|
.map(it -> it.getImageUrl().getUrl())
|
|
|
|
.map(PortalVideoController::firstReferenceUrlFromItem)
|
|
|
|
|
|
|
|
.filter(StringUtils::isNotEmpty)
|
|
|
|
.findFirst()
|
|
|
|
.findFirst()
|
|
|
|
.orElse(null);
|
|
|
|
.orElse(null);
|
|
|
|
if (StringUtils.isEmpty(firstRef)) {
|
|
|
|
// 无参考图也允许,仅 text 提示词
|
|
|
|
return AjaxResult.error("请上传参考图(content 中须含 image_url + role=reference_image)");
|
|
|
|
if (StringUtils.isNotEmpty(firstRef) && StringUtils.isEmpty(request.getReferenceUrl())) {
|
|
|
|
}
|
|
|
|
|
|
|
|
if (StringUtils.isEmpty(request.getReferenceUrl())) {
|
|
|
|
|
|
|
|
request.setReferenceUrl(firstRef);
|
|
|
|
request.setReferenceUrl(firstRef);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
} else {
|
|
|
|
@ -310,6 +322,27 @@ public class PortalVideoController extends BaseController {
|
|
|
|
return submitOrderAndCreate(request, "image-reference", body);
|
|
|
|
return submitOrderAndCreate(request, "image-reference", body);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
private static String firstReferenceUrlFromItem(ContentItem item) {
|
|
|
|
|
|
|
|
if (isReferenceImageContentItem(item)) {
|
|
|
|
|
|
|
|
return item.getImageUrl().getUrl();
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
if (isReferenceVideoContentItem(item)) {
|
|
|
|
|
|
|
|
return item.getVideoUrl().getUrl();
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
if (isReferenceAudioContentItem(item)) {
|
|
|
|
|
|
|
|
return item.getAudioUrl().getUrl();
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
return null;
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
private static boolean isValidReferenceAssetOrHttpUrl(String raw) {
|
|
|
|
|
|
|
|
if (StringUtils.isEmpty(raw)) {
|
|
|
|
|
|
|
|
return false;
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
String url = raw.trim().toLowerCase();
|
|
|
|
|
|
|
|
return url.startsWith("http://") || url.startsWith("https://") || url.startsWith("asset://");
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
private static boolean isReferenceImageContentItem(ContentItem item) {
|
|
|
|
private static boolean isReferenceImageContentItem(ContentItem item) {
|
|
|
|
if (item == null || !"image_url".equals(item.getType())) {
|
|
|
|
if (item == null || !"image_url".equals(item.getType())) {
|
|
|
|
return false;
|
|
|
|
return false;
|
|
|
|
@ -318,7 +351,38 @@ public class PortalVideoController extends BaseController {
|
|
|
|
return false;
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
ImageUrl iu = item.getImageUrl();
|
|
|
|
ImageUrl iu = item.getImageUrl();
|
|
|
|
return iu != null && StringUtils.isNotEmpty(iu.getUrl());
|
|
|
|
if (iu == null || StringUtils.isEmpty(iu.getUrl())) {
|
|
|
|
|
|
|
|
return false;
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
return isValidReferenceAssetOrHttpUrl(iu.getUrl());
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
private static boolean isReferenceVideoContentItem(ContentItem item) {
|
|
|
|
|
|
|
|
if (item == null || !"video_url".equals(item.getType())) {
|
|
|
|
|
|
|
|
return false;
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
if (!"reference_video".equals(item.getRole())) {
|
|
|
|
|
|
|
|
return false;
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
ImageUrl vu = item.getVideoUrl();
|
|
|
|
|
|
|
|
if (vu == null || StringUtils.isEmpty(vu.getUrl())) {
|
|
|
|
|
|
|
|
return false;
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
return isValidReferenceAssetOrHttpUrl(vu.getUrl());
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
private static boolean isReferenceAudioContentItem(ContentItem item) {
|
|
|
|
|
|
|
|
if (item == null || !"audio_url".equals(item.getType())) {
|
|
|
|
|
|
|
|
return false;
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
if (!"reference_audio".equals(item.getRole())) {
|
|
|
|
|
|
|
|
return false;
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
ImageUrl au = item.getAudioUrl();
|
|
|
|
|
|
|
|
if (au == null || StringUtils.isEmpty(au.getUrl())) {
|
|
|
|
|
|
|
|
return false;
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
return isValidReferenceAssetOrHttpUrl(au.getUrl());
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
private List<ContentItem> buildTextAndFirstFrame(String text, String firstUrl) {
|
|
|
|
private List<ContentItem> buildTextAndFirstFrame(String text, String firstUrl) {
|
|
|
|
|