24 lines
526 B
JavaScript
24 lines
526 B
JavaScript
|
// @ts-check
|
||
|
|
|
||
|
|
import { api } from "../../api.js";
|
||
|
|
import { ComfyButton } from "../components/button.js";
|
||
|
|
|
||
|
|
export function getInteruptButton(visibility) {
|
||
|
|
const btn = new ComfyButton({
|
||
|
|
icon: "close",
|
||
|
|
tooltip: "Cancel current generation",
|
||
|
|
enabled: false,
|
||
|
|
action: () => {
|
||
|
|
api.interrupt();
|
||
|
|
},
|
||
|
|
classList: ["comfyui-button", "comfyui-interrupt-button", visibility],
|
||
|
|
});
|
||
|
|
|
||
|
|
api.addEventListener("status", ({ detail }) => {
|
||
|
|
const sz = detail?.exec_info?.queue_remaining;
|
||
|
|
btn.enabled = sz > 0;
|
||
|
|
});
|
||
|
|
|
||
|
|
return btn;
|
||
|
|
}
|