Hide empty folders when searching in model library (#1382)

Comfy-Org · Oct 30, 2024 · e48c785 · e48c785
1 parent bf7a9bf
commit e48c785
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/src/components/sidebar/tabs/ModelLibrarySidebarTab.vue b/src/components/sidebar/tabs/ModelLibrarySidebarTab.vue
@@ -94,10 +94,9 @@ const handleSearch = async (query: string) => {
 type ModelOrFolder = ComfyModelDef | ModelFolder
 
 const root = computed<TreeNode>(() => {
-  const allNodes: ModelOrFolder[] = [
-    ...modelStore.modelFolders,
-    ...(searchQuery.value ? filteredModels.value : modelStore.models)
-  ]
+  const allNodes: ModelOrFolder[] = searchQuery.value
+    ? filteredModels.value
+    : [...modelStore.modelFolders, ...modelStore.models]
   return buildTree(allNodes, (modelOrFolder: ModelOrFolder) =>
     modelOrFolder.key.split('/')
   )