diff --git a/ui/components/ChatWindow.tsx b/ui/components/ChatWindow.tsx
index b26573f..33ef27b 100644
--- a/ui/components/ChatWindow.tsx
+++ b/ui/components/ChatWindow.tsx
@@ -58,14 +58,17 @@ const useSocket = (
         let embeddingModelProvider = localStorage.getItem(
           'embeddingModelProvider',
         );
-        let openAIBaseURL =
-          chatModelProvider === 'custom_openai'
-            ? localStorage.getItem('openAIBaseURL')
-            : null;
-        let openAIPIKey =
-          chatModelProvider === 'custom_openai'
-            ? localStorage.getItem('openAIApiKey')
-            : null;
+
+        const autoImageSearch = localStorage.getItem('autoImageSearch');
+        const autoVideoSearch = localStorage.getItem('autoVideoSearch');
+
+        if (!autoImageSearch) {
+          localStorage.setItem('autoImageSearch', 'true');
+        }
+
+        if (!autoVideoSearch) {
+          localStorage.setItem('autoVideoSearch', 'false');
+        }
 
         const providers = await fetch(
           `${process.env.NEXT_PUBLIC_API_URL}/models`,
@@ -595,6 +598,17 @@ const ChatWindow = ({ id }: { id?: string }) => {
             }),
           );
         }
+
+        const autoImageSearch = localStorage.getItem('autoImageSearch');
+        const autoVideoSearch = localStorage.getItem('autoVideoSearch');
+
+        if (autoImageSearch === 'true') {
+          document.getElementById('search-images')?.click();
+        }
+
+        if (autoVideoSearch === 'true') {
+          document.getElementById('search-videos')?.click();
+        }
       }
     };
 
diff --git a/ui/components/SearchImages.tsx b/ui/components/SearchImages.tsx
index b083af7..383f780 100644
--- a/ui/components/SearchImages.tsx
+++ b/ui/components/SearchImages.tsx
@@ -27,6 +27,7 @@ const SearchImages = ({
     <>
       {!loading && images === null && (
         <button
+          id="search-images"
           onClick={async () => {
             setLoading(true);
 
diff --git a/ui/components/SearchVideos.tsx b/ui/components/SearchVideos.tsx
index 170df61..c284dc2 100644
--- a/ui/components/SearchVideos.tsx
+++ b/ui/components/SearchVideos.tsx
@@ -42,6 +42,7 @@ const Searchvideos = ({
     <>
       {!loading && videos === null && (
         <button
+          id="search-videos"
           onClick={async () => {
             setLoading(true);