github/InvokeAI
1
1
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2026-04-23 03:00:31 -04:00
Code Issues Packages Projects Releases Wiki Activity

fix(ui): save to gallery uses auto-add board

Browse Source
This commit is contained in:
psychedelicious
2024-09-05 22:37:00 +10:00
parent 96b74f4a79
commit 029158ef3a
3 changed files with 32 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
invokeai/frontend/web/src/features/controlLayers/konva/CanvasCompositorModule.ts
View File

@@ -9,6 +9,7 @@ import {
previewBlob,
} from 'features/controlLayers/konva/util';
import type { GenerationMode, Rect } from 'features/controlLayers/store/types';
import { selectAutoAddBoardId } from 'features/gallery/store/gallerySelectors';
import type { Logger } from 'roarr';
import { getImageDTO, uploadImage } from 'services/api/endpoints/images';
import type { ImageDTO } from 'services/api/types';
@@ -134,7 +135,13 @@ export class CanvasCompositorModule extends CanvasModuleBase {
previewBlob(blob, 'Composite raster layer canvas');
}
return uploadImage(blob, 'composite-raster-layer.png', 'general', !saveToGallery);
return uploadImage({
blob,
fileName: 'composite-raster-layer.png',
image_category: 'general',
is_intermediate: !saveToGallery,
board_id: saveToGallery ? selectAutoAddBoardId(this.manager.store.getState()) : undefined,
});
};
/**
@@ -258,7 +265,13 @@ export class CanvasCompositorModule extends CanvasModuleBase {
previewBlob(blob, 'Composite inpaint mask canvas');
}
return uploadImage(blob, 'composite-inpaint-mask.png', 'general', !saveToGallery);
return uploadImage({
blob,
fileName: 'composite-inpaint-mask.png',
image_category: 'general',
is_intermediate: !saveToGallery,
board_id: saveToGallery ? selectAutoAddBoardId(this.manager.store.getState()) : undefined,
});
};
/**

7
invokeai/frontend/web/src/features/controlLayers/konva/CanvasEntityObjectRenderer.ts
View File

@@ -558,7 +558,12 @@ export class CanvasEntityObjectRenderer extends CanvasModuleBase {
if (this.manager._isDebugging) {
previewBlob(blob, 'Rasterized entity');
}
imageDTO = await uploadImage(blob, `${this.id}_rasterized.png`, 'other', true);
imageDTO = await uploadImage({
blob,
fileName: `${this.id}_rasterized.png`,
image_category: 'other',
is_intermediate: true,
});
const imageObject = imageDTOToImageObject(imageDTO);
if (replaceObjects) {
await this.setBuffer(imageObject);