From 5b66baa3ec6eae2c6700cba2da7f94edfb930d9b Mon Sep 17 00:00:00 2001
From: Mary Hipp <maryhipp@Marys-MacBook-Air.local>
Date: Thu, 22 Feb 2024 11:50:16 -0500
Subject: [PATCH] basic scan working and renders results

---
 .../subpanels/AddModelPanel/ImportQueue.tsx   |  2 +-
 .../AddModelPanel/ScanModels/ScanModels.tsx   |  5 ++
 .../ScanModels/ScanModelsForm.tsx             | 54 +++++++++++++
 .../ScanModels/ScanModelsResults.tsx          | 77 +++++++++++++++++++
 .../modelManagerV2/subpanels/ImportModels.tsx |  7 +-
 .../web/src/services/api/endpoints/models.ts  |  1 +
 6 files changed, 143 insertions(+), 3 deletions(-)
 create mode 100644 invokeai/frontend/web/src/features/modelManagerV2/subpanels/AddModelPanel/ScanModels/ScanModels.tsx
 create mode 100644 invokeai/frontend/web/src/features/modelManagerV2/subpanels/AddModelPanel/ScanModels/ScanModelsForm.tsx
 create mode 100644 invokeai/frontend/web/src/features/modelManagerV2/subpanels/AddModelPanel/ScanModels/ScanModelsResults.tsx

diff --git a/invokeai/frontend/web/src/features/modelManagerV2/subpanels/AddModelPanel/ImportQueue.tsx b/invokeai/frontend/web/src/features/modelManagerV2/subpanels/AddModelPanel/ImportQueue.tsx
index 3d80d11c29..804a728cbd 100644
--- a/invokeai/frontend/web/src/features/modelManagerV2/subpanels/AddModelPanel/ImportQueue.tsx
+++ b/invokeai/frontend/web/src/features/modelManagerV2/subpanels/AddModelPanel/ImportQueue.tsx
@@ -7,7 +7,7 @@ import { useCallback, useMemo } from 'react';
 import { RiSparklingFill } from 'react-icons/ri';
 import { useGetModelImportsQuery, usePruneModelImportsMutation } from 'services/api/endpoints/models';
 
