From 702efc5eaf980bf1262a3a53b1400fbae9dd6fd9 Mon Sep 17 00:00:00 2001
From: Tony Salomone <dadmobile@gmail.com>
Date: Mon, 27 May 2024 14:11:53 -0400
Subject: [PATCH] Pass selected folder to backend when searching for models.

---
 src/renderer/components/ModelZoo/ImportModelsBar.tsx   | 1 -
 src/renderer/components/ModelZoo/ImportModelsModal.tsx | 8 ++++----
 src/renderer/lib/transformerlab-api-sdk.ts             | 3 ++-
 3 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/src/renderer/components/ModelZoo/ImportModelsBar.tsx b/src/renderer/components/ModelZoo/ImportModelsBar.tsx
index c87bd7a..d606b18 100644
--- a/src/renderer/components/ModelZoo/ImportModelsBar.tsx
+++ b/src/renderer/components/ModelZoo/ImportModelsBar.tsx
@@ -89,7 +89,6 @@ export default function ModelDetails({}) {
             <Button
               size="sm"
               sx={{ height: '30px' }}
-              disabled={process.env.NODE_ENV != "development"}
               endDecorator={<PlusIcon />}
               onClick={() => {
                 setImportModelsModalOpen(true);
diff --git a/src/renderer/components/ModelZoo/ImportModelsModal.tsx b/src/renderer/components/ModelZoo/ImportModelsModal.tsx
index e66a68d..1a72272 100644
--- a/src/renderer/components/ModelZoo/ImportModelsModal.tsx
+++ b/src/renderer/components/ModelZoo/ImportModelsModal.tsx
@@ -25,14 +25,14 @@ const fetcher = (url) => fetch(url).then((res) => res.json());
 
 export default function ImportModelsModal({ open, setOpen}) {
     const [importing, setImporting] = useState(false);
-    const [modelFolder, setModelFolder] = useState(null); // Disabled folder support
+    const [modelFolder, setModelFolder] = useState("");
 
     const {
         data: modelsData,
         error: modelsError,
         isLoading: isLoading,
     } = useSWR(
-        chatAPI.Endpoints.Models.GetLocalUninstalled(),
+        chatAPI.Endpoints.Models.SearchForLocalUninstalledModels(modelFolder),
         fetcher
     );
     const models = modelsData?.data;
@@ -166,12 +166,12 @@ export default function ImportModelsModal({ open, setOpen}) {
                       const webkitRelativePath = firstfile.webkitRelativePath;
                       const parentPath = firstfilepath.slice(0, -1*webkitRelativePath.length);
                       const topRelativePathDir = webkitRelativePath.split('/')[0];
-                      const fullPath = parentPath + '/' + topRelativePathDir;
+                      const fullPath = parentPath + topRelativePathDir;
                       console.log(firstfile);
                       console.log(webkitRelativePath);
                       setModelFolder(fullPath);
                     } else {
-                      setModelFolder(null);
+                      setModelFolder("");
                     }
                 }}
                 />
diff --git a/src/renderer/lib/transformerlab-api-sdk.ts b/src/renderer/lib/transformerlab-api-sdk.ts
index 1caae54..d735eff 100644
--- a/src/renderer/lib/transformerlab-api-sdk.ts
+++ b/src/renderer/lib/transformerlab-api-sdk.ts
@@ -608,7 +608,8 @@ Endpoints.Models = {
     API_URL() + 'model/details/' + convertSlashInUrl(modelId),
   GetLocalHFConfig: (modelId: string) =>
     API_URL() + 'model/get_local_hfconfig?model_id=' + modelId,
-  GetLocalUninstalled: () => API_URL() + 'model/list_local_uninstalled',
+  SearchForLocalUninstalledModels: (path: string) =>
+    API_URL() + 'model/list_local_uninstalled?path=' + path,
   ImportLocal: (modelSource: string, modelId: string) =>
     API_URL() +
     'model/import_local?model_source=' +