From 49d7f3307f4db861203e949154a9108a0dbe2bb9 Mon Sep 17 00:00:00 2001
From: n4ze3m <n4ze3m@gmail.com>
Date: Tue, 16 Apr 2024 23:21:15 +0530
Subject: [PATCH] Add useStorage hook to remember recently used model

---
 src/components/Option/Models/index.tsx | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/src/components/Option/Models/index.tsx b/src/components/Option/Models/index.tsx
index 88f4804..7728484 100644
--- a/src/components/Option/Models/index.tsx
+++ b/src/components/Option/Models/index.tsx
@@ -8,6 +8,7 @@ import { useState } from "react"
 import { useForm } from "@mantine/form"
 import { Download, RotateCcw, Trash2 } from "lucide-react"
 import { useTranslation } from "react-i18next"
+import { useStorage } from "@plasmohq/storage/hook"
 
 dayjs.extend(relativeTime)
 
@@ -15,6 +16,7 @@ export const ModelsBody = () => {
   const queryClient = useQueryClient()
   const [open, setOpen] = useState(false)
   const { t } = useTranslation(["settings", "common"])
+  const [selectedModel, setSelectedModel] = useStorage("selectedModel")
 
   const form = useForm({
     initialValues: {
@@ -131,6 +133,12 @@ export const ModelsBody = () => {
                             window.confirm(t("manageModels.confirm.delete"))
                           ) {
                             deleteOllamaModel(record.model)
+                            if (
+                              selectedModel &&
+                              selectedModel === record.model
+                            ) {
+                              setSelectedModel(null)
+                            }
                           }
                         }}
                         className="text-red-500 dark:text-red-400">
@@ -193,8 +201,7 @@ export const ModelsBody = () => {
                   }}
                 />
               ),
-              defaultExpandAllRows: false,
-              
+              defaultExpandAllRows: false
             }}
             bordered
             dataSource={data}