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

fix(ui): fix uploads not loading more images correctly after generation

Browse Source
This commit is contained in:
psychedelicious 2023-05-26 16:44:13 +10:00
parent 7b0938e7e4
commit 96b4d35d43
1 changed files with 11 additions and 2 deletions

13
invokeai/frontend/web/src/features/gallery/store/uploadsSlice.ts

@ -1,4 +1,8 @@
import { createEntityAdapter, createSlice } from '@reduxjs/toolkit';
import {
PayloadAction,
createEntityAdapter,
createSlice,
} from '@reduxjs/toolkit';
import { RootState } from 'app/store/store';
import {
@ -22,6 +26,7 @@ type AdditionalUploadsState = {
pages: number;
isLoading: boolean;
nextPage: number;
upsertedImageCount: number;
};
export const initialUploadsState =
@ -30,6 +35,7 @@ export const initialUploadsState =
pages: 0,
nextPage: 0,
isLoading: false,
upsertedImageCount: 0,
});
export type UploadsState = typeof initialUploadsState;
@ -38,7 +44,10 @@ const uploadsSlice = createSlice({
name: 'uploads',
initialState: initialUploadsState,
reducers: {
uploadUpserted: uploadsAdapter.upsertOne,
uploadUpserted: (state, action: PayloadAction<UploadsImageDTO>) => {
uploadsAdapter.upsertOne(state, action.payload);
state.upsertedImageCount += 1;
},
},
extraReducers: (builder) => {
/**