-import { ImportQueueModel } from '../ImportQueueModel';
+import { ImportQueueModel } from './ImportQueueModel';
 
 export const ImportQueue = () => {
   const dispatch = useAppDispatch();
diff --git a/invokeai/frontend/web/src/features/modelManagerV2/subpanels/AddModelPanel/ScanModels/ScanModels.tsx b/invokeai/frontend/web/src/features/modelManagerV2/subpanels/AddModelPanel/ScanModels/ScanModels.tsx
new file mode 100644
index 0000000000..02eb3b50ba
--- /dev/null
+++ b/invokeai/frontend/web/src/features/modelManagerV2/subpanels/AddModelPanel/ScanModels/ScanModels.tsx
@@ -0,0 +1,5 @@
+import { ScanModelsForm } from './ScanModelsForm';
+
+export const ScanModels = () => {
+  return <ScanModelsForm />;
+};
diff --git a/invokeai/frontend/web/src/features/modelManagerV2/subpanels/AddModelPanel/ScanModels/ScanModelsForm.tsx b/invokeai/frontend/web/src/features/modelManagerV2/subpanels/AddModelPanel/ScanModels/ScanModelsForm.tsx
new file mode 100644
index 0000000000..48b8bb0dda
--- /dev/null
+++ b/invokeai/frontend/web/src/features/modelManagerV2/subpanels/AddModelPanel/ScanModels/ScanModelsForm.tsx
@@ -0,0 +1,54 @@
+import { Flex, FormControl, FormLabel, Input, Button, FormErrorMessage, Divider } from '@invoke-ai/ui-library';
+import { ChangeEventHandler, useCallback, useState } from 'react';
+import { useLazyScanModelsQuery } from '../../../../../services/api/endpoints/models';
+import { useTranslation } from 'react-i18next';
+import { ScanModelsResults } from './ScanModelsResults';
+
+export const ScanModelsForm = () => {
+  const [scanPath, setScanPath] = useState('');
+  const [errorMessage, setErrorMessage] = useState('');
+  const [results, setResults] = useState<string[] | undefined>();
+  const { t } = useTranslation();
+
+  const [_scanModels, { isLoading }] = useLazyScanModelsQuery();
+
+  const handleSubmitScan = useCallback(async () => {
+    _scanModels({ scan_path: scanPath })
+      .unwrap()
+      .then((result) => {
+        setResults(result);
+      })
+      .catch((error) => {
+        if (error) {
+          setErrorMessage(error.data.detail);
+        }
+      });
+  }, [scanPath]);
+
+  const handleSetScanPath: ChangeEventHandler<HTMLInputElement> = useCallback((e) => {
+    setScanPath(e.target.value);
+    setErrorMessage('');
+  }, []);
+
+  return (
+    <>
+      <FormControl isInvalid={!!errorMessage.length} w="full">
+        <Flex flexDir="column" w="full">
+          <Flex gap={2} alignItems="flex-end" justifyContent="space-between">
+            <Flex direction="column" w="full">
+              <FormLabel>{t('common.folder')}</FormLabel>
+              <Input value={scanPath} onChange={handleSetScanPath} />
+            </Flex>
+
+            <Button onClick={handleSubmitScan} isLoading={isLoading} isDisabled={scanPath.length === 0}>
+              {t('modelManager.scanFolder')}
+            </Button>
+          </Flex>
+          {!!errorMessage.length && <FormErrorMessage>{errorMessage}</FormErrorMessage>}
+        </Flex>
+      </FormControl>
+
+      {results && <ScanModelsResults results={results} />}
+    </>
+  );
+};
diff --git a/invokeai/frontend/web/src/features/modelManagerV2/subpanels/AddModelPanel/ScanModels/ScanModelsResults.tsx b/invokeai/frontend/web/src/features/modelManagerV2/subpanels/AddModelPanel/ScanModels/ScanModelsResults.tsx
new file mode 100644
index 0000000000..9bccf8a0d7
--- /dev/null
+++ b/invokeai/frontend/web/src/features/modelManagerV2/subpanels/AddModelPanel/ScanModels/ScanModelsResults.tsx
@@ -0,0 +1,77 @@
+import {
+  Text,
+  Flex,
+  Heading,
+  IconButton,
+  Input,
+  InputGroup,
+  InputRightElement,
+  Divider,
+  Box,
+} from '@invoke-ai/ui-library';
+import { t } from 'i18next';
+import { ChangeEventHandler, useCallback, useMemo, useState } from 'react';
+import { PiXBold } from 'react-icons/pi';
+
+export const ScanModelsResults = ({ results }: { results: string[] }) => {
+  const [searchTerm, setSearchTerm] = useState('');
+  const [filteredResults, setFilteredResults] = useState(results);
+
+  const handleSearch: ChangeEventHandler<HTMLInputElement> = useCallback(
+    (e) => {
+      setSearchTerm(e.target.value);
+      setFilteredResults(
+        results.filter((result) => {
+          const modelName = result.split('\\').slice(-1)[0];
+          return modelName?.includes(e.target.value);
+        })
+      );
+    },
+    [results]
+  );
+
+  const clearSearch = useCallback(() => {
+    setSearchTerm('');
+  }, []);
+
+  return (
+    <>
+      <Divider mt={4} />
+      <Flex flexDir="column" gap={2} mt={4}>
+        <Flex justifyContent="space-between" alignItems="center" mb={4}>
+          <Heading fontSize="md" as="h4">
+            Scan Results
+          </Heading>
+          <InputGroup maxW="300px" size="xs">
+            <Input
+              placeholder={t('modelManager.search')}
+              value={searchTerm || ''}
+              data-testid="board-search-input"
+              onChange={handleSearch}
+              size="xs"
+            />
+
+            {!!searchTerm?.length && (
+              <InputRightElement h="full" pe={2}>
+                <IconButton
+                  size="sm"
+                  variant="link"
+                  aria-label={t('boards.clearSearch')}
+                  icon={<PiXBold />}
+                  onClick={clearSearch}
+                />
+              </InputRightElement>
+            )}
+          </InputGroup>
+        </Flex>
+
+        {filteredResults.map((result) => (
+          <Flex key={result} fontSize="sm" flexDir="column">
+            <Text fontWeight="semibold">{result.split('\\').slice(-1)[0]}</Text>
+            <Text variant="subtext">{result}</Text>
+          </Flex>
+        ))}
+      </Flex>
+    </>
+  );
+};
diff --git a/invokeai/frontend/web/src/features/modelManagerV2/subpanels/ImportModels.tsx b/invokeai/frontend/web/src/features/modelManagerV2/subpanels/ImportModels.tsx
index a0dd39de2e..88423758e6 100644
--- a/invokeai/frontend/web/src/features/modelManagerV2/subpanels/ImportModels.tsx
+++ b/invokeai/frontend/web/src/features/modelManagerV2/subpanels/ImportModels.tsx
@@ -2,6 +2,7 @@ import { Box, Divider, Heading, Tab, TabList, TabPanel, TabPanels, Tabs } from '
 
 import { ImportQueue } from './AddModelPanel/ImportQueue';
 import { SimpleImport } from './AddModelPanel/SimpleImport';
+import { ScanModels } from './AddModelPanel/ScanModels/ScanModels';
 
 export const ImportModels = () => {
   return (
@@ -9,7 +10,7 @@ export const ImportModels = () => {
       <Box w="full" p={4}>
         <Heading fontSize="xl">Add Model</Heading>
       </Box>
-      <Box layerStyle="second" borderRadius="base" w="full" h="100vh">
+      <Box layerStyle="second" borderRadius="base" w="full">
         <Tabs variant="collapse">
           <TabList>
             <Tab>Simple</Tab>
@@ -21,7 +22,9 @@ export const ImportModels = () => {
               <SimpleImport />
             </TabPanel>
             <TabPanel>Advanced Import Placeholder</TabPanel>
-            <TabPanel>Scan Models Placeholder</TabPanel>
+            <TabPanel>
+              <ScanModels />
+            </TabPanel>
           </TabPanels>
         </Tabs>
 
diff --git a/invokeai/frontend/web/src/services/api/endpoints/models.ts b/invokeai/frontend/web/src/services/api/endpoints/models.ts
index a1cdb43617..61f0975920 100644
--- a/invokeai/frontend/web/src/services/api/endpoints/models.ts
+++ b/invokeai/frontend/web/src/services/api/endpoints/models.ts
@@ -349,6 +349,7 @@ export const {
   useMergeMainModelsMutation,
   useSyncModelsMutation,
   useScanModelsQuery,
+  useLazyScanModelsQuery,
   useGetCheckpointConfigsQuery,
   useGetModelImportsQuery,
   useGetModelMetadataQuery,