From be0da443ef4deaae5a5b0b4bd1015b85f78564b0 Mon Sep 17 00:00:00 2001
From: Mehdi Essoussi <messoussi@30fe.com>
Date: Mon, 9 Dec 2024 11:03:38 -0500
Subject: [PATCH] Added delete-all-chats button that deletes all chats for the
 user

---
 app/api/chat/google/route.ts                  |  1 -
 components/chat/chat-helpers/index.ts         |  7 +-
 .../sidebar/items/chat/delete-all-chats.tsx   | 73 +++++++++++++++++++
 components/utility/profile-settings.tsx       | 12 ++-
 db/chats.ts                                   | 10 +++
 db/files.ts                                   |  5 +-
 lib/build-prompt.ts                           | 54 +++++++-------
 lib/chat-setting-limits.ts                    |  2 +-
 lib/models/llm/google-llm-list.ts             |  7 +-
 9 files changed, 136 insertions(+), 35 deletions(-)
 create mode 100644 components/sidebar/items/chat/delete-all-chats.tsx

diff --git a/app/api/chat/google/route.ts b/app/api/chat/google/route.ts
index ad79139646..cb9820ee88 100644
--- a/app/api/chat/google/route.ts
+++ b/app/api/chat/google/route.ts
@@ -44,7 +44,6 @@ export async function POST(request: Request) {
     return new Response(readableStream, {
       headers: { "Content-Type": "text/plain" }
     })
-
   } catch (error: any) {
     let errorMessage = error.message || "An unexpected error occurred"
     const errorCode = error.status || 500
diff --git a/components/chat/chat-helpers/index.ts b/components/chat/chat-helpers/index.ts
index 17a2089638..ec4a548cdc 100644
--- a/components/chat/chat-helpers/index.ts
+++ b/components/chat/chat-helpers/index.ts
@@ -208,9 +208,12 @@ export const handleHostedChat = async (
 
   let draftMessages = await buildFinalMessages(payload, profile, chatImages)
 
-  let formattedMessages : any[] = []
+  let formattedMessages: any[] = []
   if (provider === "google") {
-    formattedMessages = await adaptMessagesForGoogleGemini(payload, draftMessages)
+    formattedMessages = await adaptMessagesForGoogleGemini(
+      payload,
+      draftMessages
+    )
   } else {
     formattedMessages = draftMessages
   }
diff --git a/components/sidebar/items/chat/delete-all-chats.tsx b/components/sidebar/items/chat/delete-all-chats.tsx
new file mode 100644
index 0000000000..f1bd467edf
--- /dev/null
+++ b/components/sidebar/items/chat/delete-all-chats.tsx
@@ -0,0 +1,73 @@
+import { useChatHandler } from "@/components/chat/chat-hooks/use-chat-handler"
+import { Button } from "@/components/ui/button"
+import {
+  Dialog,
+  DialogContent,
+  DialogDescription,
+  DialogFooter,
+  DialogHeader,
+  DialogTitle,
+  DialogTrigger
+} from "@/components/ui/dialog"
+import { ChatbotUIContext } from "@/context/context"
+import { deleteAllChats } from "@/db/chats"
+import { Tables } from "@/supabase/types"
+import { IconTrash } from "@tabler/icons-react"
+import { FC, useContext, useRef, useState } from "react"
+import { toast } from "sonner"
+
+interface DeleteAllChatsProps {
+  profile: Tables<"profiles">
+}
+
+export const DeleteAllChats: FC<DeleteAllChatsProps> = ({ profile }) => {
+  const { setChats, setChatMessages, setSelectedChat } =
+    useContext(ChatbotUIContext)
+  const { handleNewChat } = useChatHandler()
+
+  const buttonRef = useRef<HTMLButtonElement>(null)
+  const [showChatDialog, setShowChatDialog] = useState(false)
+
+  const handleDeleteAllChats = async () => {
+    if (!profile) return
+
+    await deleteAllChats(profile.user_id)
+
+    setChats([])
+    setChatMessages([])
+    setSelectedChat(null)
+
+    handleNewChat()
+    toast.success("All chats deleted!")
+  }
+
+  return (
+    <Dialog open={showChatDialog} onOpenChange={setShowChatDialog}>
+      <DialogTrigger asChild>
+        <IconTrash className="hover:opacity-50" size={32} />
+      </DialogTrigger>
+
+      <DialogContent>
+        <DialogHeader>
+          <DialogTitle>Delete All Chats</DialogTitle>
+          <DialogDescription>
+            Are you sure you want to delete all chats?
+          </DialogDescription>
+        </DialogHeader>
+
+        <DialogFooter>
+          <Button variant="ghost" onClick={() => setShowChatDialog(false)}>
+            Cancel
+          </Button>
+          <Button
+            ref={buttonRef}
+            variant="destructive"
+            onClick={handleDeleteAllChats}
+          >
+            Delete
+          </Button>
+        </DialogFooter>
+      </DialogContent>
+    </Dialog>
+  )
+}
diff --git a/components/utility/profile-settings.tsx b/components/utility/profile-settings.tsx
index 576cee0eb3..4ca30817f4 100644
--- a/components/utility/profile-settings.tsx
+++ b/components/utility/profile-settings.tsx
@@ -42,6 +42,9 @@ import { Tabs, TabsContent, TabsList, TabsTrigger } from "../ui/tabs"
 import { TextareaAutosize } from "../ui/textarea-autosize"
 import { WithTooltip } from "../ui/with-tooltip"
 import { ThemeSwitcher } from "./theme-switcher"
+import { deleteAllChats } from "@/db/chats"
+import { useChatHandler } from "../chat/chat-hooks/use-chat-handler"
+import { DeleteAllChats } from "../sidebar/items/chat/delete-all-chats"
 
 interface ProfileSettingsProps {}
 
@@ -52,11 +55,16 @@ export const ProfileSettings: FC<ProfileSettingsProps> = ({}) => {
     envKeyMap,
     setAvailableHostedModels,
     setAvailableOpenRouterModels,
-    availableOpenRouterModels
+    availableOpenRouterModels,
+    setChats,
+    setChatMessages,
+    setSelectedChat
   } = useContext(ChatbotUIContext)
 
   const router = useRouter()
 
+  const { handleNewChat } = useChatHandler()
+
   const buttonRef = useRef<HTMLButtonElement>(null)
 
   const [isOpen, setIsOpen] = useState(false)
@@ -747,6 +755,8 @@ export const ProfileSettings: FC<ProfileSettingsProps> = ({}) => {
             />
           </div>
 
+          <DeleteAllChats profile={profile} />
+
           <div className="ml-auto space-x-2">
             <Button variant="ghost" onClick={() => setIsOpen(false)}>
               Cancel
diff --git a/db/chats.ts b/db/chats.ts
index 13332beb73..319be6138e 100644
--- a/db/chats.ts
+++ b/db/chats.ts
@@ -79,3 +79,13 @@ export const deleteChat = async (chatId: string) => {
 
   return true
 }
+
+export const deleteAllChats = async (userId: string) => {
+  const { error } = await supabase.from("chats").delete().eq("user_id", userId)
+
+  if (error) {
+    throw new Error(error.message)
+  }
+
+  return true
+}
diff --git a/db/files.ts b/db/files.ts
index 68ad01e343..0b69625e68 100644
--- a/db/files.ts
+++ b/db/files.ts
@@ -94,7 +94,10 @@ export const createFile = async (
   let validFilename = fileRecord.name.replace(/[^a-z0-9.]/gi, "_").toLowerCase()
   const extension = file.name.split(".").pop()
   const extensionIndex = validFilename.lastIndexOf(".")
-  const baseName = validFilename.substring(0, (extensionIndex < 0) ? undefined : extensionIndex)
+  const baseName = validFilename.substring(
+    0,
+    extensionIndex < 0 ? undefined : extensionIndex
+  )
   const maxBaseNameLength = 100 - (extension?.length || 0) - 1
   if (baseName.length > maxBaseNameLength) {
     fileRecord.name = baseName.substring(0, maxBaseNameLength) + "." + extension
diff --git a/lib/build-prompt.ts b/lib/build-prompt.ts
index ddb0e76377..8bda846f9f 100644
--- a/lib/build-prompt.ts
+++ b/lib/build-prompt.ts
@@ -184,36 +184,35 @@ function buildRetrievalText(fileItems: Tables<"file_items">[]) {
 }
 
 function adaptSingleMessageForGoogleGemini(message: any) {
-
   let adaptedParts = []
 
   let rawParts = []
-  if(!Array.isArray(message.content)) {
-    rawParts.push({type: 'text', text: message.content})
+  if (!Array.isArray(message.content)) {
+    rawParts.push({ type: "text", text: message.content })
   } else {
     rawParts = message.content
   }
 
-  for(let i = 0; i < rawParts.length; i++) {
+  for (let i = 0; i < rawParts.length; i++) {
     let rawPart = rawParts[i]
 
-    if(rawPart.type == 'text') {
-      adaptedParts.push({text: rawPart.text})
-    } else if(rawPart.type === 'image_url') {
+    if (rawPart.type == "text") {
+      adaptedParts.push({ text: rawPart.text })
+    } else if (rawPart.type === "image_url") {
       adaptedParts.push({
         inlineData: {
           data: getBase64FromDataURL(rawPart.image_url.url),
-          mimeType: getMediaTypeFromDataURL(rawPart.image_url.url),
+          mimeType: getMediaTypeFromDataURL(rawPart.image_url.url)
         }
       })
     }
   }
 
-  let role = 'user'
-  if(["user", "system"].includes(message.role)) {
-    role = 'user'
-  } else if(message.role === 'assistant') {
-    role = 'model'
+  let role = "user"
+  if (["user", "system"].includes(message.role)) {
+    role = "user"
+  } else if (message.role === "assistant") {
+    role = "model"
   }
 
   return {
@@ -222,29 +221,29 @@ function adaptSingleMessageForGoogleGemini(message: any) {
   }
 }
 
-function adaptMessagesForGeminiVision(
-  messages: any[]
-) {
+function adaptMessagesForGeminiVision(messages: any[]) {
   // Gemini Pro Vision cannot process multiple messages
   // Reformat, using all texts and last visual only
 
   const basePrompt = messages[0].parts[0].text
   const baseRole = messages[0].role
-  const lastMessage = messages[messages.length-1]
-  const visualMessageParts = lastMessage.parts;
-  let visualQueryMessages = [{
-    role: "user",
-    parts: [
-      `${baseRole}:\n${basePrompt}\n\nuser:\n${visualMessageParts[0].text}\n\n`,
-      visualMessageParts.slice(1)
-    ]
-  }]
+  const lastMessage = messages[messages.length - 1]
+  const visualMessageParts = lastMessage.parts
+  let visualQueryMessages = [
+    {
+      role: "user",
+      parts: [
+        `${baseRole}:\n${basePrompt}\n\nuser:\n${visualMessageParts[0].text}\n\n`,
+        visualMessageParts.slice(1)
+      ]
+    }
+  ]
   return visualQueryMessages
 }
 
 export async function adaptMessagesForGoogleGemini(
   payload: ChatPayload,
-  messages:  any[]
+  messages: any[]
 ) {
   let geminiMessages = []
   for (let i = 0; i < messages.length; i++) {
@@ -252,9 +251,8 @@ export async function adaptMessagesForGoogleGemini(
     geminiMessages.push(adaptedMessage)
   }
 
-  if(payload.chatSettings.model === "gemini-pro-vision") {
+  if (payload.chatSettings.model === "gemini-pro-vision") {
     geminiMessages = adaptMessagesForGeminiVision(geminiMessages)
   }
   return geminiMessages
 }
-
diff --git a/lib/chat-setting-limits.ts b/lib/chat-setting-limits.ts
index c802bd657b..1dcde08490 100644
--- a/lib/chat-setting-limits.ts
+++ b/lib/chat-setting-limits.ts
@@ -47,7 +47,7 @@ export const CHAT_SETTING_LIMITS: Record<LLMID, ChatSettingLimits> = {
   },
 
   // GOOGLE MODELS
-  
+
   "gemini-1.5-flash": {
     MIN_TEMPERATURE: 0.0,
     MAX_TEMPERATURE: 1.0,
diff --git a/lib/models/llm/google-llm-list.ts b/lib/models/llm/google-llm-list.ts
index d039eda4af..833eb7ec2b 100644
--- a/lib/models/llm/google-llm-list.ts
+++ b/lib/models/llm/google-llm-list.ts
@@ -44,4 +44,9 @@ const GEMINI_PRO_VISION: LLM = {
   imageInput: true
 }
 
-export const GOOGLE_LLM_LIST: LLM[] = [GEMINI_PRO, GEMINI_PRO_VISION, GEMINI_1_5_PRO, GEMINI_1_5_FLASH]
+export const GOOGLE_LLM_LIST: LLM[] = [
+  GEMINI_PRO,
+  GEMINI_PRO_VISION,
+  GEMINI_1_5_PRO,
+  GEMINI_1_5_FLASH
+]