weitao
/
popiart-node-canvas
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
Browse Source

feat(09-01): add video history types and store support 

- Add CarouselVideoItem type to types/index.ts
- Update GenerateVideoNodeData with videoHistory and selectedVideoHistoryIndex
- Add default values in createDefaultNodeData for generateVideo
- Add video history tracking in executeWorkflow for generateVideo
- Add video history tracking in regenerateNode for generateVideo
- Save videos with videoId for carousel support

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
handoff-20260429-1057
shrimbly 6 months ago
parent
38d7e55cde
commit
bcc9108be9
2 changed files with 75 additions and 2 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Split View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 67
      src/store/workflowStore.ts
  2. 10
      src/types/index.ts

67
src/store/workflowStore.ts
View File

@ -219,6 +219,8 @@ const createDefaultNodeData = (type: NodeType): WorkflowNodeData => {
selectedModel: undefined,
status: "idle",
error: null,
videoHistory: [],
selectedVideoHistoryIndex: 0,
} as GenerateVideoNodeData;
case "llmGenerate":
return {
@ -839,7 +841,7 @@ export const useWorkflowStore = create<WorkflowStore>((set, get) => ({
};
// Helper to extract output from source node
const getSourceOutput = (sourceNode: WorkflowNode): { type: "image" | "text"; value: string | null } => {
const getSourceOutput = (sourceNode: WorkflowNode): { type: "image" | "text" | "video"; value: string | null } => {
if (sourceNode.type === "imageInput") {
return { type: "image", value: (sourceNode.data as ImageInputNodeData).image };
} else if (sourceNode.type === "annotation") {
@ -847,7 +849,8 @@ export const useWorkflowStore = create<WorkflowStore>((set, get) => ({
} else if (sourceNode.type === "nanoBanana") {
return { type: "image", value: (sourceNode.data as NanoBananaNodeData).outputImage };
} else if (sourceNode.type === "generateVideo") {
return { type: "image", value: (sourceNode.data as GenerateVideoNodeData).outputVideo };
// Return video type - generateVideo and output nodes handle this appropriately
return { type: "video", value: (sourceNode.data as GenerateVideoNodeData).outputVideo };
} else if (sourceNode.type === "prompt") {
return { type: "text", value: (sourceNode.data as PromptNodeData).prompt };
} else if (sourceNode.type === "llmGenerate") {
@ -1390,10 +1393,24 @@ export const useWorkflowStore = create<WorkflowStore>((set, get) => ({
// Handle video response (video or videoUrl field)
const videoData = result.video || result.videoUrl;
if (result.success && videoData) {
const timestamp = Date.now();
const videoId = `${timestamp}`;
// Add to node's video history
const newHistoryItem = {
id: videoId,
timestamp,
prompt: text || '',
model: nodeData.selectedModel?.modelId || '',
};
const updatedHistory = [newHistoryItem, ...(nodeData.videoHistory || [])].slice(0, 50);
updateNodeData(node.id, {
outputVideo: videoData,
status: "complete",
error: null,
videoHistory: updatedHistory,
selectedVideoHistoryIndex: 0,
});
// Auto-save video to generations folder if configured
@ -1406,6 +1423,7 @@ export const useWorkflowStore = create<WorkflowStore>((set, get) => ({
directoryPath: genPath,
video: videoData,
prompt: text,
imageId: videoId,
}),
}).catch((err) => {
console.error("Failed to save video generation:", err);
@ -1413,10 +1431,24 @@ export const useWorkflowStore = create<WorkflowStore>((set, get) => ({
}
} else if (result.success && result.image) {
// Some models might return an image preview; treat as video for now
const timestamp = Date.now();
const videoId = `${timestamp}`;
// Add to node's video history
const newHistoryItem = {
id: videoId,
timestamp,
prompt: text || '',
model: nodeData.selectedModel?.modelId || '',
};
const updatedHistory = [newHistoryItem, ...(nodeData.videoHistory || [])].slice(0, 50);
updateNodeData(node.id, {
outputVideo: result.image,
status: "complete",
error: null,
videoHistory: updatedHistory,
selectedVideoHistoryIndex: 0,
});
// Auto-save image preview to generations folder if configured
@ -1429,6 +1461,7 @@ export const useWorkflowStore = create<WorkflowStore>((set, get) => ({
directoryPath: genPath,
image: result.image,
prompt: text,
imageId: videoId,
}),
}).catch((err) => {
console.error("Failed to save video generation:", err);
@ -2134,10 +2167,24 @@ export const useWorkflowStore = create<WorkflowStore>((set, get) => ({
const result = await response.json();
const videoData = result.video || result.videoUrl;
if (result.success && videoData) {
const timestamp = Date.now();
const videoId = `${timestamp}`;
// Add to node's video history
const newHistoryItem = {
id: videoId,
timestamp,
prompt: text || '',
model: nodeData.selectedModel?.modelId || '',
};
const updatedHistory = [newHistoryItem, ...(nodeData.videoHistory || [])].slice(0, 50);
updateNodeData(nodeId, {
outputVideo: videoData,
status: "complete",
error: null,
videoHistory: updatedHistory,
selectedVideoHistoryIndex: 0,
});
// Auto-save video to generations folder if configured
@ -2150,16 +2197,31 @@ export const useWorkflowStore = create<WorkflowStore>((set, get) => ({
directoryPath: genPath,
video: videoData,
prompt: text,
imageId: videoId,
}),
}).catch((err) => {
console.error("Failed to save video generation:", err);
});
}
} else if (result.success && result.image) {
const timestamp = Date.now();
const videoId = `${timestamp}`;
// Add to node's video history
const newHistoryItem = {
id: videoId,
timestamp,
prompt: text || '',
model: nodeData.selectedModel?.modelId || '',
};
const updatedHistory = [newHistoryItem, ...(nodeData.videoHistory || [])].slice(0, 50);
updateNodeData(nodeId, {
outputVideo: result.image,
status: "complete",
error: null,
videoHistory: updatedHistory,
selectedVideoHistoryIndex: 0,
});
// Auto-save image preview to generations folder if configured
@ -2172,6 +2234,7 @@ export const useWorkflowStore = create<WorkflowStore>((set, get) => ({
directoryPath: genPath,
image: result.image,
prompt: text,
imageId: videoId,
}),
}).catch((err) => {
console.error("Failed to save video generation:", err);

10
src/types/index.ts
View File

@ -133,6 +133,14 @@ export interface CarouselImageItem {
model: ModelType;
}
// Carousel Video Item (for per-node video history)
export interface CarouselVideoItem {
id: string;
timestamp: number;
prompt: string;
model: string; // Model ID for video (not ModelType since external providers)
}
// Model input definition for dynamic handles
export interface ModelInputDef {
name: string;
@ -174,6 +182,8 @@ export interface GenerateVideoNodeData extends BaseNodeData {
inputSchema?: ModelInputDef[]; // Model's input schema for dynamic handles
status: NodeStatus;
error: string | null;
videoHistory: CarouselVideoItem[]; // Carousel history (IDs only)
selectedVideoHistoryIndex: number; // Currently selected video in carousel
}
// LLM Generate Node Data (Text Generation)

Write Preview
Loading…
Cancel
Save