feat(ui): restore image post-upload actions

eg set init image if on img2img when uploading
This commit is contained in:
psychedelicious
2023-05-15 18:52:48 +10:00
parent 835922ea8f
commit d95fe5925a
6 changed files with 34 additions and 25 deletions

View File

@@ -4,6 +4,8 @@ import { uploadAdded } from 'features/gallery/store/uploadsSlice';
import { imageSelected } from 'features/gallery/store/gallerySlice';
import { imageUploaded } from 'services/thunks/image';
import { addToast } from 'features/system/store/systemSlice';
import { initialImageSelected } from 'features/parameters/store/actions';
import { setInitialCanvasImage } from 'features/canvas/store/canvasSlice';
export const addImageUploadedListener = () => {
startAppListening({
@@ -23,6 +25,14 @@ export const addImageUploadedListener = () => {
if (state.gallery.shouldAutoSwitchToNewImages) {
dispatch(imageSelected(image));
}
if (action.meta.arg.activeTabName === 'img2img') {
dispatch(initialImageSelected(image));
}
if (action.meta.arg.activeTabName === 'unifiedCanvas') {
dispatch(setInitialCanvasImage(image));
}
},
});
};