| import { api } from "../../scripts/api.js"; | |
| import { getResolver } from "./shared_utils.js"; | |
| export class PromptExecution { | |
| constructor(id) { | |
| this.promptApi = null; | |
| this.executedNodeIds = []; | |
| this.totalNodes = 0; | |
| this.currentlyExecuting = null; | |
| this.errorDetails = null; | |
| this.apiPrompt = getResolver(); | |
| this.id = id; | |
| } | |
| setPrompt(prompt) { | |
| this.promptApi = prompt.output; | |
| this.totalNodes = Object.keys(this.promptApi).length; | |
| this.apiPrompt.resolve(null); | |
| } | |
| getApiNode(nodeId) { | |
| var _a; | |
| return ((_a = this.promptApi) === null || _a === void 0 ? void 0 : _a[String(nodeId)]) || null; | |
| } | |
| getNodeLabel(nodeId) { | |
| var _a, _b; | |
| const apiNode = this.getApiNode(nodeId); | |
| let label = ((_a = apiNode === null || apiNode === void 0 ? void 0 : apiNode._meta) === null || _a === void 0 ? void 0 : _a.title) || (apiNode === null || apiNode === void 0 ? void 0 : apiNode.class_type) || undefined; | |
| if (!label) { | |
| const graphNode = (_b = this.maybeGetComfyGraph()) === null || _b === void 0 ? void 0 : _b.getNodeById(Number(nodeId)); | |
| label = (graphNode === null || graphNode === void 0 ? void 0 : graphNode.title) || (graphNode === null || graphNode === void 0 ? void 0 : graphNode.type) || undefined; | |
| } | |
| return label; | |
| } | |
| executing(nodeId, step, maxSteps) { | |
| var _a; | |
| if (nodeId == null) { | |
| this.currentlyExecuting = null; | |
| return; | |
| } | |
| if (((_a = this.currentlyExecuting) === null || _a === void 0 ? void 0 : _a.nodeId) !== nodeId) { | |
| if (this.currentlyExecuting != null) { | |
| this.executedNodeIds.push(nodeId); | |
| } | |
| this.currentlyExecuting = { nodeId, nodeLabel: this.getNodeLabel(nodeId), pass: 0 }; | |
| this.apiPrompt.promise.then(() => { | |
| var _a; | |
| if (this.currentlyExecuting == null) { | |
| return; | |
| } | |
| const apiNode = this.getApiNode(nodeId); | |
| if (!this.currentlyExecuting.nodeLabel) { | |
| this.currentlyExecuting.nodeLabel = this.getNodeLabel(nodeId); | |
| } | |
| if ((apiNode === null || apiNode === void 0 ? void 0 : apiNode.class_type) === "UltimateSDUpscale") { | |
| this.currentlyExecuting.pass--; | |
| this.currentlyExecuting.maxPasses = -1; | |
| } | |
| else if ((apiNode === null || apiNode === void 0 ? void 0 : apiNode.class_type) === "IterativeImageUpscale") { | |
| this.currentlyExecuting.maxPasses = (_a = apiNode === null || apiNode === void 0 ? void 0 : apiNode.inputs["steps"]) !== null && _a !== void 0 ? _a : -1; | |
| } | |
| }); | |
| } | |
| if (step != null) { | |
| if (!this.currentlyExecuting.step || step < this.currentlyExecuting.step) { | |
| this.currentlyExecuting.pass++; | |
| } | |
| this.currentlyExecuting.step = step; | |
| this.currentlyExecuting.maxSteps = maxSteps; | |
| } | |
| } | |
| error(details) { | |
| this.errorDetails = details; | |
| } | |
| maybeGetComfyGraph() { | |
| var _a; | |
| return ((_a = window === null || window === void 0 ? void 0 : window.app) === null || _a === void 0 ? void 0 : _a.graph) || null; | |
| } | |
| } | |
| class PromptService extends EventTarget { | |
| constructor(api) { | |
| super(); | |
| this.promptsMap = new Map(); | |
| this.currentExecution = null; | |
| this.lastQueueRemaining = 0; | |
| const that = this; | |
| const queuePrompt = api.queuePrompt; | |
| api.queuePrompt = async function (num, prompt) { | |
| let response; | |
| try { | |
| response = await queuePrompt.apply(api, [...arguments]); | |
| } | |
| catch (e) { | |
| const promptExecution = that.getOrMakePrompt("error"); | |
| promptExecution.error({ exception_type: "Unknown." }); | |
| throw e; | |
| } | |
| const promptExecution = that.getOrMakePrompt(response.prompt_id); | |
| promptExecution.setPrompt(prompt); | |
| if (!that.currentExecution) { | |
| that.currentExecution = promptExecution; | |
| } | |
| that.promptsMap.set(response.prompt_id, promptExecution); | |
| that.dispatchEvent(new CustomEvent("queue-prompt", { | |
| detail: { | |
| prompt: promptExecution, | |
| }, | |
| })); | |
| return response; | |
| }; | |
| api.addEventListener("status", (e) => { | |
| var _a; | |
| if (!((_a = e.detail) === null || _a === void 0 ? void 0 : _a.exec_info)) | |
| return; | |
| this.lastQueueRemaining = e.detail.exec_info.queue_remaining; | |
| this.dispatchProgressUpdate(); | |
| }); | |
| api.addEventListener("execution_start", (e) => { | |
| if (!this.promptsMap.has(e.detail.prompt_id)) { | |
| console.warn("'execution_start' fired before prompt was made."); | |
| } | |
| const prompt = this.getOrMakePrompt(e.detail.prompt_id); | |
| this.currentExecution = prompt; | |
| this.dispatchProgressUpdate(); | |
| }); | |
| api.addEventListener("executing", (e) => { | |
| if (!this.currentExecution) { | |
| this.currentExecution = this.getOrMakePrompt("unknown"); | |
| console.warn("'executing' fired before prompt was made."); | |
| } | |
| this.currentExecution.executing(e.detail); | |
| this.dispatchProgressUpdate(); | |
| if (e.detail == null) { | |
| this.currentExecution = null; | |
| } | |
| }); | |
| api.addEventListener("progress", (e) => { | |
| if (!this.currentExecution) { | |
| this.currentExecution = this.getOrMakePrompt(e.detail.prompt_id); | |
| console.warn("'progress' fired before prompt was made."); | |
| } | |
| this.currentExecution.executing(e.detail.node, e.detail.value, e.detail.max); | |
| this.dispatchProgressUpdate(); | |
| }); | |
| api.addEventListener("execution_cached", (e) => { | |
| if (!this.currentExecution) { | |
| this.currentExecution = this.getOrMakePrompt(e.detail.prompt_id); | |
| console.warn("'execution_cached' fired before prompt was made."); | |
| } | |
| for (const cached of e.detail.nodes) { | |
| this.currentExecution.executing(cached); | |
| } | |
| this.dispatchProgressUpdate(); | |
| }); | |
| api.addEventListener("executed", (e) => { | |
| if (!this.currentExecution) { | |
| this.currentExecution = this.getOrMakePrompt(e.detail.prompt_id); | |
| console.warn("'executed' fired before prompt was made."); | |
| } | |
| }); | |
| api.addEventListener("execution_error", (e) => { | |
| var _a; | |
| if (!this.currentExecution) { | |
| this.currentExecution = this.getOrMakePrompt(e.detail.prompt_id); | |
| console.warn("'execution_error' fired before prompt was made."); | |
| } | |
| (_a = this.currentExecution) === null || _a === void 0 ? void 0 : _a.error(e.detail); | |
| this.dispatchProgressUpdate(); | |
| }); | |
| } | |
| async queuePrompt(prompt) { | |
| return await api.queuePrompt(-1, prompt); | |
| } | |
| dispatchProgressUpdate() { | |
| this.dispatchEvent(new CustomEvent("progress-update", { | |
| detail: { | |
| queue: this.lastQueueRemaining, | |
| prompt: this.currentExecution, | |
| }, | |
| })); | |
| } | |
| getOrMakePrompt(id) { | |
| let prompt = this.promptsMap.get(id); | |
| if (!prompt) { | |
| prompt = new PromptExecution(id); | |
| this.promptsMap.set(id, prompt); | |
| } | |
| return prompt; | |
| } | |
| } | |
| export const SERVICE = new PromptService(api); | |