change xtts generation settings

public/scripts/extensions/tts/xtts.js

@@ -32,11 +32,19 @@ class XTTSTtsProvider {
         "Hungarian": "hu",
         "Hindi": "hi",
     }
-
+   
     defaultSettings = {
         provider_endpoint: "http://localhost:8020",
         language: "en",
-        voiceMap: {}
+        temperature : 0.75,
+        length_penalty : 1.0,
+        repetition_penalty: 5.0,
+        top_k : 50,
+        top_p : 0.85,
+        speed : 1,
+        enable_text_splitting: true,
+        voiceMap: {},
+        streaming: false,
     }
 
     get settingsHtml() {
@@ -58,9 +66,37 @@ class XTTSTtsProvider {
 
         html += `
         </select>
+
+        <label">XTTS Settings:</label>
+        <label for="xtts_speed">Speed: <span id="xtts_tts_speed_output">${this.defaultSettings.speed}</span></label>
+        <input id="xtts_speed" type="range" value="${this.defaultSettings.speed}" min="0.5" max="2" step="0.1" />
+
+        <label for="xtts_temperature">Temperature: <span id="xtts_tts_temperature_output">${this.defaultSettings.temperature}</span></label>
+        <input id="xtts_temperature" type="range" value="${this.defaultSettings.temperature}" min="0.01" max="1" step="0.01" />
+
+        <label for="xtts_length_penalty">Length Penalty: <span id="xtts_length_penalty_output">${this.defaultSettings.length_penalty}</span></label>
+        <input id="xtts_length_penalty" type="range" value="${this.defaultSettings.length_penalty}" min="0.5" max="2" step="0.1" />
+
+        <label for="xtts_repetition_penalty">Repetition Penalty: <span id="xtts_repetition_penalty_output">${this.defaultSettings.repetition_penalty}</span></label>
+        <input id="xtts_repetition_penalty" type="range" value="${this.defaultSettings.repetition_penalty}" min="1" max="10" step="0.1" />
+
+        <label for="xtts_top_k">Top K: <span id="xtts_top_k_output">${this.defaultSettings.top_k}</span></label>
+        <input id="xtts_top_k" type="range" value="${this.defaultSettings.top_k}" min="0" max="100" step="1" />
+
+        <label for="xtts_top_p">Top P: <span id="xtts_top_p_output">${this.defaultSettings.top_p}</span></label>
+        <input id="xtts_top_p" type="range" value="${this.defaultSettings.top_p}" min="0" max="1" step="0.01" />
+
+        <label for="xtts_enable_text_splitting">Enable Text Splitting:</label>
+        <input id="xtts_enable_text_splitting" type="checkbox" ${this.defaultSettings.enable_text_splitting ? 'checked' : ''} />
+
         <label for="xtts_tts_endpoint">Provider Endpoint:</label>
         <input id="xtts_tts_endpoint" type="text" class="text_pole" maxlength="250" value="${this.defaultSettings.provider_endpoint}"/>
 
+        <label for="xtts_tts_streaming" class="checkbox_label">
+            <input id="xtts_tts_streaming" type="checkbox" />
+            <span>Streaming <small>(RVC not supported)</small></span>
+        </label>
+
         `;
 
         html += `
@@ -75,9 +111,30 @@ class XTTSTtsProvider {
         // Used when provider settings are updated from UI
         this.settings.provider_endpoint = $('#xtts_tts_endpoint').val()
         this.settings.language = $('#xtts_api_language').val()
+
+         // Update the default TTS settings based on input fields
+        this.settings.speed = $('#xtts_speed').val();
+        this.settings.temperature = $('#xtts_temperature').val();
+        this.settings.length_penalty = $('#xtts_length_penalty').val();
+        this.settings.repetition_penalty = $('#xtts_repetition_penalty').val();
+        this.settings.top_k = $('#xtts_top_k').val();
+        this.settings.top_p = $('#xtts_top_p').val();
+        this.settings.enable_text_splitting = $('#xtts_enable_text_splitting').is(':checked');
+        this.settings.streaming = $('#xtts_tts_streaming').is(':checked');
+        
+        // Update the UI to reflect changes
+        $('#xtts_tts_speed_output').text(this.settings.speed);
+        $('#xtts_tts_temperature_output').text(this.settings.temperature);
+        $('#xtts_length_penalty_output').text(this.settings.length_penalty);
+        $('#xtts_repetition_penalty_output').text(this.settings.repetition_penalty);
+        $('#xtts_top_k_output').text(this.settings.top_k);
+        $('#xtts_top_p_output').text(this.settings.top_p);
+        
         saveTtsProviderSettings()
+        this.changeTTSSetting()
     }
 
+
     async loadSettings(settings) {
         // Pupulate Provider UI given input settings
         if (Object.keys(settings).length == 0) {
@@ -111,6 +168,27 @@ class XTTSTtsProvider {
         $('#xtts_api_language').val(this.settings.language)
         $('#xtts_api_language').on("change", () => { this.onSettingsChange() })
 
+        // Set initial values from the settings
+        $('#xtts_speed').val(this.settings.speed);
+        $('#xtts_temperature').val(this.settings.temperature);
+        $('#xtts_length_penalty').val(this.settings.length_penalty);
+        $('#xtts_repetition_penalty').val(this.settings.repetition_penalty);
+        $('#xtts_top_k').val(this.settings.top_k);
+        $('#xtts_top_p').val(this.settings.top_p);
+        $('#xtts_enable_text_splitting').prop('checked', this.settings.enable_text_splitting);
+
+        // Register input/change event listeners to update settings on user interaction
+        $('#xtts_speed').on("input", () => { this.onSettingsChange() });
+        $('#xtts_temperature').on("input", () => { this.onSettingsChange() });
+        $('#xtts_length_penalty').on("input", () => { this.onSettingsChange() });
+        $('#xtts_repetition_penalty').on("input", () => { this.onSettingsChange() });
+        $('#xtts_top_k').on("input", () => { this.onSettingsChange() });
+        $('#xtts_top_p').on("input", () => { this.onSettingsChange() });
+        $('#xtts_enable_text_splitting').on("change", () => { this.onSettingsChange() });
+
+        $('#xtts_tts_streaming').prop('checked', this.settings.streaming);
+        $('#xtts_tts_streaming').on('change', () => { this.onSettingsChange(); });
+
         await this.checkReady()
 
         console.debug("XTTS: Settings loaded")
@@ -160,8 +238,45 @@ class XTTSTtsProvider {
         return responseJson
     }
 
+    // Each time a parameter is changed, we change the configuration
+    async changeTTSSetting() {
+        const response = await doExtrasFetch(
+            `${this.settings.provider_endpoint}/set_tts_settings`,
+            {
+                method: 'POST',
+                headers: {
+                    'Content-Type': 'application/json',
+                    'Cache-Control': 'no-cache'
+                },
+                body: JSON.stringify({
+                    "temperature": this.settings.temperature,
+                    "speed": this.settings.speed,
+                    "length_penalty": this.settings.length_penalty,
+                    "repetition_penalty": this.settings.repetition_penalty,
+                    "top_p": this.settings.top_p,
+                    "top_k": this.settings.top_k,
+                    "enable_text_splitting": this.settings.enable_text_splitting,
+                })
+            }
+        )
+        if (!response.ok) {
+            toastr.error(response.statusText, 'An error occurred when changing TTS settings');
+            throw new Error(`HTTP ${response.status}: ${await response.text()}`);
+        }
+        return response
+    }
+
     async fetchTtsGeneration(inputText, voiceId) {
         console.info(`Generating new TTS for voice_id ${voiceId}`)
+
+        if (this.settings.streaming) {
+            const params = new URLSearchParams();
+            params.append('text', inputText);
+            params.append('speaker_wav', voiceId);
+            params.append('language', this.settings.language);
+            return `${this.settings.provider_endpoint}/tts_stream/?${params.toString()}`;
+        }
+
         const response = await doExtrasFetch(
             `${this.settings.provider_endpoint}/tts_to_audio/`,
             {