Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fixes Canvas Auto Save to Gallery

Browse Source
This commit is contained in:
psychedelicious 2022-11-19 16:07:36 +11:00 committed by blessedcoolant
parent 2ab868314f
commit ae4a44de3e
1 changed files with 13 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
frontend/src/app/socketio/listeners.ts
View File

@ -98,7 +98,8 @@ const makeSocketIOListeners = (
*/
onGenerationResult: (data: InvokeAI.ImageResultResponse) => {
try {
const { shouldLoopback, activeTab } = getState().options;
const state = getState();
const { shouldLoopback, activeTab } = state.options;
const { boundingBox: _, generationMode, ...rest } = data;
const newImage = {
@ -107,24 +108,31 @@ const makeSocketIOListeners = (
};
if (['txt2img', 'img2img'].includes(generationMode)) {
newImage.category = 'result';
dispatch(
addImage({
category: 'result',
image: newImage,
image: { ...newImage, category: 'result' },
})
);
}
if (generationMode === 'unifiedCanvas' && data.boundingBox) {
newImage.category = 'temp';
const { boundingBox } = data;
dispatch(
addImageToStagingArea({
image: newImage,
image: { ...newImage, category: 'temp' },
boundingBox,
})
);
if (state.canvas.shouldAutoSave) {
dispatch(
addImage({
image: { ...newImage, category: 'result' },
category: 'result',
})
);
}
}
if (shouldLoopback) {