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

ui(model_manager): Remember Scan Path

Browse Source
This commit is contained in:
blessedcoolant 2024-03-08 19:54:30 +05:30 committed by Mary Hipp Rogers
parent 281222df3c
commit a42812d78d
2 changed files with 22 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
invokeai/frontend/web/src/features/modelManagerV2/store/modelManagerV2Slice.ts
View File

@ -8,6 +8,7 @@ type ModelManagerState = {
selectedModelMode: 'edit' | 'view';
searchTerm: string;
filteredModelType: string | null;
scanPath: string | undefined;
};
const initialModelManagerState: ModelManagerState = {
@ -16,6 +17,7 @@ const initialModelManagerState: ModelManagerState = {
selectedModelMode: 'view',
filteredModelType: null,
searchTerm: '',
scanPath: undefined,
};
export const modelManagerV2Slice = createSlice({
@ -36,10 +38,13 @@ export const modelManagerV2Slice = createSlice({
setFilteredModelType: (state, action: PayloadAction<string | null>) => {
state.filteredModelType = action.payload;
},
setScanPath: (state, action: PayloadAction<string | undefined>) => {
state.scanPath = action.payload;
},
},
});
export const { setSelectedModelKey, setSearchTerm, setFilteredModelType, setSelectedModelMode } =
export const { setSelectedModelKey, setSearchTerm, setFilteredModelType, setSelectedModelMode, setScanPath } =
modelManagerV2Slice.actions;
/* eslint-disable-next-line @typescript-eslint/no-explicit-any */

22
invokeai/frontend/web/src/features/modelManagerV2/subpanels/AddModelPanel/ScanFolder/ScanFolderForm.tsx
View File

@ -1,4 +1,6 @@
import { Button, Flex, FormControl, FormErrorMessage, FormLabel, Input } from '@invoke-ai/ui-library';
import { useAppDispatch, useAppSelector } from 'app/store/storeHooks';
import { setScanPath } from 'features/modelManagerV2/store/modelManagerV2Slice';
import type { ChangeEventHandler } from 'react';
import { useCallback, useState } from 'react';
import { useTranslation } from 'react-i18next';
@ -7,7 +9,8 @@ import { useLazyScanFolderQuery } from 'services/api/endpoints/models';
import { ScanModelsResults } from './ScanFolderResults';
export const ScanModelsForm = () => {
const [scanPath, setScanPath] = useState('');
const scanPath = useAppSelector((state) => state.modelmanagerV2.scanPath);
const dispatch = useAppDispatch();
const [errorMessage, setErrorMessage] = useState('');
const { t } = useTranslation();
@ -21,10 +24,13 @@ export const ScanModelsForm = () => {
});
}, [_scanFolder, scanPath]);
const handleSetScanPath: ChangeEventHandler<HTMLInputElement> = useCallback((e) => {
setScanPath(e.target.value);
setErrorMessage('');
}, []);
const handleSetScanPath: ChangeEventHandler<HTMLInputElement> = useCallback(
(e) => {
dispatch(setScanPath(e.target.value));
setErrorMessage('');
},
[dispatch]
);
return (
<Flex flexDir="column" height="100%">
@ -36,7 +42,11 @@ export const ScanModelsForm = () => {
<Input value={scanPath} onChange={handleSetScanPath} />
</Flex>
<Button onClick={scanFolder} isLoading={isLoading} isDisabled={scanPath.length === 0}>
<Button
onClick={scanFolder}
isLoading={isLoading}
isDisabled={scanPath === undefined || scanPath.length === 0}
>
{t('modelManager.scanFolder')}
</Button>
</Flex>