Add SD character-prefixed free-mode

public/scripts/extensions/stable-diffusion/index.js

@@ -58,6 +58,7 @@ const generationMode = {
     CHARACTER_MULTIMODAL: 8,
     USER_MULTIMODAL: 9,
     FACE_MULTIMODAL: 10,
+    CHARACTER_FREE: 11,
 };
 
 const multimodalMap = {
@@ -1563,6 +1564,10 @@ function getGenerationType(prompt) {
         mode = multimodalMap[mode];
     }
 
+    if (mode === generationMode.FREE && /^char\s/.test(prompt)) {
+        mode = generationMode.CHARACTER_FREE;
+    }
+
     return mode;
 }
 
@@ -1718,6 +1723,9 @@ async function getPrompt(generationType, message, trigger, quietPrompt) {
         case generationMode.USER_MULTIMODAL:
             prompt = await generateMultimodalPrompt(generationType, quietPrompt);
             break;
+        case generationMode.CHARACTER_FREE:
+            prompt = generateFreeCharacterPrompt(trigger);
+            break;
         default:
             prompt = await generatePrompt(quietPrompt);
             break;
@@ -1730,6 +1738,41 @@ async function getPrompt(generationType, message, trigger, quietPrompt) {
     return prompt;
 }
 
+/**
+ * Generates a free prompt with a character-specific prompt prefix.
+ * @param {string} trigger - The prompt to use for the image generation.
+ * @returns {string}
+ */
+function generateFreeCharacterPrompt(trigger) {
+    const context = getContext();
+
+    const getCharacterPrefix = () => {
+        const key = getLastCharacterKey();
+        const value = (extension_settings.sd.character_prompts[key] || '').trim();
+        return value ? value + ' ' : '';
+    };
+
+    const getLastCharacterKey = () => {
+        if (typeof this_chid !== 'undefined') {
+            return getCharaFilename(this_chid);
+        }
+        for (let i = context.chat.length - 1; i >= 0; i--) {
+            const message = context.chat[i];
+            if (context.is_user || message.is_system) {
+                continue;
+            } else if (typeof message.original_avatar === 'string') {
+                return message.original_avatar.replace(/\.[^/.]+$/, '');
+            }
+        }
+        throw new Error('No usable messages found.');
+    };
+
+    const prefix = getCharacterPrefix();
+    const pieces = trigger.split(/\s+/g);
+    const prompt = pieces.slice(1).join(' ');
+    return (prefix + prompt).trim();
+}
+
 /**
  * Generates a prompt using multimodal captioning.
  * @param {number} generationType - The type of image generation to perform.