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