Add Extension Hook for Stable Diffusion Prompt Processing (#4167)

* Update index.js * Update index.js * Update script.js * Use a named constant in event emit --------- Co-authored-by: Cohee <18619528+Cohee1207@users.noreply.github.com>
2025-06-16 21:48:38 +03:00
parent 3e03b9f0a2
commit 0a136cbb2a
2 changed files with 7 additions and 1 deletions
@@ -2468,9 +2468,14 @@ async function generatePicture(initiator, args, trigger, message, callback) {
        const combineNegatives = (prefix) => { negativePromptPrefix = combinePrefixes(negativePromptPrefix, prefix); };

        // generate the text prompt for the image
-        const prompt = await getPrompt(generationType, message, trigger, quietPrompt, combineNegatives);
+        let prompt = await getPrompt(generationType, message, trigger, quietPrompt, combineNegatives);
        console.log('Processed image prompt:', prompt);

+        // Extension hook for prompt processing
+        const eventData = { prompt, generationType, message, trigger };
+        await eventSource.emit(event_types.SD_PROMPT_PROCESSING, eventData);
+        prompt = eventData.prompt; // Allow extensions to modify the prompt
+
        $(stopButton).show();
        eventSource.once(CUSTOM_STOP_EVENT, stopListener);