Merge pull request #3000 from error-usernotavailable/staging ADetailer functionality for stable-diffusion extension

cb987706eba8303e7d11ea0f8e129b73b0618c85

Cohee <18619528+Cohee1207@users.noreply.github.com>

Signed

Parents: ada44497dea61f47c29c5236879e42ce5127028a ae784ddb1a7fe8dfe34f73f66c3deb3a66c44e49

public/scripts/extensions/stable-diffusion/index.js+36 -11 Viewed

 } from '../../../script.js';
 import { getApiUrl, getContext, extension_settings, doExtrasFetch, modules, renderExtensionTemplateAsync, writeExtensionField } from '../../extensions.js';
 import { selected_group } from '../../group-chats.js';
 import { stringFormat, initScrollHeight, resetScrollHeight, getCharaFilename, saveBase64AsFile, getBase64Async, delay, isTrueBoolean, debounce, isFalseBoolean, deepMerge } from '../../utils.js';
 import { getMessageTimeStamp, humanizedDateTime } from '../../RossAscends-mods.js';
 import { SECRET_KEYS, secret_state, writeSecret } from '../../secrets.js';
 import { getNovelUnlimitedImageGeneration, getNovelAnlas, loadNovelSubscriptionData } from '../../nai-settings.js';

     // Automatic1111/Horde exclusives
     restore_faces: false,
     enable_hr: false,
 +    adetailer_face: false,
     // Horde settings
     horde: false,

     $('#sd_horde_sanitize').prop('checked', extension_settings.sd.horde_sanitize);
     $('#sd_restore_faces').prop('checked', extension_settings.sd.restore_faces);
     $('#sd_enable_hr').prop('checked', extension_settings.sd.enable_hr);
 +    $('#sd_adetailer_face').prop('checked', extension_settings.sd.adetailer_face);
     $('#sd_refine_mode').prop('checked', extension_settings.sd.refine_mode);
     $('#sd_multimodal_captioning').prop('checked', extension_settings.sd.multimodal_captioning);
     $('#sd_auto_url').val(extension_settings.sd.auto_url);

     saveSettingsDebounced();
+}
 +function onADetailerFaceChange() {
 +    extension_settings.sd.adetailer_face = !!$('#sd_adetailer_face').prop('checked');
 +    saveSettingsDebounced();
 +}
++
 const resolutionOptions = {
     sd_res_512x512: { width: 512, height: 512, name: '512x512 (1:1, icons, profile pictures)' },
     sd_res_600x600: { width: 600, height: 600, name: '600x600 (1:1, icons, profile pictures)' },

  */
 async function generateAutoImage(prompt, negativePrompt, signal) {
     const isValidVae = extension_settings.sd.vae && !['N/A', placeholderVae].includes(extension_settings.sd.vae);
 -    const result = await fetch('/api/sd/generate', {
 +    let payload = {
 -        method: 'POST',
 -        headers: getRequestHeaders(),
 -        signal: signal,
 -        body: JSON.stringify({
         ...getSdRequestBody(),
         prompt: prompt,
         negative_prompt: negativePrompt,

         denoising_strength: extension_settings.sd.denoising_strength,
         hr_second_pass_steps: extension_settings.sd.hr_second_pass_steps,
         seed: extension_settings.sd.seed >= 0 ? extension_settings.sd.seed : undefined,
 -            // For AUTO1111
         override_settings: {
             CLIP_stop_at_last_layers: extension_settings.sd.clip_skip,
             sd_vae: isValidVae ? extension_settings.sd.vae : undefined,
         },
         override_settings_restore_afterwards: true,
                     clip_skip: extension_settings.sd.clip_skip, // For SD.Next
 -            clip_skip: extension_settings.sd.clip_skip,
 -            // Ensure generated img is saved to disk
         save_images: true,
         send_images: true,
         do_not_save_grid: false,
         do_not_save_samples: false,
 -        }),
 +    };
++
 +    // Conditionally add the ADetailer if adetailer_face is enabled
 +    if (extension_settings.sd.adetailer_face) {
 +        payload = deepMerge(payload, {
 +            alwayson_scripts: {
 +                ADetailer: {
 +                    args: [
 +                        true, // ad_enable
 +                        true, // skip_img2img
 +                        {
 +                            'ad_model': 'face_yolov8n.pt',
 +                        },
 +                    ],
 +                },
 +            },
 +        });
 +    }
++
 +    // Make the fetch call with the payload
 +    const result = await fetch('/api/sd/generate', {
 +        method: 'POST',
 +        headers: getRequestHeaders(),
 +        signal: signal,
 +        body: JSON.stringify(payload),
     });
     if (result.ok) {

     $('#sd_horde_sanitize').on('input', onHordeSanitizeInput);
     $('#sd_restore_faces').on('input', onRestoreFacesInput);
     $('#sd_enable_hr').on('input', onHighResFixInput);
 +    $('#sd_adetailer_face').on('change', onADetailerFaceChange);
     $('#sd_refine_mode').on('input', onRefineModeInput);
     $('#sd_character_prompt').on('input', onCharacterPromptInput);
     $('#sd_character_negative_prompt').on('input', onCharacterNegativePromptInput);

public/scripts/extensions/stable-diffusion/settings.html+10 -0 Viewed

                 </label>
             </div>
 +            <div class="flex-container marginTopBot5" data-sd-source="auto,vlad">
 +                <label for="sd_adetailer_face" class="flex1 checkbox_label" data-i18n="[title]sd_adetailer_face" title="Use ADetailer with face model during the generation. The ADetailer extension must be installed on the backend.">
 +                    <input id="sd_adetailer_face" type="checkbox" />
 +                    <small data-i18n="Use ADetailer (Face)">Use ADetailer (Face)</small>
 +                </label>
 +                <div class="flex1">
 +                    <!-- I will be useful later! -->
 +                </div>
 +            </div>
++
             <div class="flex-container marginTopBot5" data-sd-source="novel">
                 <label class="flex1 checkbox_label" data-i18n="[title]SMEA versions of samplers are modified to perform better at high resolution." title="SMEA versions of samplers are modified to perform better at high resolution.">
                     <input id="sd_novel_sm" type="checkbox" />

public/scripts/utils.js+32 -0 Viewed

     previous: -1000,
 };
 +/**
 + * Determines if a value is an object.
 + * @param {any} item The item to check.
 + * @returns {boolean} True if the item is an object, false otherwise.
 + */
 +function isObject(item) {
 +    return (item && typeof item === 'object' && !Array.isArray(item));
 +}
++
 +/**
 + * Merges properties of two objects. If the property is an object, it will be merged recursively.
 + * @param {object} target The target object
 + * @param {object} source The source object
 + * @returns {object} Merged object
 + */
 +export function deepMerge(target, source) {
 +    let output = Object.assign({}, target);
 +    if (isObject(target) && isObject(source)) {
 +        Object.keys(source).forEach(key => {
 +            if (isObject(source[key])) {
 +                if (!(key in target))
 +                    Object.assign(output, { [key]: source[key] });
 +                else
 +                    output[key] = deepMerge(target[key], source[key]);
 +            } else {
 +                Object.assign(output, { [key]: source[key] });
 +            }
 +        });
 +    }
 +    return output;
 +}
++
 export function escapeHtml(str) {
     return String(str).replace(/&/g, '&amp;').replace(/</g, '&lt;').replace(/>/g, '&gt;').replace(/"/g, '&quot;');
+}