// What a run is doing, beside Generate: the step it is on, a bar, and how long it has been going (this page's own clock). import { composer as c } from "../../composer/composer.js"; import { QUEUED, RUNNING } from "../../shell/run.js"; export const elapsed = (ms) => { const s = Math.max(0, Math.floor(ms / 1000)); return `${String(Math.floor(s / 60)).padStart(2, "0")}m ${String(s % 60).padStart(2, "0")}s`; }; export default { id: "generate-progress", mount: "timeline.actions", needs: ["generate"], setup({ host, app }) { const g = app.generate; const bar = c.progress.bar({ value: 0, max: 100, label: "Generating" }); const text = c.text.sm({ text: "" }); const wrap = document.createElement("span"); wrap.style.cssText = "display:inline-flex;align-items:center;gap:8px;min-inline-size:0"; bar.node.style.inlineSize = "90px"; wrap.append(bar.node, text.node); wrap.hidden = true; host.append(wrap); let since = 0; const draw = () => { const s = g.state(), working = s.phase === QUEUED || s.phase === RUNNING; if (working && !since) since = Date.now(); if (!working) since = 0; wrap.hidden = !working; if (!working) { return; } const pr = s.progress; bar.setValue(pr && pr.max ? (pr.value / pr.max) * 100 : 0); text.setText(`${s.phase === QUEUED ? "Queued" : pr && pr.max ? `Step ${pr.value}/${pr.max}` : "Running"} ยท ${elapsed(Date.now() - since)}`); }; const off = g.subscribe(draw); const timer = setInterval(() => { if (since) draw(); }, 1000); // the clock ticks between ComfyUI's messages draw(); return () => { off(); clearInterval(timer); wrap.remove(); }; }, };