46 lines
1.7 KiB
TypeScript
46 lines
1.7 KiB
TypeScript
import type { ReferenceImage } from "@/types/image";
|
||
|
||
import type { RequestOptions } from "./video-contracts";
|
||
|
||
export type VideoImageRole = "first_frame" | "last_frame" | "reference_image";
|
||
|
||
export type ResolvedVideoImageReference = {
|
||
image: ReferenceImage;
|
||
role: VideoImageRole;
|
||
};
|
||
|
||
type VideoReferenceContext = {
|
||
videoCount?: number;
|
||
audioCount?: number;
|
||
};
|
||
|
||
export function resolveVideoImageReferences(images: ReferenceImage[], options?: RequestOptions, context: VideoReferenceContext = {}): ResolvedVideoImageReference[] {
|
||
const operation = options?.videoEditOperation?.trim();
|
||
if (operation === "reference_to_video") {
|
||
return images.map((image) => ({ image, role: "reference_image" }));
|
||
}
|
||
|
||
const startFrameId = options?.videoStartFrameNodeId?.trim();
|
||
const endFrameId = options?.videoEndFrameNodeId?.trim();
|
||
if (startFrameId || endFrameId) {
|
||
return images.map((image) => ({
|
||
image,
|
||
role: image.id === startFrameId ? "first_frame" : image.id === endFrameId ? "last_frame" : "reference_image",
|
||
}));
|
||
}
|
||
|
||
if (context.videoCount || context.audioCount) {
|
||
return images.map((image) => ({ image, role: "reference_image" }));
|
||
}
|
||
|
||
// 传统创作页没有帧节点 ID：保留单图首帧、双图首尾帧的历史语义。
|
||
return images.map((image, index) => ({
|
||
image,
|
||
role: index === 0 ? "first_frame" : index === 1 && images.length === 2 ? "last_frame" : "reference_image",
|
||
}));
|
||
}
|
||
|
||
export function hasExplicitVideoFrames(options?: RequestOptions) {
|
||
return options?.videoEditOperation !== "reference_to_video" && Boolean(options?.videoStartFrameNodeId || options?.videoEndFrameNodeId);
|
||
}
|