Image Generation: Fix race condition in adding images to chat

8ff9ef3610f7eb1b793062528f6b548859f54431

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

public/scripts/extensions/stable-diffusion/index.js+5 -3 Viewed

@@ -2305,9 +2305,9 @@ async function generatePicture(initiator, args, trigger, message, callback) {
2305	2305	eventSource.emit(event_types.FORCE_SET_BACKGROUND, { url: imgUrl, path: imagePath });
2306	2306
2307	2307	if (typeof callbackOriginal === 'function') {
2308	2308	await callbackOriginal(prompt, imagePath, generationType, negativePromptPrefix, initiator);
2309	2309	} else {
2310	2310	await sendMessage(prompt, imagePath, generationType, negativePromptPrefix, initiator);
2311	2311	}
2312	2312	};
2313	2313	}

     const filename = `${characterName}_${humanizedDateTime()}`;
     const base64Image = await saveBase64AsFile(result.data, characterName, filename, result.format);
 -    callback ? callback(prompt, base64Image, generationType, additionalNegativePrefix, initiator) : sendMessage(prompt, base64Image, generationType, additionalNegativePrefix, initiator);
 +    callback
 +        ? await callback(prompt, base64Image, generationType, additionalNegativePrefix, initiator)
 +        : await sendMessage(prompt, base64Image, generationType, additionalNegativePrefix, initiator);
     return base64Image;
+}