From 5cbc51232243607ece7b50b2e65e2ee3811d422d Mon Sep 17 00:00:00 2001
From: ItzCrazyKns <95534749+ItzCrazyKns@users.noreply.github.com>
Date: Sat, 15 Feb 2025 11:29:59 +0530
Subject: [PATCH] feat(app): add auto video & image search

---
 ui/components/ChatWindow.tsx   | 30 ++++++++++++++++++++++--------
 ui/components/SearchImages.tsx |  1 +
 ui/components/SearchVideos.tsx |  1 +
 3 files changed, 24 insertions(+), 8 deletions(-)

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);