From 5b30091c83b4a84910bbcbc0015c8cf9a56c5982 Mon Sep 17 00:00:00 2001
From: Nicolas Mowen <nickmowen213@gmail.com>
Date: Wed, 11 Sep 2024 08:35:50 -0600
Subject: [PATCH] Fix similarity search

---
 web/src/components/filter/SearchFilterGroup.tsx |  2 +-
 web/src/pages/Explore.tsx                       | 12 ++++++++----
 2 files changed, 9 insertions(+), 5 deletions(-)

diff --git a/web/src/components/filter/SearchFilterGroup.tsx b/web/src/components/filter/SearchFilterGroup.tsx
index 7471b14c3..261738d82 100644
--- a/web/src/components/filter/SearchFilterGroup.tsx
+++ b/web/src/components/filter/SearchFilterGroup.tsx
@@ -954,7 +954,7 @@ export function SearchTypeContent({
           <Button
             onClick={() => {
               updateSearchSourceFilter(undefined);
-              setCurrentSearchSources(undefined);
+              setCurrentSearchSources(["thumbnail", "description"]);
             }}
           >
             Reset
diff --git a/web/src/pages/Explore.tsx b/web/src/pages/Explore.tsx
index 137c6c68c..de13a9ee0 100644
--- a/web/src/pages/Explore.tsx
+++ b/web/src/pages/Explore.tsx
@@ -31,21 +31,25 @@ export default function Explore() {
 
   // search filter
 
+  const [similaritySearch, setSimilaritySearch] =
+    useState<PartialSearchResult>();
+
   const [searchFilter, setSearchFilter, searchSearchParams] =
     useApiFilterArgs<SearchFilter>();
 
   const onUpdateFilter = useCallback(
     (newFilter: SearchFilter) => {
       setSearchFilter(newFilter);
+
+      if (similaritySearch && !newFilter.search_type?.includes("similarity")) {
+        setSimilaritySearch(undefined);
+      }
     },
-    [setSearchFilter],
+    [similaritySearch, setSearchFilter],
   );
 
   // search api
 
-  const [similaritySearch, setSimilaritySearch] =
-    useState<PartialSearchResult>();
-
   const updateFilterWithSimilarity = useCallback(
     (similaritySearch: PartialSearchResult) => {
       let newFilter = searchFilter;