Merge pull request #6366 from menloresearch/fix/search-model-hub

fix: search hgf repo and downloaded filter

web-app/src/routes/hub/index.tsx

@@ -194,35 +194,30 @@ function Hub() {
     fetchSources()
   }, [fetchSources])
 
-  const handleSearchChange = (e: ChangeEvent<HTMLInputElement>) => {
-    setIsSearching(false)
-    setSearchValue(e.target.value)
-    setHuggingFaceRepo(null) // Clear previous repo info
+  const fetchHuggingFaceModel = async (searchValue: string) => {
+    if (
+      !searchValue.length ||
+      (!searchValue.includes('/') && !searchValue.startsWith('http'))
+    ) {
+      return
+    }
 
+    setIsSearching(true)
     if (addModelSourceTimeoutRef.current) {
       clearTimeout(addModelSourceTimeoutRef.current)
     }
 
-    if (
-      e.target.value.length &&
-      (e.target.value.includes('/') || e.target.value.startsWith('http'))
-    ) {
-      setIsSearching(true)
-
     addModelSourceTimeoutRef.current = setTimeout(async () => {
       try {
-          // Fetch HuggingFace repository information
-          const repoInfo = await fetchHuggingFaceRepo(
-            e.target.value,
-            huggingfaceToken
-          )
+        const repoInfo = await fetchHuggingFaceRepo(searchValue, huggingfaceToken)
         if (repoInfo) {
           const catalogModel = convertHfRepoToCatalogModel(repoInfo)
           if (
             !sources.some(
               (s) =>
                 catalogModel.model_name.trim().split('/').pop() ===
-                  s.model_name.trim()
+                  s.model_name.trim() &&
+                catalogModel.developer.trim() === s.developer?.trim()
             )
           ) {
             setHuggingFaceRepo(catalogModel)
@@ -235,6 +230,15 @@ function Hub() {
       }
     }, 500)
   }
+
+  const handleSearchChange = (e: ChangeEvent<HTMLInputElement>) => {
+    setIsSearching(false)
+    setSearchValue(e.target.value)
+    setHuggingFaceRepo(null) // Clear previous repo info
+
+    if (!showOnlyDownloaded) {
+      fetchHuggingFaceModel(e.target.value)
+    }
   }
 
   const { downloads, localDownloadingModels, addLocalDownloadingModel } =
@@ -508,7 +512,15 @@ function Hub() {
         <div className="flex items-center gap-2">
           <Switch
             checked={showOnlyDownloaded}
-            onCheckedChange={setShowOnlyDownloaded}
+            onCheckedChange={(checked) => {
+              setShowOnlyDownloaded(checked)
+              if (checked) {
+                setHuggingFaceRepo(null)
+              } else {
+                // Re-trigger HuggingFace search when switching back to "All models"
+                fetchHuggingFaceModel(searchValue)
+              }
+            }}
           />
           <span className="text-xs text-main-view-fg/70 font-medium whitespace-nowrap">
             {t('hub:downloaded')}