Add swipes when using paintbrush

4fd8d8e0eedabc955653884ee87e49e05bcce404

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

1 files changed, +14 -1Ignore whitespace
public/scripts/extensions/stable-diffusion/index.js+14 -1
@@ -3617,10 +3617,23 @@ async function sdMessageButton(e) {
36173617
36183618 function saveGeneratedImage(prompt, image, generationType, negative) {
36193619 // Some message sources may not create the extra object
36203620 if (typeof message.extra !== 'object' || message.extra === null) {
36213621 message.extra = {};
36223622 }
36233623
3624+ // Add image to the swipe list if it's not already there
3625+ if (!Array.isArray(message.extra.image_swipes)) {
3626+ message.extra.image_swipes = [];
3627+ }
3628+
3629+ const swipes = message.extra.image_swipes;
3630+
3631+ if (message.extra.image && !swipes.includes(message.extra.image)) {
3632+ swipes.push(message.extra.image);
3633+ }
3634+
3635+ swipes.push(image);
3636+
36243637 // If already contains an image and it's not inline - leave it as is
36253638 message.extra.inline_image = message.extra.image && !message.extra.inline_image ? false : true;
36263639 message.extra.image = image;