DemonGatanjieuandClaude Opus 5.5
9abd596186
- Each run keeps its shape read from the prompt ComfyUI queued (ai_run_map.py): boxes by role with their settings, typed links, plumbing wired through, and what changed since the task's run before. The run step shows a one-line summary, the changes as chips and the boxes (top down, fitted to the column, each locating its node); show_run_map puts it on the stage. - By default the AI works in a tab of the task's own (ai_task_canvas.py, mcp_task_tab.js): the open workflow is copied before its first change, the tab is brought back (a click on ComfyUI's tab) before later changes, reads and runs, and the user's tab stays untouched. Per-task switch in the header, default in the Gates window; the AI is told once. - Output pictures can be handed to the task an AI is on (gallery card menu, the inspector's ✦), with the user's words: the AI gets them attached with what made them (ai_pictures.py). search_images(with_pictures) attaches the first six. Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
69 lines
3.5 KiB
JavaScript
69 lines
3.5 KiB
JavaScript
/**
|
|
* The task's own canvas tab (api/ai_task_canvas.py), on the page: before an AI's canvas action
|
|
* (mcp_bridge.js), with the task set to "new", the task's tab is brought to the front the way a
|
|
* click on it does (ComfyUI's store alone would not load its graph); with no tab yet, the first
|
|
* change copies the open workflow into a new tab named after the task and says so in a toast,
|
|
* so the user's own tab is never changed. Reads and runs follow the task's tab when it is open
|
|
* and use the canvas as it is before it has one; new_workflow and open_workflow make the tab
|
|
* they open the task's. A tab the user closed is made again from the open workflow.
|
|
*/
|
|
|
|
import { app } from '../../../scripts/app.js';
|
|
import { translate as t } from './locales.js';
|
|
import { showWorkbenchToast } from './ui_prompt_toast.js';
|
|
import { tabName } from './mcp_canvas_edit.js';
|
|
|
|
const CHANGES = new Set(['set_prompt', 'set_model', 'add_lora', 'place_combo', 'fix_models', 'add_node', 'connect_nodes', 'set_values', 'remove_nodes']);
|
|
const FOLLOWS = new Set(['describe_canvas', 'check_models', 'run_workflow']);
|
|
const OPENS = new Set(['new_workflow', 'open_workflow']);
|
|
const SWITCH_WAIT_MS = 4000;
|
|
|
|
const store = () => app.extensionManager?.workflow || null;
|
|
const wait = ms => new Promise(resolve => setTimeout(resolve, ms));
|
|
const shown = path => store()?.activeWorkflow?.path === path && !store()?.activeWorkflow?.isLoading;
|
|
|
|
function frontTab(made = false, again = false) {
|
|
const workflow = store()?.activeWorkflow;
|
|
return workflow ? { path: workflow.path, name: String(workflow.filename || workflow.path).replace(/\.json$/i, ''), made, again } : null;
|
|
}
|
|
|
|
async function settled(path, limit = SWITCH_WAIT_MS) {
|
|
for (let waited = 0; waited < limit && !shown(path); waited += 50) await wait(50);
|
|
if (shown(path)) await new Promise(resolve => requestAnimationFrame(resolve));
|
|
return shown(path);
|
|
}
|
|
|
|
/** Brings open tab `path` to the front as a click on it does; false when it is not open. */
|
|
export async function showTab(path) {
|
|
const workflows = store()?.openWorkflows || [];
|
|
const index = workflows.findIndex(workflow => workflow.path === path);
|
|
if (index < 0) return false;
|
|
if (shown(path)) return true;
|
|
document.querySelectorAll('.workflow-tabs > button')[index]?.click();
|
|
if (await settled(path)) return true;
|
|
// Tabs not on screen (another layout): step through the open ones, as Ctrl+Tab does.
|
|
for (let step = 0; step < workflows.length && !shown(path); step++) {
|
|
await app.extensionManager?.command?.execute?.('Workspace.NextOpenedWorkflow');
|
|
await settled(path, 600);
|
|
}
|
|
return shown(path);
|
|
}
|
|
|
|
/** Runs page action `action` (`run()`) on the task's tab per `spec` ({mode, tab, title}); `opened(tab)`
|
|
* hears of a tab made or opened for the task ({path, name, made, again}). Returns what `run()` does. */
|
|
export async function onTaskTab(action, spec, run, opened) {
|
|
if (spec?.mode !== 'new' || !(CHANGES.has(action) || FOLLOWS.has(action) || OPENS.has(action))) return run();
|
|
if (OPENS.has(action)) {
|
|
const value = await run();
|
|
opened(frontTab());
|
|
return value;
|
|
}
|
|
const path = spec.tab?.path;
|
|
if ((path && await showTab(path)) || !CHANGES.has(action)) return run();
|
|
const name = tabName(spec.title || t('aiTaskTabDefault'));
|
|
await app.loadGraphData(app.graph.serialize(), true, true, name);
|
|
opened(frontTab(true, Boolean(path)));
|
|
showWorkbenchToast(t(path ? 'aiTaskTabAgain' : 'aiTaskTabMade', { name }));
|
|
return run();
|
|
}
|