Files
ComfyUI_frontend/src/scripts/ui/menu/interruptButton.ts
2024-07-25 10:10:18 -04:00

22 lines
529 B
TypeScript

import { api } from '../../api'
import { ComfyButton } from '../components/button'
export function getInteruptButton(visibility: string) {
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
}