SillyTavern/public/scripts/extensions/randomize/index.js

import { saveSettingsDebounced } from "../../../script.js";
import { extension_settings } from "../../extensions.js";

function toggleRandomizedSetting(buttonRef, forId) {
    if (extension_settings.randomizer.controls.indexOf(forId) === -1) {
        extension_settings.randomizer.controls.push(forId);
    } else {
        extension_settings.randomizer.controls = extension_settings.randomizer.controls.filter(x => x !== forId);
    }

    buttonRef.toggleClass('active');
    console.debug('Randomizer controls:', extension_settings.randomizer.controls);
    saveSettingsDebounced();
}

function addRandomizeButton() {
    const counterRef = $(this);
    const labelRef = $(this).find('div[data-for]');
    const isDisabled = counterRef.data('randomization-disabled');

    if (labelRef.length === 0 || isDisabled == true) {
        return;
    }

    const forId = labelRef.data('for');
    const buttonRef = $('<div class="randomize_button menu_button fa-solid fa-shuffle"></div>');
    buttonRef.toggleClass('active', extension_settings.randomizer.controls.indexOf(forId) !== -1);
    buttonRef.hide();
    buttonRef.on('click', () => toggleRandomizedSetting(buttonRef, forId));
    counterRef.append(buttonRef);
}

function onRandomizerEnabled() {
    extension_settings.randomizer.enabled = $(this).prop('checked');
    $('.randomize_button').toggle(extension_settings.randomizer.enabled);
    console.debug('Randomizer enabled:', extension_settings.randomizer.enabled);
}

window['randomizerInterceptor'] = (function () {
    if (extension_settings.randomizer.enabled === false) {
        console.debug('Randomizer skipped: disabled.');
        return;
    }

    if (extension_settings.randomizer.fluctuation === 0 || extension_settings.randomizer.controls.length === 0) {
        console.debug('Randomizer skipped: nothing to do.');
        return;
    }

    for (const control of extension_settings.randomizer.controls) {
        const controlRef = $('#' + control);

        if (controlRef.length === 0) {
            console.debug(`Randomizer skipped: control ${control} not found.`);
            continue;
        }

        if (!controlRef.is(':visible')) {
            console.debug(`Randomizer skipped: control ${control} is not visible.`);
            continue;
        }

        let previousValue = parseFloat(controlRef.data('previous-value'));
        let originalValue = parseFloat(controlRef.data('original-value'));
        let currentValue = parseFloat(controlRef.val());

        let value;

        // Initialize originalValue and previousValue if they are NaN
        if (isNaN(originalValue)) {
            originalValue = currentValue;
            controlRef.data('original-value', originalValue);
        }
        if (isNaN(previousValue)) {
            previousValue = currentValue;
            controlRef.data('previous-value', previousValue);
        }

        // If the current value hasn't changed compared to the previous value, use the original value as a base for the calculation
        if (currentValue === previousValue) {
            console.debug(`Randomizer for ${control} reusing original value: ${originalValue}`);
            value = originalValue;
        } else {
            console.debug(`Randomizer for ${control} using current value: ${currentValue}`);
            value = currentValue;
            controlRef.data('previous-value', currentValue); // Update the previous value when using the current value
            controlRef.data('original-value', currentValue); // Update the original value when using the current value
        }

        if (isNaN(value)) {
            console.debug('Randomizer skipped: NaN.');
            continue;
        }

        const fluctuation = extension_settings.randomizer.fluctuation;
        const min = parseFloat(controlRef.attr('min'));
        const max = parseFloat(controlRef.attr('max'));
        const delta = (Math.random() * fluctuation * 2 - fluctuation) * value;
        const newValue = Math.min(Math.max(value + delta, min), max);
        console.debug(`Randomizer for ${control}: ${value} -> ${newValue} (delta: ${delta}, min: ${min}, max: ${max})`);
        controlRef.val(newValue).trigger('input');
        controlRef.data('previous-value', parseFloat(controlRef.val()));
    }
});

jQuery(() => {
    const html = `
    <div class="randomizer_settings">
        <div class="inline-drawer">
            <div class="inline-drawer-toggle inline-drawer-header">
                <b>Parameter Randomizer</b>
                <div class="inline-drawer-icon fa-solid fa-circle-chevron-down down"></div>
            </div>
            <div class="inline-drawer-content">
                <label for="randomizer_enabled" class="checkbox_label">
                    <input type="checkbox" id="randomizer_enabled" name="randomizer_enabled" >
                    Enabled
                </label>
                <div  class="range-block">
                    <div class="range-block-title">
                        Fluctuation (0-1)
                    </div>
                    <div class="range-block-range-and-counter">
                        <div class="range-block-range-and-counter">
                            <div class="range-block-range">
                                <input type="range" id="randomizer_fluctuation" min="0" max="1" step="0.1">
                            </div>
                            <div class="range-block-counter">
                                <div contenteditable="true" data-for="randomizer_fluctuation" id="randomizer_fluctuation_counter">
                                </div>
                            </div>
                        </div>
                    </div>
                </div>
            </div>
        </div>
    </div>`;

    $('#extensions_settings').append(html);
    $('#ai_response_configuration .range-block-counter').each(addRandomizeButton);
    $('#randomizer_enabled').on('input', onRandomizerEnabled);
    $('#randomizer_enabled').prop('checked', extension_settings.randomizer.enabled).trigger('input');
    $('#randomizer_fluctuation').val(extension_settings.randomizer.fluctuation).trigger('input');
    $('#randomizer_fluctuation_counter').text(extension_settings.randomizer.fluctuation);
    $('#randomizer_fluctuation').on('input', function () {
        const value = parseFloat($(this).val());
        $('#randomizer_fluctuation_counter').text(value);
        extension_settings.randomizer.fluctuation = value;
        console.debug('Randomizer fluctuation:', extension_settings.randomizer.fluctuation);
        saveSettingsDebounced();
    });
});