From ade6ef8af8e01173aa184b5edb15512832998fd0 Mon Sep 17 00:00:00 2001
From: Cohee <18619528+Cohee1207@users.noreply.github.com>
Date: Sun, 22 Sep 2024 12:20:07 +0300
Subject: [PATCH] Better customized sysprompt migration

---
 public/script.js            |  2 +-
 public/scripts/sysprompt.js | 19 +++++++++++++------
 2 files changed, 14 insertions(+), 7 deletions(-)

diff --git a/public/script.js b/public/script.js
index 81638634c..e40512b0c 100644
--- a/public/script.js
+++ b/public/script.js
@@ -934,6 +934,7 @@ async function firstLoadInit() {
     initDefaultSlashCommands();
     initTextGenModels();
     initSystemPrompts();
+    await initPresetManager();
     await getSystemMessages();
     sendSystemMessage(system_message_types.WELCOME);
     sendSystemMessage(system_message_types.WELCOME_PROMPT);
@@ -947,7 +948,6 @@ async function firstLoadInit() {
     await getCharacters();
     await getBackgrounds();
     await initTokenizers();
-    await initPresetManager();
     initBackgrounds();
     initAuthorsNote();
     initPersonas();
diff --git a/public/scripts/sysprompt.js b/public/scripts/sysprompt.js
index b4103eb0d..1e4106333 100644
--- a/public/scripts/sysprompt.js
+++ b/public/scripts/sysprompt.js
@@ -17,14 +17,21 @@ const $select = $('#sysprompt_select');
 const $content = $('#sysprompt_content');
 const $contentBlock = $('#SystemPromptBlock');
 
-function migrateSystemPromptFromInstructMode() {
+async function migrateSystemPromptFromInstructMode() {
     if ('system_prompt' in power_user.instruct) {
-        power_user.sysprompt.enabled = power_user.instruct.enabled;
-        power_user.sysprompt.content = String(power_user.instruct.system_prompt);
+        const prompt = String(power_user.instruct.system_prompt);
         delete power_user.instruct.system_prompt;
+        power_user.sysprompt.enabled = power_user.instruct.enabled;
+        power_user.sysprompt.content = prompt;
 
-        if (system_prompts.some(x => x.name === power_user.instruct.preset)) {
-            power_user.sysprompt.name = power_user.instruct.preset;
+        const existingPromptName = system_prompts.find(x => x.content === prompt)?.name;
+
+        if (existingPromptName) {
+            power_user.sysprompt.name = existingPromptName;
+        } else {
+            const data = { name: `${power_user.instruct.preset} (Migrated)`, content: prompt };
+            await getPresetManager('sysprompt')?.savePreset(data.name, data);
+            power_user.sysprompt.name = data.name;
         }
 
         saveSettingsDebounced();
@@ -41,7 +48,7 @@ export async function loadSystemPrompts(data) {
         system_prompts = data.sysprompt;
     }
 
-    migrateSystemPromptFromInstructMode();
+    await migrateSystemPromptFromInstructMode();
     toggleSystemPromptDisabledControls();
 
     for (const prompt of system_prompts) {