From ee5a22ee04e36135d8f4e5676b8852f17d2bed65 Mon Sep 17 00:00:00 2001
From: DrJKL <DrJKL0424@gmail.com>
Date: Mon, 1 Dec 2025 17:08:16 -0800
Subject: [PATCH] Feat: Add model import button to Vue model list popover.

---
 .../assets/components/AssetBrowserModal.vue   | 30 ++--------------
 .../assets/composables/useModelUpload.ts      | 35 +++++++++++++++++++
 .../form/dropdown/FormDropdownMenuFilter.vue  | 16 ++++++++-
 3 files changed, 53 insertions(+), 28 deletions(-)
 create mode 100644 src/platform/assets/composables/useModelUpload.ts

diff --git a/src/platform/assets/components/AssetBrowserModal.vue b/src/platform/assets/components/AssetBrowserModal.vue
index 381a20b73fe..d6cd2302775 100644
--- a/src/platform/assets/components/AssetBrowserModal.vue
+++ b/src/platform/assets/components/AssetBrowserModal.vue
@@ -35,7 +35,7 @@
           size="md"
           class="!h-10 [&>span]:hidden md:[&>span]:inline"
           :label="$t('assetBrowser.uploadModel')"
-          :on-click="handleUploadClick"
+          :on-click="showUploadDialog"
         >
           <template #icon>
             <i class="icon-[lucide--package-plus]" />
@@ -70,17 +70,14 @@ import IconTextButton from '@/components/button/IconTextButton.vue'
 import SearchBox from '@/components/input/SearchBox.vue'
 import BaseModalLayout from '@/components/widget/layout/BaseModalLayout.vue'
 import LeftSidePanel from '@/components/widget/panel/LeftSidePanel.vue'
-import { useFeatureFlags } from '@/composables/useFeatureFlags'
 import AssetFilterBar from '@/platform/assets/components/AssetFilterBar.vue'
 import AssetGrid from '@/platform/assets/components/AssetGrid.vue'
-import UploadModelDialog from '@/platform/assets/components/UploadModelDialog.vue'
-import UploadModelDialogHeader from '@/platform/assets/components/UploadModelDialogHeader.vue'
 import type { AssetDisplayItem } from '@/platform/assets/composables/useAssetBrowser'
 import { useAssetBrowser } from '@/platform/assets/composables/useAssetBrowser'
+import { useModelUpload } from '@/platform/assets/composables/useModelUpload'
 import type { AssetItem } from '@/platform/assets/schemas/assetSchema'
 import { assetService } from '@/platform/assets/services/assetService'
 import { formatCategoryLabel } from '@/platform/assets/utils/categoryLabel'
-import { useDialogStore } from '@/stores/dialogStore'
 import { useModelToNodeStore } from '@/stores/modelToNodeStore'
 import { OnCloseKey } from '@/types/widgetTypes'
 
@@ -95,7 +92,6 @@ const props = defineProps<{
 }>()
 
 const { t } = useI18n()
-const dialogStore = useDialogStore()
 
 const emit = defineEmits<{
   'asset-select': [asset: AssetDisplayItem]
@@ -189,25 +185,5 @@ function handleAssetSelectAndEmit(asset: AssetDisplayItem) {
   props.onSelect?.(asset)
 }
 
-const { flags } = useFeatureFlags()
-const isUploadButtonEnabled = computed(() => flags.modelUploadButtonEnabled)
-
-function handleUploadClick() {
-  dialogStore.showDialog({
-    key: 'upload-model',
-    headerComponent: UploadModelDialogHeader,
-    component: UploadModelDialog,
-    props: {
-      onUploadSuccess: async () => {
-        await execute()
-      }
-    },
-    dialogComponentProps: {
-      pt: {
-        header: 'py-0! pl-0!',
-        content: 'p-0!'
-      }
-    }
-  })
-}
+const { isUploadButtonEnabled, showUploadDialog } = useModelUpload(execute)
 </script>
diff --git a/src/platform/assets/composables/useModelUpload.ts b/src/platform/assets/composables/useModelUpload.ts
new file mode 100644
index 00000000000..e765d5c55dd
--- /dev/null
+++ b/src/platform/assets/composables/useModelUpload.ts
@@ -0,0 +1,35 @@
+import { useFeatureFlags } from '@/composables/useFeatureFlags'
+import UploadModelDialog from '@/platform/assets/components/UploadModelDialog.vue'
+import UploadModelDialogHeader from '@/platform/assets/components/UploadModelDialogHeader.vue'
+import type { AssetItem } from '@/platform/assets/schemas/assetSchema'
+import { useDialogStore } from '@/stores/dialogStore'
+import type { UseAsyncStateReturn } from '@vueuse/core'
+import { computed } from 'vue'
+
+export function useModelUpload(
+  execute?: UseAsyncStateReturn<AssetItem[], [], true>['execute']
+) {
+  const dialogStore = useDialogStore()
+  const { flags } = useFeatureFlags()
+  const isUploadButtonEnabled = computed(() => flags.modelUploadButtonEnabled)
+
+  function showUploadDialog() {
+    dialogStore.showDialog({
+      key: 'upload-model',
+      headerComponent: UploadModelDialogHeader,
+      component: UploadModelDialog,
+      props: {
+        onUploadSuccess: async () => {
+          await execute?.()
+        }
+      },
+      dialogComponentProps: {
+        pt: {
+          header: 'py-0! pl-0!',
+          content: 'p-0!'
+        }
+      }
+    })
+  }
+  return { isUploadButtonEnabled, showUploadDialog }
+}
diff --git a/src/renderer/extensions/vueNodes/widgets/components/form/dropdown/FormDropdownMenuFilter.vue b/src/renderer/extensions/vueNodes/widgets/components/form/dropdown/FormDropdownMenuFilter.vue
index db8fc9cff3d..3af54edb9d1 100644
--- a/src/renderer/extensions/vueNodes/widgets/components/form/dropdown/FormDropdownMenuFilter.vue
+++ b/src/renderer/extensions/vueNodes/widgets/components/form/dropdown/FormDropdownMenuFilter.vue
@@ -1,4 +1,6 @@
 <script setup lang="ts">
+import IconTextButton from '@/components/button/IconTextButton.vue'
+import { useModelUpload } from '@/platform/assets/composables/useModelUpload'
 import { cn } from '@/utils/tailwindUtil'
 
 import type { FilterOption, OptionId } from './types'
@@ -8,10 +10,12 @@ defineProps<{
 }>()
 
 const filterSelected = defineModel<OptionId>('filterSelected')
+
+const { isUploadButtonEnabled, showUploadDialog } = useModelUpload()
 </script>
 
 <template>
-  <div class="text-secondary mb-4 flex gap-1 px-4">
+  <div class="text-secondary mb-4 flex gap-1 px-4 justify-between">
     <div
       v-for="option in filterOptions"
       :key="option.id"
@@ -30,5 +34,15 @@ const filterSelected = defineModel<OptionId>('filterSelected')
     >
       {{ option.name }}
     </div>
+    <IconTextButton
+      v-if="isUploadButtonEnabled"
+      :label="$t('g.import')"
+      type="secondary"
+      @click="showUploadDialog"
+    >
+      <template #icon>
+        <i class="icon-[lucide--folder-input]" />
+      </template>
+    </IconTextButton>
   </div>
 </template>