diff --git a/javascript/prompt_format.js b/javascript/prompt_format.js index a1f573c..5fce7ba 100644 --- a/javascript/prompt_format.js +++ b/javascript/prompt_format.js @@ -226,9 +226,18 @@ onUiLoaded(async () => { tools.after(formatter); ['txt', 'img'].forEach((mode) => { - gradioApp().getElementById(`${mode}2img_generate`).addEventListener('click', () => { - if (autoRun) + const generateButton = gradioApp().getElementById(`${mode}2img_generate`); + const enqueueButton = gradioApp().getElementById(`${mode}2img_enqueue`); + + const handleClick = () => { + if (autoRun) { promptFields.forEach((field) => LeFormatter.formatPipeline(field, dedupe, removeUnderscore, refresh)); - }); + } + }; + + generateButton.addEventListener('click', handleClick); + if (enqueueButton) { + enqueueButton.addEventListener('click', handleClick); + } }); });