github/InvokeAI
1
1
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2026-02-14 10:34:55 -05:00
Code Issues Packages Projects Releases Wiki Activity

feat(ui): restore image post-upload actions

Browse Source 
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
Download Patch File Download Diff File Expand all files Collapse all files

11
invokeai/frontend/web/src/services/thunks/image.ts
View File

@@ -1,5 +1,6 @@
import { log } from 'app/logging/useLogger';
import { createAppAsyncThunk } from 'app/store/storeUtils';
import { InvokeTabName } from 'features/ui/store/tabMap';
import { ImagesService } from 'services/api';
import { getHeaders } from 'services/util/getHeaders';
@@ -39,7 +40,11 @@ export const thumbnailReceived = createAppAsyncThunk(
}
);
type ImageUploadedArg = Parameters<(typeof ImagesService)['uploadImage']>[0];
type ImageUploadedArg = Parameters<(typeof ImagesService)['uploadImage']>[0] & {
// extra arg to determine post-upload actions - we check for this when the image is uploaded
// to determine if we should set the init image
activeTabName?: InvokeTabName;
};
/**
* `ImagesService.uploadImage()` thunk
@@ -47,7 +52,9 @@ type ImageUploadedArg = Parameters<(typeof ImagesService)['uploadImage']>[0];
export const imageUploaded = createAppAsyncThunk(
'api/imageUploaded',
async (arg: ImageUploadedArg) => {
const response = await ImagesService.uploadImage(arg);
// strip out `activeTabName` from arg - the route does not need it
const { activeTabName, ...rest } = arg;
const response = await ImagesService.uploadImage(rest);
const { location } = getHeaders(response);
imagesLog.info(