Use eventSource instead of DOM click

3cbbb2fe3927734897f3b1c743e86f6603269609

Cohee <18619528+Cohee1207@users.noreply.github.com>

Parents: 3c059d19efe8975276582c6d4087d250250a7e01
1 files changed, +2 -3Ignore whitespace
public/scripts/extensions/stable-diffusion/index.js+2 -3
@@ -2294,14 +2294,13 @@ async function generatePicture(initiator, args, trigger, message, callback) {
2294 let imagePath = '';2294 let imagePath = '';
22952295
2296 const stopListener = () => abortController.abort('Aborted by user');2296 const stopListener = () => abortController.abort('Aborted by user');
2297 const mesStop = document.getElementById('mes_stop');
22982297
2299 try {2298 try {
2300 const combineNegatives = (prefix) => { negativePromptPrefix = combinePrefixes(negativePromptPrefix, prefix); };2299 const combineNegatives = (prefix) => { negativePromptPrefix = combinePrefixes(negativePromptPrefix, prefix); };
2301 const prompt = await getPrompt(generationType, message, trigger, quietPrompt, combineNegatives);2300 const prompt = await getPrompt(generationType, message, trigger, quietPrompt, combineNegatives);
2302 console.log('Processed image prompt:', prompt);2301 console.log('Processed image prompt:', prompt);
23032302
2304 mesStop?.addEventListener('click', stopListener);2303 eventSource.once(event_types.GENERATION_STOPPED, stopListener);
2305 context.deactivateSendButtons();2304 context.deactivateSendButtons();
2306 hideSwipeButtons();2305 hideSwipeButtons();
23072306
@@ -2316,7 +2315,7 @@ async function generatePicture(initiator, args, trigger, message, callback) {
2316 }2315 }
2317 finally {2316 finally {
2318 restoreOriginalDimensions(dimensions);2317 restoreOriginalDimensions(dimensions);
2319 mesStop?.removeEventListener('click', stopListener);2318 eventSource.removeListener(event_types.GENERATION_STOPPED, stopListener);
2320 context.activateSendButtons();2319 context.activateSendButtons();
2321 showSwipeButtons();2320 showSwipeButtons();
2322 }2321 }