diff --git a/invokeai/frontend/web/src/features/canvas/store/canvasSlice.ts b/invokeai/frontend/web/src/features/canvas/store/canvasSlice.ts index ad0581e42f..7f41066ba1 100644 --- a/invokeai/frontend/web/src/features/canvas/store/canvasSlice.ts +++ b/invokeai/frontend/web/src/features/canvas/store/canvasSlice.ts @@ -29,6 +29,7 @@ import { isCanvasMaskLine, } from './canvasTypes'; import { ImageDTO } from 'services/api'; +import { sessionCanceled } from 'services/thunks/session'; export const initialLayerState: CanvasLayerState = { objects: [], @@ -844,6 +845,13 @@ export const canvasSlice = createSlice({ state.isTransformingBoundingBox = false; }, }, + extraReducers: (builder) => { + builder.addCase(sessionCanceled.pending, (state) => { + if (!state.layerState.stagingArea.images.length) { + state.layerState.stagingArea = initialLayerState.stagingArea; + } + }); + }, }); export const {