49 lines
3.7 KiB
JavaScript
49 lines
3.7 KiB
JavaScript
import {T} from './i18n.js';
|
|
import {app} from '../../scripts/app.js';
|
|
import {api} from '../../scripts/api.js';
|
|
import {workspaceRequest} from './workspace_transport.js';
|
|
import {workspaceRoutingNodes} from './workspace_activation.js';
|
|
import {applyExecutionBinding,graphNodeEntries,executionId,executionNode} from './workspace_flow.js';
|
|
const graphTasks=new WeakMap();
|
|
const link=value=>Array.isArray(value)&&value.length===2&&typeof value[0]==='string'&&Number.isInteger(value[1]);
|
|
function depends(prompt,ref,target,seen=new Set()) {
|
|
if(!link(ref)||seen.has(ref[0]))return false;if(ref[0]===target)return true;seen.add(ref[0]);
|
|
return Object.values(prompt[ref[0]]?.inputs||{}).some(input=>depends(prompt,input,target,seen));
|
|
}
|
|
export function prepareWorkspacePrompt(options) {
|
|
const graph=app.rootGraph||app.graph;
|
|
const previous=graphTasks.get(graph)||Promise.resolve();
|
|
const task=previous.catch(()=>{}).then(()=>prepareSnapshot(graph,options));graphTasks.set(graph,task);return task;
|
|
}
|
|
async function prepareSnapshot(graph,{editorNode,samplerNode,compareNode,binding={},getBinding,allowEmpty=false,controller}) {
|
|
if(!editorNode?.adachiEditor?.ready && !allowEmpty)throw Error(T("Upload a source image first"));
|
|
const entries=graphNodeEntries(graph),roots=[editorNode,samplerNode,compareNode];
|
|
if(graph!==(app.rootGraph||app.graph)||roots.some(n=>!entries.some(e=>e.node===n)))throw Error(T("The workflow changed; reopen the workbench"));
|
|
const ids=roots.map(n=>executionId(graph,n)),[editorId,samplerId,compareId]=ids;
|
|
const parents=roots.flatMap(node=>{const path=executionId(graph,node).split(':');return path.slice(0,-1).map((_,i)=>executionNode(graph,path.slice(0,i+1).join(':'))).filter(Boolean);});
|
|
const routers=workspaceRoutingNodes(roots,controller),modes=new Map([...parents,...roots,...routers].map(n=>[n,n.mode]));
|
|
if(editorNode.adachiEditor.ready)editorNode.adachiEditor.flush();let prepared;
|
|
try {
|
|
for(const node of modes.keys())node.mode=0;
|
|
prepared=typeof app.ue_modified_prompt==='function'?await app.ue_modified_prompt():await app.graphToPrompt();
|
|
if(!prepared?.output)throw Error(T("Workflow connections could not be resolved; check bypass states, Get/Set and broadcast connections"));
|
|
} finally {for(const [node,mode]of modes)node.mode=mode;}
|
|
if(graph!==(app.rootGraph||app.graph))throw Error(T("The workflow changed; check connections again"));
|
|
const output=prepared.output,sampler=output[samplerId],compare=output[compareId];
|
|
if(!sampler||!compare||!output[editorId])throw Error(T("The selected image, inpaint or comparison node is missing from the execution graph"));
|
|
// Existing image processors and result postprocessing remain connected. An
|
|
// unconnected workbench can use its explicitly selected inline source/output.
|
|
if(!depends(output,sampler.inputs.image,editorId)){sampler.inputs.image=[editorId,0];sampler.inputs.mask=[editorId,1];}
|
|
else if(!link(sampler.inputs.mask))sampler.inputs.mask=[editorId,1];
|
|
if(!link(compare.inputs.original))compare.inputs.original=[editorId,0];
|
|
if(!depends(output,compare.inputs.result,samplerId))compare.inputs.result=[samplerId,0];
|
|
const resolved=getBinding?await getBinding(output,samplerId):binding;
|
|
const prompt=applyExecutionBinding(output,samplerId,resolved);
|
|
// Keep the editable workflow, including ignored groups and original routers,
|
|
// in PNG metadata. Only the separate API execution graph is specialized.
|
|
const workflow=JSON.parse(JSON.stringify(graph.serialize()));
|
|
return {prompt,compare_id:compareId,client_id:api.clientId||'',extra_data:{extra_pnginfo:{workflow},auth_token_comfy_org:api.authToken,api_key_comfy_org:api.apiKey}};
|
|
}
|
|
export const validateWorkspace=(prepared,options)=>workspaceRequest(api,'validate',prepared,options);
|
|
export const submitWorkspace=prepared=>workspaceRequest(api,'submit',prepared);
|