rixty
/
SillyTavern
mirror of https://github.com/SillyTavern/SillyTavern.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Use eventSource instead of DOM click

This commit is contained in:
Cohee 2024-07-19 02:12:01 +03:00
parent 3c059d19ef
commit 3cbbb2fe39
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
public/scripts/extensions/stable-diffusion/index.js
View File

@ -2294,14 +2294,13 @@ async function generatePicture(initiator, args, trigger, message, callback) {
let imagePath = '';
const stopListener = () => abortController.abort('Aborted by user');
const mesStop = document.getElementById('mes_stop');
try {
const combineNegatives = (prefix) => { negativePromptPrefix = combinePrefixes(negativePromptPrefix, prefix); };
const prompt = await getPrompt(generationType, message, trigger, quietPrompt, combineNegatives);
console.log('Processed image prompt:', prompt);
mesStop?.addEventListener('click', stopListener);
eventSource.once(event_types.GENERATION_STOPPED, stopListener);
context.deactivateSendButtons();
hideSwipeButtons();
@ -2316,7 +2315,7 @@ async function generatePicture(initiator, args, trigger, message, callback) {
}
finally {
restoreOriginalDimensions(dimensions);
mesStop?.removeEventListener('click', stopListener);
eventSource.removeListener(event_types.GENERATION_STOPPED, stopListener);
context.activateSendButtons();
showSwipeButtons();
}