Add parameter randomizer plugin

This commit is contained in:
Cohee
2023-07-08 20:24:15 +03:00
parent 68551ae15c
commit 474a9cb99e
6 changed files with 177 additions and 11 deletions

View File

@@ -115,7 +115,7 @@
Click slider numbers to input manually. Click slider numbers to input manually.
</div> </div>
<div class="scrollableInner"> <div class="scrollableInner">
<div class="flex-container"> <div class="flex-container" id="ai_response_configuration">
<div id="respective-presets-block" class="width100p"> <div id="respective-presets-block" class="width100p">
<div id="kobold_api-presets"> <div id="kobold_api-presets">
<h3><span data-i18n="kobldpresets">Kobold Presets</span> <h3><span data-i18n="kobldpresets">Kobold Presets</span>
@@ -178,7 +178,7 @@
<div class="range-block-range"> <div class="range-block-range">
<input type="range" id="amount_gen" name="volume" min="16" max="1024" step="1"> <input type="range" id="amount_gen" name="volume" min="16" max="1024" step="1">
</div> </div>
<div class="range-block-counter"> <div class="range-block-counter" data-randomization-disabled="true">
<div contenteditable="true" data-for="amount_gen" id="amount_gen_counter"> <div contenteditable="true" data-for="amount_gen" id="amount_gen_counter">
select select
</div> </div>
@@ -194,7 +194,7 @@
<div class="range-block-range"> <div class="range-block-range">
<input type="range" id="max_context" name="volume" min="512" max="2048" step="1"> <input type="range" id="max_context" name="volume" min="512" max="2048" step="1">
</div> </div>
<div class="range-block-counter"> <div class="range-block-counter" data-randomization-disabled="true">
<div contenteditable="true" data-for="max_context" id="max_context_counter"> <div contenteditable="true" data-for="max_context" id="max_context_counter">
select select
</div> </div>
@@ -535,7 +535,7 @@
<div class="range-block-range"> <div class="range-block-range">
<input type="range" id="openai_max_context" name="volume" min="512" max="4095" step="1"> <input type="range" id="openai_max_context" name="volume" min="512" max="4095" step="1">
</div> </div>
<div class="range-block-counter"> <div class="range-block-counter" data-randomization-disabled="true">
<div contenteditable="true" data-for="openai_max_context" id="openai_max_context_counter"> <div contenteditable="true" data-for="openai_max_context" id="openai_max_context_counter">
select select
</div> </div>
@@ -3624,4 +3624,4 @@
</script> </script>
</body> </body>
</html> </html>

View File

@@ -63,6 +63,11 @@ const extension_settings = {
translate: {}, translate: {},
objective: {}, objective: {},
quickReply: {}, quickReply: {},
randomizer: {
controls: [],
fluctuation: 0.1,
enabled: false,
},
}; };
let modules = []; let modules = [];

View File

@@ -0,0 +1,152 @@
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();
});
});

View File

@@ -0,0 +1,12 @@
{
"display_name": "Parameter Randomizer",
"loading_order": 15,
"requires": [],
"optional": [],
"js": "index.js",
"css": "style.css",
"author": "Cohee#1207",
"version": "1.0.0",
"generate_interceptor": "randomizerInterceptor",
"homePage": "https://github.com/SillyTavern/SillyTavern"
}

View File

@@ -1891,7 +1891,7 @@ grammarly-extension {
aspect-ratio: 1 / 1; aspect-ratio: 1 / 1;
} }
.menu_button:hover { .menu_button:hover, .menu_button.active {
background-color: var(--white30a); background-color: var(--white30a);
} }
@@ -1899,10 +1899,6 @@ grammarly-extension {
height: 32px; height: 32px;
} }
#dialogue_del_mes .menu_button { #dialogue_del_mes .menu_button {
margin-left: 25px; margin-left: 25px;
@@ -2340,6 +2336,7 @@ input[type='checkbox']:not(#nav-toggle):not(#rm_button_panel_pin):not(#lm_button
font-size: calc(var(--mainFontSize) * 0.95); font-size: calc(var(--mainFontSize) * 0.95);
color: var(--SmartThemeBodyColor); color: var(--SmartThemeBodyColor);
flex: 1; flex: 1;
gap: 5px;
display: flex; display: flex;
} }
@@ -5115,4 +5112,4 @@ body.waifuMode .zoomed_avatar {
background-color: var(--SmartThemeBlurTintColor); background-color: var(--SmartThemeBlurTintColor);
text-align: center; text-align: center;
line-height: 14px; line-height: 14px;
} }