#612 SD manual prompt refine mode

public/index.html

@@ -2865,7 +2865,7 @@
                 <div id="dialogue_popup_text">
                     <h3>text</h3>
                 </div>
-                <input id="dialogue_popup_input" class="text_pole" type="text" />
+                <textarea id="dialogue_popup_input" class="text_pole" rows="1"></textarea>
                 <div id="dialogue_popup_controls">
                     <div id="dialogue_popup_ok" class="menu_button" data-i18n="Delete">Delete</div>
                     <div id="dialogue_popup_cancel" class="menu_button" data-i18n="Cancel">Cancel</div>

public/script.js

@@ -5531,7 +5531,7 @@ function onScenarioOverrideRemoveClick() {
     $(this).closest('.scenario_override').find('.chat_scenario').val('').trigger('input');
 }
 
-function callPopup(text, type, inputValue = '', okButton) {
+function callPopup(text, type, inputValue = '', { okButton, rows } = {}) {
     if (type) {
         popup_type = type;
     }
@@ -5559,6 +5559,7 @@ function callPopup(text, type, inputValue = '', okButton) {
     }
 
     $("#dialogue_popup_input").val(inputValue);
+    $("#dialogue_popup_input").attr("rows", rows ?? 1);
 
     if (popup_type == 'input') {
         $("#dialogue_popup_input").css("display", "block");

public/scripts/extensions/dice/index.js

@@ -10,7 +10,7 @@ async function doDiceRoll(customDiceFormula) {
     let value = typeof customDiceFormula === 'string' ? customDiceFormula.trim() : $(this).data('value');
 
     if (value == 'custom') {
-        value = await callPopup('Enter the dice formula:<br><i>(for example, <tt>2d6</tt>)</i>', 'input');x
+        value = await callPopup('Enter the dice formula:<br><i>(for example, <tt>2d6</tt>)</i>', 'input');
     }
 
     if (!value) {

public/scripts/extensions/regex/index.js

@@ -11,7 +11,7 @@ async function saveRegexScript(regexScript, existingScriptIndex) {
             toastr.error(`Could not save regex script: The script name was undefined or empty!`);
             return;
         }
-    
+
         // Does the script name already exist?
         if (extension_settings.regex.find((e) => e.scriptName === regexScript.scriptName)) {
             toastr.error(`Could not save regex script: A script with name ${regexScript.scriptName} already exists.`);
@@ -144,7 +144,7 @@ async function onRegexEditorOpenClick(existingId) {
             .prop("checked", true);
     }
 
-    const popupResult = await callPopup(editorHtml, "confirm", undefined, "Save");
+    const popupResult = await callPopup(editorHtml, "confirm", undefined, { okButton: "Save" });
     if (popupResult) {
         const newRegexScript = {
             scriptName: editorHtml.find(".regex_script_name").val(),

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

@@ -131,6 +131,9 @@ const defaultSettings = {
     horde: false,
     horde_nsfw: false,
     horde_karras: true,
+
+    // Refine mode
+    refine_mode: false,
 }
 
 async function loadSettings() {
@@ -149,10 +152,16 @@ async function loadSettings() {
     $('#sd_horde_karras').prop('checked', extension_settings.sd.horde_karras);
     $('#sd_restore_faces').prop('checked', extension_settings.sd.restore_faces);
     $('#sd_enable_hr').prop('checked', extension_settings.sd.enable_hr);
+    $('#sd_refine_mode').prop('checked', extension_settings.sd.refine_mode);
 
     await Promise.all([loadSamplers(), loadModels()]);
 }
 
+function onRefineModeInput() {
+    extension_settings.sd.refine_mode = !!$('#sd_refine_mode').prop('checked');
+    saveSettingsDebounced();
+}
+
 function onScaleInput() {
     extension_settings.sd.scale = Number($('#sd_scale').val());
     $('#sd_scale_value').text(extension_settings.sd.scale.toFixed(1));
@@ -470,7 +479,7 @@ async function getPrompt(generationType, message, trigger, quiet_prompt) {
 }
 
 async function generatePrompt(quiet_prompt) {
-    return processReply(await new Promise(
+    let reply = processReply(await new Promise(
         async function promptPromise(resolve, reject) {
             try {
                 await getContext().generate('quiet', { resolve, reject, quiet_prompt, force_name2: true, });
@@ -479,6 +488,18 @@ async function generatePrompt(quiet_prompt) {
                 reject();
             }
         }));
+
+    if (extension_settings.sd.refine_mode) {
+        const refinedPrompt = await callPopup('<h3>Review and edit the generated prompt:</h3>Press "Cancel" to abort the image generation.', 'input', reply, { rows: 5 });
+
+        if (refinedPrompt) {
+            reply = refinedPrompt;
+        } else {
+            throw new Error('Generation aborted by user.');
+        }
+    }
+
+    return reply;
 }
 
 async function sendGenerationRequest(prompt, callback) {
@@ -583,7 +604,6 @@ function addSDGenButtons() {
     `
     const dropdownHtml = `
     <div id="sd_dropdown">
-    
         <ul class="list-group">
         <span>Send me a picture of:</span>
             <li class="list-group-item" id="sd_you" data-value="you">Yourself</li>
@@ -750,6 +770,10 @@ jQuery(async () => {
             <small><i>Use slash commands or the bottom Paintbrush button to generate images. Type <span class="monospace">/help</span> in chat for more details</i></small>
             <br>
             <small><i>Hint: Save an API key in Horde KoboldAI API settings to use it here.</i></small>
+            <label for="sd_refine_mode" class="checkbox_label" title="Allow to edit prompts manually before sending them to generation API">
+                <input id="sd_refine_mode" type="checkbox" />
+                Edit prompts before generation
+            </label>
             <div class="flex-container flexGap5 marginTop10 margin-bot-10px">
                 <label class="checkbox_label">
                     <input id="sd_horde" type="checkbox" />
@@ -810,6 +834,7 @@ jQuery(async () => {
     $('#sd_horde_karras').on('input', onHordeKarrasInput);
     $('#sd_restore_faces').on('input', onRestoreFacesInput);
     $('#sd_enable_hr').on('input', onHighResFixInput);
+    $('#sd_refine_mode').on('input', onRefineModeInput);
 
     $('.sd_settings .inline-drawer-toggle').on('click', function () {
         initScrollHeight($("#sd_prompt_prefix"));