From 8d60cbdc7a47c2e017020b3e8282860573cf5d01 Mon Sep 17 00:00:00 2001
From: Daniel Avila <messagedaniel@protonmail.com>
Date: Sun, 5 Mar 2023 17:02:00 -0500
Subject: [PATCH] feat: dynamic custom model saving/updating in dialog

---
 src/components/Models/ModelDialog.jsx | 8 ++++++--
 src/components/Models/ModelMenu.jsx   | 2 +-
 2 files changed, 7 insertions(+), 3 deletions(-)

diff --git a/src/components/Models/ModelDialog.jsx b/src/components/Models/ModelDialog.jsx
index b613734248..3de7370a26 100644
--- a/src/components/Models/ModelDialog.jsx
+++ b/src/components/Models/ModelDialog.jsx
@@ -16,7 +16,7 @@ import {
   DialogTitle
 } from '../ui/Dialog.tsx';
 
-export default function ModelDialog({ mutate }) {
+export default function ModelDialog({ mutate, modelMap }) {
   const dispatch = useDispatch();
   const [chatGptLabel, setChatGptLabel] = useState('');
   const [promptPrefix, setPromptPrefix] = useState('');
@@ -50,7 +50,7 @@ export default function ModelDialog({ mutate }) {
     updateCustomGpt.trigger({ value, chatGptLabel, promptPrefix });
 
     mutate();
-    setSaveText('Saved!');
+    setSaveText((prev) => prev + 'd!');
     setTimeout(() => {
       setSaveText('Save');
     }, 2500);
@@ -60,6 +60,10 @@ export default function ModelDialog({ mutate }) {
     // dispatch(setDisabled(false));
   };
 
+  if (modelMap[chatGptLabel.toLowerCase()] && saveText === 'Save') {
+    setSaveText('Update');
+  }
+
   const requiredProp = required ? { required: true } : {};
 
   return (
diff --git a/src/components/Models/ModelMenu.jsx b/src/components/Models/ModelMenu.jsx
index c5f49dcb61..a53472f9fb 100644
--- a/src/components/Models/ModelMenu.jsx
+++ b/src/components/Models/ModelMenu.jsx
@@ -133,7 +133,7 @@ export default function ModelMenu() {
           </DropdownMenuRadioGroup>
         </DropdownMenuContent>
       </DropdownMenu>
-      <ModelDialog mutate={trigger} />
+      <ModelDialog mutate={trigger} modelMap={modelMap}/>
     </Dialog>
   );
 }