Fix elevenlabs multi-model

This commit is contained in:
Cohee 2023-10-22 14:46:54 +03:00
parent f49f9c1f96
commit 0517f1bbbc
2 changed files with 44 additions and 29 deletions

View File

@ -10,15 +10,12 @@ class ElevenLabsTtsProvider {
voices = [] voices = []
separator = ' ... ... ... ' separator = ' ... ... ... '
get settings() {
return this.settings
}
defaultSettings = { defaultSettings = {
stability: 0.75, stability: 0.75,
similarity_boost: 0.75, similarity_boost: 0.75,
apiKey: "", apiKey: "",
multilingual: false, model: 'eleven_monolingual_v1',
voiceMap: {} voiceMap: {}
} }
@ -27,15 +24,17 @@ class ElevenLabsTtsProvider {
<div class="elevenlabs_tts_settings"> <div class="elevenlabs_tts_settings">
<label for="elevenlabs_tts_api_key">API Key</label> <label for="elevenlabs_tts_api_key">API Key</label>
<input id="elevenlabs_tts_api_key" type="text" class="text_pole" placeholder="<API Key>"/> <input id="elevenlabs_tts_api_key" type="text" class="text_pole" placeholder="<API Key>"/>
<label for="elevenlabs_tts_model">Model</label>
<select id="elevenlabs_tts_model" class="text_pole">
<option value="eleven_monolingual_v1">Monolingual</option>
<option value="eleven_multilingual_v1">Multilingual v1</option>
<option value="eleven_multilingual_v2">Multilingual v2</option>
</select>
<input id="eleven_labs_connect" class="menu_button" type="button" value="Connect" /> <input id="eleven_labs_connect" class="menu_button" type="button" value="Connect" />
<label for="elevenlabs_tts_stability">Stability: <span id="elevenlabs_tts_stability_output"></span></label> <label for="elevenlabs_tts_stability">Stability: <span id="elevenlabs_tts_stability_output"></span></label>
<input id="elevenlabs_tts_stability" type="range" value="${this.defaultSettings.stability}" min="0" max="1" step="0.05" /> <input id="elevenlabs_tts_stability" type="range" value="${this.defaultSettings.stability}" min="0" max="1" step="0.05" />
<label for="elevenlabs_tts_similarity_boost">Similarity Boost: <span id="elevenlabs_tts_similarity_boost_output"></span></label> <label for="elevenlabs_tts_similarity_boost">Similarity Boost: <span id="elevenlabs_tts_similarity_boost_output"></span></label>
<input id="elevenlabs_tts_similarity_boost" type="range" value="${this.defaultSettings.similarity_boost}" min="0" max="1" step="0.05" /> <input id="elevenlabs_tts_similarity_boost" type="range" value="${this.defaultSettings.similarity_boost}" min="0" max="1" step="0.05" />
<label class="checkbox_label" for="elevenlabs_tts_multilingual">
<input id="elevenlabs_tts_multilingual" type="checkbox" value="${this.defaultSettings.multilingual}" />
Enable Multilingual
</label>
</div> </div>
` `
return html return html
@ -45,11 +44,10 @@ class ElevenLabsTtsProvider {
// Update dynamically // Update dynamically
this.settings.stability = $('#elevenlabs_tts_stability').val() this.settings.stability = $('#elevenlabs_tts_stability').val()
this.settings.similarity_boost = $('#elevenlabs_tts_similarity_boost').val() this.settings.similarity_boost = $('#elevenlabs_tts_similarity_boost').val()
this.settings.multilingual = $('#elevenlabs_tts_multilingual').prop('checked') this.settings.model = $('#elevenlabs_tts_model').find(':selected').val()
saveTtsProviderSettings() saveTtsProviderSettings()
} }
async loadSettings(settings) { async loadSettings(settings) {
// Pupulate Provider UI given input settings // Pupulate Provider UI given input settings
if (Object.keys(settings).length == 0) { if (Object.keys(settings).length == 0) {
@ -59,26 +57,39 @@ class ElevenLabsTtsProvider {
// Only accept keys defined in defaultSettings // Only accept keys defined in defaultSettings
this.settings = this.defaultSettings this.settings = this.defaultSettings
for (const key in settings){ // Migrate old settings
if (key in this.settings){ if (settings['multilingual'] !== undefined) {
settings.model = settings.multilingual ? 'eleven_multilingual_v1' : 'eleven_monolingual_v1';
delete settings['multilingual'];
}
for (const key in settings) {
if (key in this.settings) {
this.settings[key] = settings[key] this.settings[key] = settings[key]
} else { } else {
throw `Invalid setting passed to TTS Provider: ${key}` throw `Invalid setting passed to TTS Provider: ${key}`
} }
} }
$('#elevenlabs_tts_stability').val(this.settings.stability) $('#elevenlabs_tts_stability').val(this.settings.stability)
$('#elevenlabs_tts_similarity_boost').val(this.settings.similarity_boost) $('#elevenlabs_tts_similarity_boost').val(this.settings.similarity_boost)
$('#elevenlabs_tts_api_key').val(this.settings.apiKey) $('#elevenlabs_tts_api_key').val(this.settings.apiKey)
$('#elevenlabs_tts_multilingual').prop('checked', this.settings.multilingual) $('#elevenlabs_tts_model').val(this.settings.model);
$('#eleven_labs_connect').on('click', () => {this.onConnectClick()}) $('#eleven_labs_connect').on('click', () => { this.onConnectClick() })
$('#elevenlabs_tts_settings').on('input',this.onSettingsChange) $('#elevenlabs_tts_similarity_boost').on('input', this.onSettingsChange.bind(this))
$('#elevenlabs_tts_stability').on('input', this.onSettingsChange.bind(this))
$('#elevenlabs_tts_model').on('change', this.onSettingsChange.bind(this))
await this.checkReady() try {
console.debug("ElevenLabs: Settings loaded") await this.checkReady()
console.debug("ElevenLabs: Settings loaded")
} catch {
console.debug("ElevenLabs: Settings loaded, but not ready")
}
} }
// Perform a simple readiness check by trying to fetch voiceIds // Perform a simple readiness check by trying to fetch voiceIds
async checkReady(){ async checkReady() {
await this.fetchTtsVoiceObjects() await this.fetchTtsVoiceObjects()
} }
@ -87,7 +98,7 @@ class ElevenLabsTtsProvider {
async onConnectClick() { async onConnectClick() {
// Update on Apply click // Update on Apply click
return await this.updateApiKey().catch( (error) => { return await this.updateApiKey().catch((error) => {
toastr.error(`ElevenLabs: ${error}`) toastr.error(`ElevenLabs: ${error}`)
}) })
} }
@ -102,6 +113,7 @@ class ElevenLabsTtsProvider {
}) })
this.settings.apiKey = this.settings.apiKey this.settings.apiKey = this.settings.apiKey
console.debug(`Saved new API_KEY: ${this.settings.apiKey}`) console.debug(`Saved new API_KEY: ${this.settings.apiKey}`)
$('#tts_status').text('')
this.onSettingsChange() this.onSettingsChange()
} }
@ -123,7 +135,7 @@ class ElevenLabsTtsProvider {
} }
async generateTts(text, voiceId){ async generateTts(text, voiceId) {
const historyId = await this.findTtsGenerationInHistory(text, voiceId) const historyId = await this.findTtsGenerationInHistory(text, voiceId)
let response let response
@ -189,11 +201,8 @@ class ElevenLabsTtsProvider {
} }
async fetchTtsGeneration(text, voiceId) { async fetchTtsGeneration(text, voiceId) {
let model = "eleven_monolingual_v1" let model = this.settings.model ?? "eleven_monolingual_v1";
if (this.settings.multilingual == true) { console.info(`Generating new TTS for voice_id ${voiceId}, model ${model}`)
model = "eleven_multilingual_v1"
}
console.info(`Generating new TTS for voice_id ${voiceId}`)
const response = await fetch( const response = await fetch(
`https://api.elevenlabs.io/v1/text-to-speech/${voiceId}`, `https://api.elevenlabs.io/v1/text-to-speech/${voiceId}`,
{ {
@ -203,9 +212,12 @@ class ElevenLabsTtsProvider {
'Content-Type': 'application/json' 'Content-Type': 'application/json'
}, },
body: JSON.stringify({ body: JSON.stringify({
model: model, model_id: model,
text: text, text: text,
voice_settings: this.settings voice_settings: {
stability: Number(this.settings.stability),
similarity_boost: Number(this.settings.similarity_boost),
},
}) })
} }
) )

View File

@ -429,7 +429,7 @@ async function processTtsQueue() {
console.debug('New message found, running TTS') console.debug('New message found, running TTS')
currentTtsJob = ttsJobQueue.shift() currentTtsJob = ttsJobQueue.shift()
let text = extension_settings.tts.narrate_translated_only ? currentTtsJob?.extra?.display_text : currentTtsJob.mes let text = extension_settings.tts.narrate_translated_only ? (currentTtsJob?.extra?.display_text || currentTtsJob.mes) : currentTtsJob.mes
text = extension_settings.tts.narrate_dialogues_only text = extension_settings.tts.narrate_dialogues_only
? text.replace(/\*[^\*]*?(\*|$)/g, '').trim() // remove asterisks content ? text.replace(/\*[^\*]*?(\*|$)/g, '').trim() // remove asterisks content
: text.replaceAll('*', '').trim() // remove just the asterisks : text.replaceAll('*', '').trim() // remove just the asterisks
@ -614,8 +614,8 @@ function onTtsProviderChange() {
// Ensure that TTS provider settings are saved to extension settings. // Ensure that TTS provider settings are saved to extension settings.
export function saveTtsProviderSettings() { export function saveTtsProviderSettings() {
updateVoiceMap()
extension_settings.tts[ttsProviderName] = ttsProvider.settings extension_settings.tts[ttsProviderName] = ttsProvider.settings
updateVoiceMap()
saveSettingsDebounced() saveSettingsDebounced()
console.info(`Saved settings ${ttsProviderName} ${JSON.stringify(ttsProvider.settings)}`) console.info(`Saved settings ${ttsProviderName} ${JSON.stringify(ttsProvider.settings)}`)
} }
@ -695,6 +695,9 @@ function updateVoiceMap() {
voiceMap = tempVoiceMap voiceMap = tempVoiceMap
console.log(`Voicemap updated to ${JSON.stringify(voiceMap)}`) console.log(`Voicemap updated to ${JSON.stringify(voiceMap)}`)
} }
if (!extension_settings.tts[ttsProviderName].voiceMap) {
extension_settings.tts[ttsProviderName].voiceMap = {}
}
Object.assign(extension_settings.tts[ttsProviderName].voiceMap, voiceMap) Object.assign(extension_settings.tts[ttsProviderName].voiceMap, voiceMap)
saveSettingsDebounced() saveSettingsDebounced()
} }