diff --git a/invokeai/frontend/web/src/features/controlLayers/store/controlAdaptersReducers.ts b/invokeai/frontend/web/src/features/controlLayers/store/controlAdaptersReducers.ts index 0a744c6c28..5c2813c2d3 100644 --- a/invokeai/frontend/web/src/features/controlLayers/store/controlAdaptersReducers.ts +++ b/invokeai/frontend/web/src/features/controlLayers/store/controlAdaptersReducers.ts @@ -44,13 +44,16 @@ export const controlAdaptersReducers = { processorPendingBatchId: null, ...config, }); + state.selectedEntityIdentifier = { type: 'control_adapter', id }; }, prepare: (config: ControlNetConfig | T2IAdapterConfig) => ({ payload: { id: uuidv4(), config }, }), }, caRecalled: (state, action: PayloadAction<{ data: ControlAdapterData }>) => { - state.controlAdapters.push(action.payload.data); + const { data } = action.payload; + state.controlAdapters.push(data); + state.selectedEntityIdentifier = { type: 'control_adapter', id: data.id }; }, caIsEnabledToggled: (state, action: PayloadAction<{ id: string }>) => { const { id } = action.payload; diff --git a/invokeai/frontend/web/src/features/controlLayers/store/ipAdaptersReducers.ts b/invokeai/frontend/web/src/features/controlLayers/store/ipAdaptersReducers.ts index fb6ff8d987..a4e950767b 100644 --- a/invokeai/frontend/web/src/features/controlLayers/store/ipAdaptersReducers.ts +++ b/invokeai/frontend/web/src/features/controlLayers/store/ipAdaptersReducers.ts @@ -25,11 +25,14 @@ export const ipAdaptersReducers = { ...config, }; state.ipAdapters.push(layer); + state.selectedEntityIdentifier = { type: 'ip_adapter', id }; }, prepare: (config: IPAdapterConfig) => ({ payload: { id: uuidv4(), config } }), }, ipaRecalled: (state, action: PayloadAction<{ data: IPAdapterData }>) => { - state.ipAdapters.push(action.payload.data); + const { data } = action.payload; + state.ipAdapters.push(data); + state.selectedEntityIdentifier = { type: 'ip_adapter', id: data.id }; }, ipaIsEnabledToggled: (state, action: PayloadAction<{ id: string }>) => { const { id } = action.payload; diff --git a/invokeai/frontend/web/src/features/controlLayers/store/layersReducers.ts b/invokeai/frontend/web/src/features/controlLayers/store/layersReducers.ts index 6b3e58f8a2..dc06f8d188 100644 --- a/invokeai/frontend/web/src/features/controlLayers/store/layersReducers.ts +++ b/invokeai/frontend/web/src/features/controlLayers/store/layersReducers.ts @@ -38,11 +38,14 @@ export const layersReducers = { x: 0, y: 0, }); + state.selectedEntityIdentifier = { type: 'layer', id }; }, prepare: () => ({ payload: { id: uuidv4() } }), }, layerRecalled: (state, action: PayloadAction<{ data: LayerData }>) => { - state.layers.push(action.payload.data); + const { data } = action.payload; + state.layers.push(data); + state.selectedEntityIdentifier = { type: 'layer', id: data.id }; }, layerIsEnabledToggled: (state, action: PayloadAction<{ id: string }>) => { const { id } = action.payload; diff --git a/invokeai/frontend/web/src/features/controlLayers/store/regionsReducers.ts b/invokeai/frontend/web/src/features/controlLayers/store/regionsReducers.ts index 9d75e0e93f..680eb35a82 100644 --- a/invokeai/frontend/web/src/features/controlLayers/store/regionsReducers.ts +++ b/invokeai/frontend/web/src/features/controlLayers/store/regionsReducers.ts @@ -72,6 +72,7 @@ export const regionsReducers = { imageCache: null, }; state.regions.push(rg); + state.selectedEntityIdentifier = { type: 'regional_guidance', id }; }, prepare: () => ({ payload: { id: uuidv4() } }), }, @@ -89,6 +90,7 @@ export const regionsReducers = { rgRecalled: (state, action: PayloadAction<{ data: RegionalGuidanceData }>) => { const { data } = action.payload; state.regions.push(data); + state.selectedEntityIdentifier = { type: 'regional_guidance', id: data.id }; }, rgIsEnabledToggled: (state, action: PayloadAction<{ id: string }>) => { const { id } = action.payload;