Pendrokar's picture
relocate folders
ed18ebf
raw
history blame contribute delete
No virus
70.6 kB
"use strict"
window.appVersion = "v3.0.3"
const {Editor} = require("./javascript/editor.js")
require("./javascript/textarea.js")
const xVASpeech = require("./javascript/speech2speech.js")
// Newly introduced in v3. I will slowly start moving global context variables into this, and update code throughout to reference this
// instead of old variables such as window.games, window.currentModel, etc.
window.appState = {}
// Start the server
if (window.PRODUCTION) {
window.pythonProcess = spawn(`${path}/cpython_${window.userSettings.installation}/server.exe`, {stdio: "ignore"})
}
let themeColour
let secondaryThemeColour
const oldCError = console.error
console.error = (...rest) => {
window.appLogger.log(`console.error: ${rest}`)
oldCError(rest)
}
window.addEventListener("error", function (e) {window.appLogger.log(`error: ${e.error.stack}`)})
window.addEventListener('unhandledrejection', function (e) {window.appLogger.log(`unhandledrejection: ${e.stack}`)})
setTimeout(() => {
window.electron = require("electron")
}, 1000)
window.games = {}
window.models = {}
window.sequenceEditor = new Editor()
window.currentModel = undefined
window.currentModelButton = undefined
window.watchedModelsDirs = []
window.appLogger.log(`Settings: ${JSON.stringify(window.userSettings)}`)
// Set up folders
try {fs.mkdirSync(`${path}/models`)} catch (e) {/*Do nothing*/}
try {fs.mkdirSync(`${path}/output`)} catch (e) {/*Do nothing*/}
try {fs.mkdirSync(`${path}/assets`)} catch (e) {/*Do nothing*/}
// Clean up temp files
const clearOldTempFiles = () => {
fs.readdir(`${__dirname.replace("/javascript", "")}/output`, (err, files) => {
if (err) {
window.appLogger.log(`Error cleaning up temp files: ${err}`)
}
if (files && files.length) {
files.filter(f => f.startsWith("temp-")).forEach(file => {
fs.unlink(`${__dirname.replace("/javascript", "")}/output/${file}`, err => err&&console.log(err))
})
}
})
}
clearOldTempFiles()
let fileRenameCounter = 0
let fileChangeCounter = 0
window.isGenerating = false
window.registerModel = (modelsPath, gameFolder, model, {gameId, voiceId, voiceName, voiceDescription, gender, variant, modelType, emb_i}) => {
// Add game, if the game hasn't been added yet
let audioPreviewPath
try {
audioPreviewPath = `${modelsPath}/${model.games.find(({gameId}) => gameId==gameFolder).voiceId}`
} catch (e) {}
if (!window.games.hasOwnProperty(gameId)) {
const gameAsset = fs.readdirSync(`${path}/assets`).find(f => f==gameId+".json")
if (gameAsset && gameAsset.length && gameAsset[0]) {
const gameTheme = JSON.parse(fs.readFileSync(`${path}/assets/${gameAsset}`))
window.games[gameId] = {
models: [],
gameTheme,
gameAsset
}
} else {
window.appLogger.log(`Something not right with loading model: ${voiceId} . The asset file for its game (${gameId}) could not be found here: ${path}/assets. You need a ${gameId}.json file. Loading a generic theme for this voice's game/category.`)
const dummyGameTheme = {
"gameName": gameId,
"assetFile": "other.jpg",
"themeColourPrimary": "aaaaaa",
"themeColourSecondary": null,
"gameCode": "x",
"nexusGamePageIDs": []
}
const dummyGameAsset = "other.json"
window.games[gameId] = {
models: [],
dummyGameTheme,
dummyGameAsset
}
audioPreviewPath = `${modelsPath}/${model.games[0].voiceId}`
}
}
// Catch duplicates, for when/if a model is registered for multiple games, but there's already the same model in that game, from another version
// const existingDuplicates = []
// window.games[gameId].models.forEach((item,i) => {
// if (item.voiceId==voiceId) {
// existingDuplicates.push([item, i])
// }
// })
// Check if a variant has already been added for this voice name, for this game
let foundVariantIndex = undefined
window.games[gameId].models.forEach((item,i) => {
if (foundVariantIndex!=undefined) return
if (item.voiceName.toLowerCase().trim()==voiceName.toLowerCase().trim()) {
foundVariantIndex = i
}
})
// Add the initial model metadata, if no existing variant has been added (will happen most of the time)
if (!foundVariantIndex) {
const modelData = {
gameId,
modelsPath,
voiceName,
lang_capabilities: model.lang_capabilities,
embOverABaseModel: model.embOverABaseModel,
variants: []
}
window.games[gameId].models.push(modelData)
foundVariantIndex = window.games[gameId].models.length-1
}
const variantData = {
author: model.author,
version: model.version,
modelVersion: model.modelVersion,
modelType: model.modelType,
base_speaker_emb: model.modelType=="xVAPitch" ? model.games[0].base_speaker_emb : undefined,
voiceId,
audioPreviewPath,
hifi: undefined,
num_speakers: model.emb_size,
emb_i,
variantName: variant ? variant.replace("Default :", "Default:").replace("Default:", "").trim() : "Default",
voiceDescription,
lang: model.lang,
gender,
modelType: modelType||model.modelType,
model,
}
const potentialHiFiPath = `${modelsPath}/${voiceId}.hg.pt`
if (fs.existsSync(potentialHiFiPath)) {
variantData.hifi = potentialHiFiPath
}
const isDefaultVariant = !variant || variant.toLowerCase().startsWith("default")
if (isDefaultVariant) {
// Place first in the list, if it's default
window.games[gameId].models[foundVariantIndex].audioPreviewPath = audioPreviewPath
window.games[gameId].models[foundVariantIndex].variants.splice(0,0,variantData)
} else {
window.games[gameId].models[foundVariantIndex].variants.push(variantData)
}
// // Using the detected duplicates, use only the latest version
// if (existingDuplicates.length) {
// if (existingDuplicates[0][0].modelVersion<model.modelVersion) {
// window.games[gameId].models.splice(existingDuplicates[0][1], 1)
// window.games[gameId].models.push(modelData)
// }
// } else {
// window.games[gameId].models.push(modelData)
// }
}
window.loadAllModels = (forceUpdate=false) => {
return new Promise(resolve => {
if (!forceUpdate && window.nexusState.installQueue.length) {
return
}
let gameFolder
let modelPathsKeys = Object.keys(window.userSettings).filter(key => key.includes("modelspath_"))
window.games = {}
// Do the current game first, and stop blocking the render process
if (window.currentGame) {
const currentGameFolder = window.userSettings[`modelspath_${window.currentGame.gameId}`]
gameFolder = currentGameFolder
try {
const files = fs.readdirSync(modelsPath).filter(f => f.endsWith(".json"))
files.forEach(fileName => {
try {
if (!models.hasOwnProperty(`${gameFolder}/${fileName}`)) {
models[`${gameFolder}/${fileName}`] = null
}
const model = JSON.parse(fs.readFileSync(`${modelsPath}/${fileName}`, "utf8"))
model.games.forEach(({gameId, voiceId, voiceName, voiceDescription, gender, variant, modelType, emb_i}) => {
window.registerModel(currentGameFolder, gameFolder, model, {gameId, voiceId, voiceName, voiceDescription, gender, variant, modelType, emb_i})
})
} catch (e) {
console.log(e)
// window.appLogger.log(`${window.i18n.ERR_LOADING_MODELS_FOR_GAME_WITH_FILENAME.replace("_1", gameFolder)} `+fileName)
// window.appLogger.log(e)
// window.appLogger.log(e.stack)
}
})
} catch (e) {
window.appLogger.log(`${window.i18n.ERR_LOADING_MODELS_FOR_GAME}: `+ gameFolder)
window.appLogger.log(e)
}
resolve() // Continue the rest but asynchronously
}
modelPathsKeys.forEach(modelsPathKey => {
const modelsPath = window.userSettings[modelsPathKey]
try {
const files = fs.readdirSync(modelsPath).filter(f => f.endsWith(".json"))
if (!files.length) {
return
}
files.forEach(fileName => {
gameFolder = modelsPathKey.split("_")[1]
try {
if (!models.hasOwnProperty(`${gameFolder}/${fileName}`)) {
models[`${gameFolder}/${fileName}`] = null
}
const model = JSON.parse(fs.readFileSync(`${modelsPath}/${fileName}`, "utf8"))
model.games.forEach(({gameId, voiceId, voiceName, voiceDescription, gender, variant, modelType, emb_i}) => {
window.registerModel(modelsPath, gameFolder, model, {gameId, voiceId, voiceName, voiceDescription, gender, variant, modelType, emb_i})
})
} catch (e) {
console.log(e)
setTimeout(() => {
window.errorModal(`${fileName}<br><br>${e.stack}`)
}, 1000)
window.appLogger.log(`${window.i18n.ERR_LOADING_MODELS_FOR_GAME_WITH_FILENAME.replace("_1", gameFolder)} `+fileName)
window.appLogger.log(e)
window.appLogger.log(e.stack)
}
})
} catch (e) {
window.appLogger.log(`${window.i18n.ERR_LOADING_MODELS_FOR_GAME}: `+ gameFolder)
window.appLogger.log(e)
}
})
window.updateGameList(false)
resolve()
})
}
// Change variant
let oldVariantSelection = undefined // For reverting, if versioning checks fail
variant_select.addEventListener("change", () => {
const model = window.games[window.currentGame.gameId].models.find(model => model.voiceName== window.currentModel.voiceName)
const variant = model.variants.find(variant => variant.variantName==variant_select.value)
const appVersionOk = window.checkVersionRequirements(variant.version, appVersion)
if (!appVersionOk) {
window.errorModal(`${window.i18n.MODEL_REQUIRES_VERSION} v${variant.version}<br><br>${window.i18n.THIS_APP_VERSION}: ${window.appVersion}`)
variant_select.value = oldVariantSelection
return
}
generateVoiceButton.dataset.modelQuery = JSON.stringify({
outputs: parseInt(model.outputs),
model: model.embOverABaseModel ? window.userSettings[`modelspath_${model.embOverABaseModel.split("/")[0]}`]+`/${model.embOverABaseModel.split("/")[1]}` : `${model.modelsPath}/${variant.voiceId}`,
modelType: variant.modelType,
version: variant.version,
model_speakers: model.num_speakers,
base_lang: model.lang || "en"
})
oldVariantSelection = variant_select.value
titleInfoVoiceID.innerHTML = variant.voiceId
titleInfoGender.innerHTML = variant.gender || "?"
titleInfoAppVersion.innerHTML = variant.version || "?"
titleInfoModelVersion.innerHTML = variant.modelVersion || "?"
titleInfoModelType.innerHTML = variant.modelType || "?"
titleInfoLanguage.innerHTML = variant.lang || window.currentModel.games[0].lang || "en"
titleInfoAuthor.innerHTML = variant.author || "?"
generateVoiceButton.click()
})
// Change game
window.changeGame = (meta) => {
titleInfo.style.display = "none"
window.currentGame = meta
themeColour = meta.themeColourPrimary
secondaryThemeColour = meta.themeColourSecondary
let titleID = meta.gameCode
generateVoiceButton.disabled = true
generateVoiceButton.innerHTML = window.i18n.GENERATE_VOICE
selectedGameDisplay.innerHTML = meta.gameName
// Change the app title
titleName.innerHTML = window.i18n.SELECT_VOICE_TYPE
if (window.games[window.currentGame.gameId] == undefined) {
titleName.innerHTML = `${window.i18n.NO_MODELS_IN}: ${window.userSettings[`modelspath_${window.currentGame.gameId}`]}`
}
const gameFolder = meta.gameId
const gameName = meta.gameName
setting_models_path_container.style.display = "flex"
setting_out_path_container.style.display = "flex"
setting_models_path_label.innerHTML = `<i style="display:inline">${gameName}</i><span>${window.i18n.SETTINGS_MODELS_PATH}</span>`
setting_models_path_input.value = window.userSettings[`modelspath_${gameFolder}`]
setting_out_path_label.innerHTML = `<i style="display:inline">${gameName}</i> ${window.i18n.SETTINGS_OUTPUT_PATH}`
setting_out_path_input.value = window.userSettings[`outpath_${gameFolder}`]
window.setTheme(window.currentGame)
try {
window.displayAllModels()
} catch (e) {console.log(e)}
try {fs.mkdirSync(`${path}/output/${meta.gameId}`)} catch (e) {/*Do nothing*/}
localStorage.setItem("lastGame", JSON.stringify(meta))
// Populate models
voiceTypeContainer.innerHTML = ""
voiceSamples.innerHTML = ""
const buttons = []
const totalNumVoices = (window.games[meta.gameId] ? window.games[meta.gameId].models : []).reduce((p,c)=>p+c.variants.length, 0)
voiceSearchInput.placeholder = window.i18n.SEARCH_N_VOICES.replace("_", window.games[meta.gameId] ? totalNumVoices : "0")
voiceSearchInput.value = ""
if (!window.games[meta.gameId]) {
return
}
(window.games[meta.gameId] ? window.games[meta.gameId].models : []).forEach(({modelsPath, audioPreviewPath, gameId, variants, voiceName, embOverABaseModel}) => {
const {voiceId, voiceDescription, hifi, model} = variants[0]
const modelVersion = variants[0].version
const button = createElem("div.voiceType", voiceName)
button.style.background = `#${themeColour}`
if (embOverABaseModel) {
button.style.fontStyle = "italic"
}
if (window.userSettings.do_model_version_highlight && parseFloat(modelVersion)<window.userSettings.model_version_highlight) {
button.style.border = `2px solid #${themeColour}`
button.style.padding = "0"
button.style.background = "none"
}
button.dataset.modelId = voiceId
if (secondaryThemeColour) {
button.style.color = `#${secondaryThemeColour}`
button.style.textShadow = `none`
}
// Quick voice set preview, if there is a preview file
button.addEventListener("contextmenu", () => {
window.appLogger.log(`${audioPreviewPath}.wav`)
const audioPreview = createElem("audio", {autoplay: false}, createElem("source", {
src: `${audioPreviewPath}.wav`
}))
audioPreview.style.height = "25px"
audioPreview.setSinkId(window.userSettings.base_speaker)
})
if (embOverABaseModel) {
const gameOfBaseModel = embOverABaseModel.split("/")[0]
if (gameOfBaseModel=="<base>") {
// For included base v3 models
modelsPath = `${window.path}/python/xvapitch/${embOverABaseModel.split("/")[1]}`
} else {
// For any other model
const gameModelsPath = `${window.userSettings[`outpath_${gameOfBaseModel}`]}`
modelsPath = `${gameModelsPath}/${embOverABaseModel.split("/")[1]}`
}
}
button.addEventListener("click", event => window.selectVoice(event, variants, hifi, gameId, voiceId, model, button, audioPreviewPath, modelsPath, meta, embOverABaseModel))
buttons.push(button)
})
buttons.sort((a,b) => a.innerHTML.toLowerCase()<b.innerHTML.toLowerCase()?-1:1)
.forEach(button => voiceTypeContainer.appendChild(button))
}
window.selectVoice = (event, variants, hifi, gameId, voiceId, model, button, audioPreviewPath, modelsPath, meta, embOverABaseModel) => {
// Just for easier packaging of the voice models for publishing - yes, lazy
if (event.ctrlKey && event.shiftKey) {
window.packageVoice(event.altKey, variants, {modelsPath, gameId})
}
variant_select.innerHTML = ""
oldVariantSelection = undefined
if (variants.length==1) {
variantElements.style.display = "none"
} else {
variantElements.style.display = "flex"
variants.forEach(variant => {
const option = createElem("option", {value: variant.variantName})
option.innerHTML = variant.variantName
variant_select.appendChild(option)
if (!oldVariantSelection) {
oldVariantSelection = variant.variantName
}
})
}
if (hifi) {
// Remove the bespoke hifi option if there was one already there
Array.from(vocoder_select.children).forEach(opt => {
if (opt.innerHTML=="Bespoke HiFi GAN") {
vocoder_select.removeChild(opt)
}
})
bespoke_hifi_bolt.style.opacity = 1
const option = createElem("option", "Bespoke HiFi GAN")
option.value = `${gameId}/${voiceId}.hg.pt`
vocoder_select.appendChild(option)
} else {
bespoke_hifi_bolt.style.opacity = 0
// Set the vocoder select to quick-and-dirty if bespoke hifi-gan was selected
if (vocoder_select.value.includes(".hg.")) {
vocoder_select.value = "qnd"
window.changeVocoder("qnd")
}
// Remove the bespoke hifi option if there was one already there
Array.from(vocoder_select.children).forEach(opt => {
if (opt.innerHTML=="Bespoke HiFi GAN") {
vocoder_select.removeChild(opt)
}
})
}
window.currentModel = model
window.currentModel.voiceId = voiceId
window.currentModel.voiceName = button.innerHTML
window.currentModel.hifi = hifi
window.currentModel.audioPreviewPath = audioPreviewPath
window.currentModelButton = button
generateVoiceButton.dataset.modelQuery = null
// The model is already loaded. Don't re-load it.
if (generateVoiceButton.dataset.modelIDLoaded == voiceId) {
generateVoiceButton.innerHTML = window.i18n.GENERATE_VOICE
generateVoiceButton.dataset.modelQuery = "null"
} else {
generateVoiceButton.innerHTML = window.i18n.LOAD_MODEL
generateVoiceButton.dataset.modelQuery = JSON.stringify({
outputs: parseInt(model.outputs),
model: model.embOverABaseModel ? window.userSettings[`modelspath_${model.embOverABaseModel.split("/")[0]}`]+`/${model.embOverABaseModel.split("/")[1]}` : `${modelsPath}/${model.voiceId}`,
modelType: model.modelType,
version: model.version,
model_speakers: model.emb_size,
cmudict: model.cmudict,
base_lang: model.lang || "en"
})
generateVoiceButton.dataset.modelIDToLoad = voiceId
}
generateVoiceButton.disabled = false
titleName.innerHTML = button.innerHTML
titleInfo.style.display = "flex"
titleInfoName.innerHTML = window.currentModel.voiceName
titleInfoVoiceID.innerHTML = voiceId
titleInfoGender.innerHTML = window.currentModel.games[0].gender || "?"
titleInfoAppVersion.innerHTML = window.currentModel.version || "?"
titleInfoModelVersion.innerHTML = window.currentModel.modelVersion || "?"
titleInfoModelType.innerHTML = window.currentModel.modelType || "?"
titleInfoLanguage.innerHTML = window.currentModel.lang || window.currentModel.games[0].lang || "en"
titleInfoAuthor.innerHTML = window.currentModel.author || "?"
titleInfoLicense.innerHTML = window.currentModel.license || window.i18n.UNKNOWN
title.dataset.modelId = voiceId
keepSampleButton.style.display = "none"
samplePlayPause.style.display = "none"
// Voice samples
voiceSamples.innerHTML = ""
window.initMainPagePagination(`${window.userSettings[`outpath_${meta.gameId}`]}/${button.dataset.modelId}`)
window.refreshRecordsList()
}
titleInfo.addEventListener("click", () => titleDetails.style.display = titleDetails.style.display=="none" ? "block" : "none")
window.addEventListener("click", event => {
if (event.target!=titleInfo && event.target!=titleDetails && event.target.parentNode && event.target.parentNode!=titleDetails && event.target.parentNode.parentNode!=titleDetails) {
titleDetails.style.display = "none"
}
})
titleDetails.style.display = "none"
window.loadModel = () => {
return new Promise(resolve => {
if (window.batch_state.state) {
window.errorModal(window.i18n.BATCH_ERR_IN_PROGRESS)
return
}
const body = JSON.parse(generateVoiceButton.dataset.modelQuery)
const appVersionOk = window.checkVersionRequirements(body.version, appVersion)
if (!appVersionOk) {
window.errorModal(`${window.i18n.MODEL_REQUIRES_VERSION} v${body.version}<br><br>${window.i18n.THIS_APP_VERSION}: ${window.appVersion}`)
return
}
window.appLogger.log(`${window.i18n.LOADING_VOICE}: ${JSON.parse(generateVoiceButton.dataset.modelQuery).model}`)
window.batch_state.lastModel = JSON.parse(generateVoiceButton.dataset.modelQuery).model.split("/").reverse()[0]
body["pluginsContext"] = JSON.stringify(window.pluginsContext)
spinnerModal(`${window.i18n.LOADING_VOICE}`)
doFetch(`http://localhost:8008/loadModel`, {
method: "Post",
body: JSON.stringify(body)
}).then(r=>r.text()).then(res => {
window.currentModel.loaded = true
generateVoiceButton.dataset.modelQuery = null
generateVoiceButton.innerHTML = window.i18n.GENERATE_VOICE
generateVoiceButton.dataset.modelIDLoaded = generateVoiceButton.dataset.modelIDToLoad
// Set the editor pitch/energy dropdowns to pitch, and freeze them, if energy is not supported by the model
window.appState.currentModelEmbeddings = {}
if (window.currentModel.modelType.toLowerCase()=="xvapitch" && !window.currentModel.isBaseModel) {
vocoder_options_container.style.display = "none"
base_lang_select.disabled = false
style_emb_select.disabled = false
window.loadStyleEmbsForVoice(window.currentModel)
mic_SVG.children[0].style.fill = "white"
base_lang_select.value = window.currentModel.lang
} else {
vocoder_options_container.style.display = "inline-block"
base_lang_select.disabled = true
style_emb_select.disabled = true
mic_SVG.children[0].style.fill = "grey"
}
if (window.currentModel.modelType.toLowerCase()=="fastpitch") {
seq_edit_view_select.value = "pitch"
seq_edit_edit_select.value = "pitch"
seq_edit_view_select.disabled = true
seq_edit_edit_select.disabled = true
} else {
seq_edit_view_select.value = "pitch_energy"
seq_edit_view_select.disabled = false
seq_edit_edit_select.disabled = false
}
window.populateLanguagesDropdownsFromModel(base_lang_select, window.currentModel)
base_lang_select.value = window.currentModel.lang
if (window.userSettings.defaultToHiFi && window.currentModel.hifi) {
vocoder_select.value = Array.from(vocoder_select.children).find(opt => opt.innerHTML=="Bespoke HiFi GAN").value
window.changeVocoder(vocoder_select.value).then(() => dialogueInput.focus())
} else if (window.userSettings.vocoder.includes(".hg.pt")) {
window.changeVocoder("qnd").then(() => dialogueInput.focus())
} else {
closeModal(null, [workbenchContainer]).then(() => dialogueInput.focus())
}
resolve()
}).catch(e => {
console.log(e)
if (e.code =="ENOENT") {
closeModal(null, [modalContainer, workbenchContainer]).then(() => {
window.errorModal(window.i18n.ERR_SERVER)
resolve()
})
}
})
})
}
// Return true/false for if the prompt is the same - BUT: allow phoneme swaps
window.checkIfPromptIsTheSame = (sequence) => {
// False if there was no previous prompt
if (!window.sequenceEditor.historyState.length) {
return false
}
const lastPrompt = window.sequenceEditor.historyState.at(-1)
// False if they're different lengths
if (sequence.length != lastPrompt.length) {
return false
}
// Split into words (and phonemes)
const currentParts = sequence.split(" ")
const lastParts = lastPrompt.split(" ")
for (let si=0; si<currentParts.length; si++) {
// False if a word is different, but not if it's an ARPAbet symbol
const cleaned = currentParts[si].replace(/[^a-zA-Z]/g, "")
if (currentParts[si]!=lastParts[si] && !window.ARPAbetSymbols.includes(cleaned)) {
return false
}
}
return true
}
window.synthesizeSample = () => {
const game = window.currentGame.gameId
if (window.isGenerating) {
return
}
if (!window.speech2speechState.s2s_running) {
clearOldTempFiles()
}
let sequence = dialogueInput.value.replace("…", "...").replace("’", "'")
if (window.userSettings.spacePadding && !window.sequenceEditor.isEditingFromFile) { // Pad start and end of the input sequence with spaces
sequence = " "+sequence.trim()+" "
}
window.sequenceEditor.isEditingFromFile = false
if (sequence.length==0) {
return
}
window.isGenerating = true
window.pluginsManager.runPlugins(window.pluginsManager.pluginsModules["generate-voice"]["pre"], event="pre generate-voice")
if (window.wavesurfer) {
try {
window.wavesurfer.stop()
} catch (e) {
console.log(e)
}
wavesurferContainer.style.opacity = 0
}
toggleSpinnerButtons()
const voiceType = title.dataset.modelId
const outputFileName = dialogueInput.value.slice(0, 260).replace(/\n/g, " ").replace(/[\/\\:\*?<>"|]*/g, "").replace(/^[\.\s]+/, "")
try {fs.unlinkSync(localStorage.getItem("tempFileLocation"))} catch (e) {/*Do nothing*/}
// For some reason, the samplePlay audio element does not update the source when the file name is the same
const tempFileNum = `${Math.random().toString().split(".")[1]}`
let tempFileLocation = `${path}/output/temp-${tempFileNum}.wav`
let pitch = []
let duration = []
let energy = []
let emAngry = []
let emHappy = []
let emSad = []
let emSurprise = []
let editorStyles = {}
let isFreshRegen = true
let old_sequence = undefined
if (editorContainer.innerHTML && editorContainer.innerHTML.length && generateVoiceButton.dataset.modelIDLoaded==window.sequenceEditor.currentVoice) {
if (window.sequenceEditor.audioInput || window.sequenceEditor.sequence && sequence!=window.sequenceEditor.inputSequence) {
old_sequence = window.sequenceEditor.inputSequence
}
}
// Don't use the old_sequence if running speech-to-speech
if (window.speech2speechState.s2s_running) {
old_sequence = undefined
window.speech2speechState.s2s_running = false
}
// Check if editing an existing line (otherwise it's a fresh new line)
const languageHasChanged = window.sequenceEditor.base_lang && window.sequenceEditor.base_lang != base_lang_select.value
const promptHasChanged = !window.checkIfPromptIsTheSame(sequence)
if (!promptHasChanged && !languageHasChanged && !window.arpabetMenuState.hasChangedARPAbet && !window.styleEmbsMenuState.hasChangedEmb &&
(speech2speechState.s2s_autogenerate || (editorContainer.innerHTML && editorContainer.innerHTML.length && (window.userSettings.keepEditorOnVoiceChange || generateVoiceButton.dataset.modelIDLoaded==window.sequenceEditor.currentVoice)))) {
speech2speechState.s2s_autogenerate = false
pitch = window.sequenceEditor.pitchNew.map(v=> v==undefined?0:v)
duration = window.sequenceEditor.dursNew.map(v => v*pace_slid.value).map(v=> v==undefined?0:v)
energy = window.sequenceEditor.energyNew ? window.sequenceEditor.energyNew.map(v => v==undefined?0:v).filter(v => !isNaN(v)) : []
if (window.currentModel.modelType=="xVAPitch") {
emAngry = window.sequenceEditor.emAngryNew ? window.sequenceEditor.emAngryNew.map(v => v==undefined?0:v).filter(v => !isNaN(v)) : []
emHappy = window.sequenceEditor.emHappyNew ? window.sequenceEditor.emHappyNew.map(v => v==undefined?0:v).filter(v => !isNaN(v)) : []
emSad = window.sequenceEditor.emSadNew ? window.sequenceEditor.emSadNew.map(v => v==undefined?0:v).filter(v => !isNaN(v)) : []
emSurprise = window.sequenceEditor.emSurpriseNew ? window.sequenceEditor.emSurpriseNew.map(v => v==undefined?0:v).filter(v => !isNaN(v)) : []
if (window.sequenceEditor.registeredStyleKeys) {
window.sequenceEditor.registeredStyleKeys.forEach(styleKey => {
editorStyles[styleKey] = {
embedding: window.appState.currentModelEmbeddings[styleKey][1],
sliders: window.sequenceEditor.styleValuesNew[styleKey].map(v => v==undefined?0:v).filter(v => !isNaN(v))// : []
}
})
}
}
isFreshRegen = false
}
window.arpabetMenuState.hasChangedARPAbet = false
window.styleEmbsMenuState.hasChangedEmb = false
window.sequenceEditor.currentVoice = generateVoiceButton.dataset.modelIDLoaded
const speaker_i = window.currentModel.games[0].emb_i
const pace = (window.userSettings.keepPaceOnNew && isFreshRegen)?pace_slid.value:1
window.appLogger.log(`${window.i18n.SYNTHESIZING}: ${sequence}`)
doFetch(`http://localhost:8008/synthesize`, {
method: "Post",
body: JSON.stringify({
sequence, pitch, duration, energy, emAngry, emHappy, emSad, emSurprise, editorStyles, speaker_i, pace,
base_lang: base_lang_select.value,
base_emb: style_emb_select.value||"",
modelType: window.currentModel.modelType,
old_sequence, // For partial re-generation
device: window.userSettings.installation=="cpu"?"cpu":(window.userSettings.useGPU?"cuda:0":"cpu"),
// device: window.userSettings.useGPU?"gpu":"cpu", // Switch to this once DirectML is installed
useSR: useSRCkbx.checked,
useCleanup: useCleanupCkbx.checked,
outfile: tempFileLocation,
pluginsContext: JSON.stringify(window.pluginsContext),
vocoder: window.currentModel.modelType=="xVAPitch" ? "n/a" : window.userSettings.vocoder,
waveglowPath: vocoder_select.value=="256_waveglow" ? window.userSettings.waveglow_path : window.userSettings.bigwaveglow_path
})
}).then(r=>r.text()).then(res => {
window.isGenerating = false
if (res=="ENOENT" || res.startsWith("ERR:")) {
console.log(res)
if (res.startsWith("ERR:")) {
if (res.includes("ARPABET_NOT_IN_LIST")) {
const symbolNotInList = res.split(":").reverse()[0]
window.errorModal(`${window.i18n.SOMETHING_WENT_WRONG}<br><br>${window.i18n.ERR_ARPABET_NOT_EXIST.replace("_1", symbolNotInList)}`)
} else {
window.errorModal(`${window.i18n.SOMETHING_WENT_WRONG}<br><br>${res.replace("ERR:","").replaceAll(/\n/g, "<br>")}`)
}
} else {
window.appLogger.log(res)
window.errorModal(`${window.i18n.BATCH_MODEL_NOT_FOUND}.${vocoder_select.value.includes("waveglow")?" "+window.i18n.BATCH_DOWNLOAD_WAVEGLOW:""}`)
}
toggleSpinnerButtons()
return
}
dialogueInput.focus()
window.sequenceEditor.historyState.push(sequence)
if (window.userSettings.clear_text_after_synth) {
dialogueInput.value = ""
}
res = res.split("\n")
let pitchData = res[0]
let durationsData = res[1]
let energyData = res[2]
let em_angryData = res[3]
let em_happyData = res[4]
let em_sadData = res[5]
let em_surpriseData = res[6]
const editorStyles = res[7]&&res[7].length ? JSON.parse(res[7]) : undefined
let cleanedSequence = res[8].split("|").map(c=>c.replaceAll("{", "").replaceAll("}", "").replace(/\s/g, "_"))
const start_index = res[9]
const end_index = res[10]
pitchData = pitchData.split(",").map(v => parseFloat(v))
// For use in adjusting editor range
const maxPitchVal = pitchData.reduce((p,c)=>Math.max(p, Math.abs(c)), 0)
if (maxPitchVal>window.sequenceEditor.default_pitchSliderRange) {
window.sequenceEditor.pitchSliderRange = maxPitchVal
} else {
window.sequenceEditor.pitchSliderRange = window.sequenceEditor.default_pitchSliderRange
}
em_angryData = em_angryData.length ? em_angryData.split(",").map(v => parseFloat(v)).filter(v => !isNaN(v)) : []
em_happyData = em_happyData.length ? em_happyData.split(",").map(v => parseFloat(v)).filter(v => !isNaN(v)) : []
em_sadData = em_sadData.length ? em_sadData.split(",").map(v => parseFloat(v)).filter(v => !isNaN(v)) : []
em_surpriseData = em_surpriseData.length ? em_surpriseData.split(",").map(v => parseFloat(v)).filter(v => !isNaN(v)) : []
if (energyData.length) {
energyData = energyData.split(",").map(v => parseFloat(v)).filter(v => !isNaN(v))
// For use in adjusting editor range
const maxEnergyVal = energyData.reduce((p,c)=>Math.max(p, c), 0)
const minEnergyVal = energyData.reduce((p,c)=>Math.min(p, c), 100)
if (minEnergyVal<window.sequenceEditor.default_MIN_ENERGY) {
window.sequenceEditor.MIN_ENERGY = minEnergyVal
} else {
window.sequenceEditor.MIN_ENERGY = window.sequenceEditor.default_MIN_ENERGY
}
if (maxEnergyVal>window.sequenceEditor.default_MAX_ENERGY) {
window.sequenceEditor.MAX_ENERGY = maxEnergyVal
} else {
window.sequenceEditor.MAX_ENERGY = window.sequenceEditor.default_MAX_ENERGY
}
} else {
energyData = []
}
durationsData = durationsData.split(",").map(v => isFreshRegen ? parseFloat(v) : parseFloat(v)/pace_slid.value)
const doTheRest = () => {
window.sequenceEditor.base_lang = base_lang_select.value
window.sequenceEditor.inputSequence = sequence
window.sequenceEditor.sequence = cleanedSequence
if (pitch.length==0 || isFreshRegen) {
window.sequenceEditor.resetPitch = pitchData
window.sequenceEditor.resetDurs = durationsData
window.sequenceEditor.resetEnergy = energyData
window.sequenceEditor.resetEmAngry = em_angryData
window.sequenceEditor.resetEmHappy = em_happyData
window.sequenceEditor.resetEmSad = em_sadData
window.sequenceEditor.resetEmSurprise = em_surpriseData
}
window.sequenceEditor.letters = cleanedSequence
window.sequenceEditor.pitchNew = pitchData.map(p=>p)
window.sequenceEditor.dursNew = durationsData.map(v=>v)
window.sequenceEditor.energyNew = energyData.map(v=>v)
if (window.currentModel.modelType=="xVAPitch") {
window.sequenceEditor.emAngryNew = em_angryData.map(v=>v)
window.sequenceEditor.emHappyNew = em_happyData.map(v=>v)
window.sequenceEditor.emSadNew = em_sadData.map(v=>v)
window.sequenceEditor.emSurpriseNew = em_surpriseData.map(v=>v)
window.sequenceEditor.loadStylesData(editorStyles)
}
window.sequenceEditor.init()
const pitchRange = window.userSettings.pitchrangeoverride ? window.userSettings.pitchrangeoverride : window.sequenceEditor.pitchSliderRange
window.sequenceEditor.update(window.currentModel.modelType, pitchRange)
window.sequenceEditor.sliderBoxes.forEach((box, i) => {box.setValueFromValue(window.sequenceEditor.dursNew[i])})
window.sequenceEditor.autoInferTimer = null
window.sequenceEditor.hasChanged = false
toggleSpinnerButtons()
if (keepSampleButton.dataset.newFileLocation && keepSampleButton.dataset.newFileLocation.startsWith("BATCH_EDIT")) {
console.log("_debug_")
} else {
if (window.userSettings[`outpath_${game}`]) {
keepSampleButton.dataset.newFileLocation = `${window.userSettings[`outpath_${game}`]}/${voiceType}/${outputFileName}.wav`
} else {
keepSampleButton.dataset.newFileLocation = `${__dirname.replace(/\\/g,"/")}/output/${voiceType}/${outputFileName}.wav`
}
}
keepSampleButton.disabled = false
window.tempFileLocation = tempFileLocation
// Wavesurfer
if (!window.wavesurfer) {
window.initWaveSurfer(`${__dirname.replace("/javascript", "")}/output/${tempFileLocation.split("/").reverse()[0]}`)
} else {
window.wavesurfer.load(`${__dirname.replace("/javascript", "")}/output/${tempFileLocation.split("/").reverse()[0]}`)
}
window.wavesurfer.on("ready", () => {
wavesurferContainer.style.opacity = 1
if (window.userSettings.autoPlayGen) {
if (window.userSettings.playChangedAudio) {
const playbackStartEnd = window.sequenceEditor.getChangedTimeStamps(start_index, end_index, window.wavesurfer.getDuration())
if (playbackStartEnd) {
wavesurfer.play(playbackStartEnd[0], playbackStartEnd[1])
} else {
wavesurfer.play()
}
} else {
wavesurfer.play()
}
window.sequenceEditor.adjustedLetters = new Set()
samplePlayPause.innerHTML = window.i18n.PAUSE
}
})
// Persistance across sessions
localStorage.setItem("tempFileLocation", tempFileLocation)
}
if (window.userSettings.audio.ffmpeg) {
const options = {
hz: window.userSettings.audio.hz,
padStart: window.userSettings.audio.padStart,
padEnd: window.userSettings.audio.padEnd,
bit_depth: window.userSettings.audio.bitdepth,
amplitude: window.userSettings.audio.amplitude,
pitchMult: window.userSettings.audio.pitchMult,
tempo: window.userSettings.audio.tempo,
deessing: window.userSettings.audio.deessing,
nr: window.userSettings.audio.nr,
nf: window.userSettings.audio.nf,
useNR: window.userSettings.audio.useNR,
useSR: useSRCkbx.checked,
useCleanup: useCleanupCkbx.checked,
}
const extraInfo = {
game: window.currentGame.gameId,
voiceId: window.currentModel.voiceId,
voiceName: window.currentModel.voiceName,
inputSequence: sequence,
letters: cleanedSequence,
pitch: pitchData.map(p=>p),
energy: energyData.map(p=>p),
em_angry: em_angryData.map(p=>p),
em_happy: em_happyData.map(p=>p),
em_sad: em_sadData.map(p=>p),
em_surprise: em_surpriseData.map(p=>p),
durations: durationsData.map(v=>v)
}
doFetch(`http://localhost:8008/outputAudio`, {
method: "Post",
body: JSON.stringify({
input_path: tempFileLocation,
output_path: tempFileLocation.replace(".wav", `_ffmpeg.${window.userSettings.audio.format}`),
pluginsContext: JSON.stringify(window.pluginsContext),
extraInfo: JSON.stringify(extraInfo),
isBatchMode: false,
options: JSON.stringify(options)
})
}).then(r=>r.text()).then(res => {
if (res.length && res!="-") {
console.log("res", res)
window.errorModal(`${window.i18n.SOMETHING_WENT_WRONG}<br><br>${res}`).then(() => toggleSpinnerButtons())
} else {
tempFileLocation = tempFileLocation.replace(".wav", `_ffmpeg.${window.userSettings.audio.format}`)
doTheRest()
}
}).catch(res => {
console.log(res)
window.appLogger.log(`outputAudio error: ${res}`)
// closeModal().then(() => {
window.errorModal(`${window.i18n.SOMETHING_WENT_WRONG}<br><br>${res}`)
// })
})
} else {
doTheRest()
}
}).catch(res => {
window.isGenerating = false
console.log(res)
window.appLogger.log(res)
window.errorModal(window.i18n.SOMETHING_WENT_WRONG)
toggleSpinnerButtons()
})
}
generateVoiceButton.addEventListener("click", () => {
try {fs.mkdirSync(window.userSettings[`outpath_${game}`])} catch (e) {/*Do nothing*/}
try {fs.mkdirSync(`${window.userSettings[`outpath_${game}`]}/${voiceId}`)} catch (e) {/*Do nothing*/}
if (generateVoiceButton.dataset.modelQuery && generateVoiceButton.dataset.modelQuery!="null") {
window.loadModel()
} else {
window.synthesizeSample()
}
})
window.saveFile = (from, to, skipUIRecord=false) => {
to = to.split("%20").join(" ")
to = to.replace(".wav", `.${window.userSettings.audio.format}`)
// Make the containing folder if it does not already exist
let containerFolderPath = to.split("/")
containerFolderPath = containerFolderPath.slice(0,containerFolderPath.length-1).join("/")
try {fs.mkdirSync(containerFolderPath)} catch (e) {/*Do nothing*/}
// For plugins
const pluginData = {
game: window.currentGame.gameId,
voiceId: window.currentModel.voiceId,
voiceName: window.currentModel.voiceName,
inputSequence: window.sequenceEditor.inputSequence,
letters: window.sequenceEditor.letters,
pitch: window.sequenceEditor.pitchNew,
durations: window.sequenceEditor.dursNew,
vocoder: vocoder_select.value,
from, to
}
const options = {
hz: window.userSettings.audio.hz,
padStart: window.userSettings.audio.padStart,
padEnd: window.userSettings.audio.padEnd,
bit_depth: window.userSettings.audio.bitdepth,
amplitude: window.userSettings.audio.amplitude,
pitchMult: window.userSettings.audio.pitchMult,
tempo: window.userSettings.audio.tempo,
deessing: window.userSettings.audio.deessing,
nr: window.userSettings.audio.nr,
nf: window.userSettings.audio.nf,
useNR: window.userSettings.audio.useNR,
useSR: useSRCkbx.checked
}
pluginData.audioOptions = options
window.pluginsManager.runPlugins(window.pluginsManager.pluginsModules["keep-sample"]["pre"], event="pre keep-sample", pluginData)
const jsonDataOut = {
modelType: window.currentModel.modelType,
modelVersion: window.currentModel.modelVersion,
version: window.currentModel.version,
inputSequence: dialogueInput.value.trim(),
pacing: parseFloat(pace_slid.value),
letters: window.sequenceEditor.letters,
currentVoice: window.sequenceEditor.currentVoice,
resetEnergy: window.sequenceEditor.resetEnergy,
resetPitch: window.sequenceEditor.resetPitch,
resetDurs: window.sequenceEditor.resetDurs,
resetEmAngry: window.sequenceEditor.resetEmAngry,
resetEmHappy: window.sequenceEditor.resetEmHappy,
resetEmSad: window.sequenceEditor.resetEmSad,
resetEmSurprise: window.sequenceEditor.resetEmSurprise,
styleValuesReset: window.sequenceEditor.styleValuesReset,
ampFlatCounter: window.sequenceEditor.ampFlatCounter,
inputSequence: window.sequenceEditor.inputSequence,
sequence: window.sequenceEditor.sequence,
pitchNew: window.sequenceEditor.pitchNew,
energyNew: window.sequenceEditor.energyNew,
dursNew: window.sequenceEditor.dursNew,
emAngryNew: window.sequenceEditor.emAngryNew,
emHappyNew: window.sequenceEditor.emHappyNew,
emSadNew: window.sequenceEditor.emSadNew,
emSurpriseNew: window.sequenceEditor.emSurpriseNew,
styleValuesNew: window.sequenceEditor.styleValuesNew,
}
let outputFileName = to.split("/").reverse()[0].split(".").reverse().slice(1, 1000)
const toExt = to.split(".").reverse()[0]
if (window.userSettings.filenameNumericalSeq) {
outputFileName = outputFileName[0]+"."+outputFileName.slice(1,1000).reverse().join(".")
} else {
outputFileName = outputFileName.reverse().join(".")
}
to = `${to.split("/").reverse().slice(1,10000).reverse().join("/")}/${outputFileName}`
const allFiles = fs.readdirSync(`${path}/output`).filter(fname => fname.includes(from.split("/").reverse()[0].split(".")[0]))
const toFolder = to.split("/").reverse().slice(1, 1000).reverse().join("/")
allFiles.forEach(fname => {
const ext = fname.split(".").reverse()[0]
fs.copyFile(`${path}/output/${fname}`, `${toFolder}/${outputFileName}.${ext}`, err => {
if (err) {
console.log(err)
window.appLogger.log(`Error in saveFile->outputAudio[no ffmpeg]: ${err}`)
if (!fs.existsSync(from)) {
window.appLogger.log(`${window.i18n.TEMP_FILE_NOT_EXIST}: ${from}`)
}
if (!fs.existsSync(toFolder)) {
window.appLogger.log(`${window.i18n.OUT_DIR_NOT_EXIST}: ${toFolder}`)
}
} else {
if (window.userSettings.outputJSON && window.sequenceEditor.letters.length) {
fs.writeFileSync(`${to}.${toExt}.json`, JSON.stringify(jsonDataOut, null, 4))
}
if (!skipUIRecord) {
window.initMainPagePagination(`${window.userSettings[`outpath_${window.currentGame.gameId}`]}/${window.currentModel.voiceId}`)
window.refreshRecordsList()
}
window.pluginsManager.runPlugins(window.pluginsManager.pluginsModules["keep-sample"]["post"], event="post keep-sample", pluginData)
}
})
})
}
window.keepSampleFunction = shiftClick => {
if (keepSampleButton.dataset.newFileLocation) {
const skipUIRecord = keepSampleButton.dataset.newFileLocation.includes("BATCH_EDIT")
let fromLocation = window.tempFileLocation
let toLocation = keepSampleButton.dataset.newFileLocation.replace("BATCH_EDIT", "")
if (!skipUIRecord) {
toLocation = toLocation.split("/")
toLocation[toLocation.length-1] = toLocation[toLocation.length-1].replace(/[\/\\:\*?<>"|]*/g, "")
toLocation[toLocation.length-1] = toLocation[toLocation.length-1].replace(/\.wav$/, "").slice(0, window.userSettings.max_filename_chars).replace(/\.$/, "")
}
// Numerical file name counter
if (!skipUIRecord && window.userSettings.filenameNumericalSeq) {
let existingFiles = []
try {
existingFiles = fs.readdirSync(toLocation.slice(0, toLocation.length-1).join("/")).filter(fname => !fname.endsWith(".json"))
} catch (e) {
console.log(e)
}
existingFiles = existingFiles.filter(fname => fname.includes(toLocation[toLocation.length-1]))
existingFiles = existingFiles.map(fname => {
const parts = fname.split(".")
parts.reverse()
if (parts.length>2 && parts.reverse()[0].length) {
if (parseInt(parts[0]) != NaN) {
return parseInt(parts[0])
}
}
return null
})
existingFiles = existingFiles.filter(val => !!val)
if (existingFiles.length==0) {
existingFiles.push(0)
}
if (existingFiles.length) {
existingFiles = existingFiles.sort((a,b) => {a<b?-1:1})
toLocation[toLocation.length-1] = `${toLocation[toLocation.length-1]}.${String(existingFiles[existingFiles.length-1]+1).padStart(4, "0")}`
}
}
if (!skipUIRecord) {
toLocation[toLocation.length-1] += ".wav"
toLocation = toLocation.join("/")
}
const outFolder = toLocation.split("/").reverse().slice(2, 100).reverse().join("/")
if (!fs.existsSync(outFolder)) {
return void window.errorModal(`${window.i18n.OUT_DIR_NOT_EXIST}:<br><br><i>${outFolder}</i><br><br>${window.i18n.YOU_CAN_CHANGE_IN_SETTINGS}`)
}
// File name conflict
const alreadyExists = fs.existsSync(toLocation)
if (alreadyExists || shiftClick) {
const promptText = alreadyExists ? window.i18n.FILE_EXISTS_ADJUST : window.i18n.ENTER_FILE_NAME
createModal("prompt", {
prompt: promptText,
value: toLocation.split("/").reverse()[0].replace(".wav", `.${window.userSettings.audio.format}`)
}).then(newFileName => {
let toLocationOut = toLocation.split("/").reverse()
toLocationOut[0] = newFileName.replace(`.${window.userSettings.audio.format}`, "") + `.${window.userSettings.audio.format}`
let outDir = toLocationOut
outDir.shift()
newFileName = (newFileName.replace(`.${window.userSettings.audio.format}`, "") + `.${window.userSettings.audio.format}`).replace(/[\/\\:\*?<>"|]*/g, "")
toLocationOut.reverse()
toLocationOut.push(newFileName)
if (fs.existsSync(outDir.slice(0, outDir.length-1).join("/"))) {
const existingFiles = fs.readdirSync(outDir.slice(0, outDir.length-1).join("/"))
const existingFileConflict = existingFiles.filter(name => name==newFileName)
const finalOutLocation = toLocationOut.join("/")
if (existingFileConflict.length) {
// Remove the entry from the output files' preview
Array.from(voiceSamples.querySelectorAll("div.sample")).forEach(sampleElem => {
let sourceSrc = sampleElem.children[0].children[0].innerText
sourceSrc = sourceSrc.split("/").reverse()
const finalFileName = finalOutLocation.split("/").reverse()
if (sourceSrc[0] == finalFileName[0]) {
sampleElem.parentNode.removeChild(sampleElem)
}
})
// Remove the old file and write the new one in
fs.unlink(finalOutLocation, err => {
if (err) {
console.log(err)
window.appLogger.log(`Error in keepSample: ${err}`)
}
window.saveFile(fromLocation, finalOutLocation, skipUIRecord)
})
return
} else {
window.saveFile(fromLocation, toLocationOut.join("/"), skipUIRecord)
return
}
}
window.saveFile(fromLocation, toLocationOut.join("/"), skipUIRecord)
})
} else {
window.saveFile(fromLocation, toLocation, skipUIRecord)
}
}
}
keepSampleButton.addEventListener("click", event => keepSampleFunction(event.shiftKey))
// Weird recursive intermittent promises to repeatedly check if the server is up yet - but it works!
window.serverIsUp = false
const serverStartingMessage = `${window.i18n.LOADING}...<br>${window.i18n.MAY_TAKE_A_MINUTE}<br><br>${window.i18n.STARTING_PYTHON}...`
window.doWeirdServerStartupCheck = () => {
const check = () => {
return new Promise(topResolve => {
if (window.serverIsUp) {
topResolve()
} else {
(new Promise((resolve, reject) => {
// Gather the model paths to send to the server
const modelsPaths = {}
Object.keys(window.userSettings).filter(key => key.includes("modelspath_")).forEach(key => {
modelsPaths[key.split("_")[1]] = window.userSettings[key]
})
doFetch(`http://localhost:8008/checkReady`, {
method: "Post",
body: JSON.stringify({
device: (window.userSettings.useGPU&&window.userSettings.installation=="gpu")?"gpu":"cpu",
modelsPaths: JSON.stringify(modelsPaths)
})
}).then(r => r.text()).then(r => {
closeModal([document.querySelector("#activeModal"), modalContainer], [totdContainer, EULAContainer], true).then(() => {
window.pluginsManager.updateUI()
if (!window.pluginsManager.hasRunPostStartPlugins) {
window.pluginsManager.hasRunPostStartPlugins = true
window.pluginsManager.runPlugins(window.pluginsManager.pluginsModules["start"]["post"], event="post start")
window.electronBrowserWindow.setProgressBar(-1)
window.checkForWorkshopInstallations()
}
})
window.serverIsUp = true
if (window.userSettings.installation=="cpu") {
if (useGPUCbx.checked) {
doFetch(`http://localhost:8008/setDevice`, {
method: "Post",
body: JSON.stringify({device: "cpu"})
})
}
useGPUCbx.checked = false
useGPUCbx.disabled = true
window.userSettings.useGPU = false
saveUserSettings()
}
resolve()
}).catch((err) => {
reject()
})
})).catch(() => {
setTimeout(async () => {
await check()
topResolve()
}, 100)
})
}
})
}
check()
}
window.doWeirdServerStartupCheck()
modalContainer.addEventListener("click", event => {
try {
if (event.target==modalContainer && activeModal.dataset.type!="spinner") {
closeModal()
}
} catch (e) {}
})
// Cached UI stuff
// =========
dialogueInput.addEventListener("keyup", (event) => {
localStorage.setItem("dialogueInput", " "+dialogueInput.value.trim()+" ")
window.sequenceEditor.hasChanged = true
})
const dialogueInputCache = localStorage.getItem("dialogueInput")
if (dialogueInputCache) {
dialogueInput.value = dialogueInputCache
}
// =========
vocoder_select.value = window.userSettings.vocoder.includes(".hg.") ? "qnd" : window.userSettings.vocoder
window.changeVocoder = vocoder => {
return new Promise(resolve => {
spinnerModal(window.i18n.CHANGING_MODELS)
doFetch(`http://localhost:8008/setVocoder`, {
method: "Post",
body: JSON.stringify({
vocoder,
modelPath: vocoder=="256_waveglow" ? window.userSettings.waveglow_path : window.userSettings.bigwaveglow_path
})
}).then(r=>r.text()).then((res) => {
closeModal().then(() => {
setTimeout(() => {
if (res=="ENOENT") {
vocoder_select.value = window.userSettings.vocoder
window.errorModal(`${window.i18n.BATCH_MODEL_NOT_FOUND}.${vocoder.includes("waveglow")?" "+window.i18n.BATCH_DOWNLOAD_WAVEGLOW:""}`)
resolve()
} else {
window.batch_state.lastVocoder = vocoder
window.userSettings.vocoder = vocoder
saveUserSettings()
resolve()
}
}, 300)
})
})
})
}
vocoder_select.addEventListener("change", () => window.changeVocoder(vocoder_select.value))
useSRCkbx.addEventListener("click", () => {
let userHasSeenThisAlready = localStorage.getItem("useSRHintSeen")
if (useSRCkbx.checked && !userHasSeenThisAlready) {
window.confirmModal(window.i18n.USE_SR_HINT).then(resp => {
if (resp) {
localStorage.setItem("useSRHintSeen", "true")
}
})
}
})
dialogueInput.addEventListener("contextmenu", event => {
event.preventDefault()
ipcRenderer.send('show-context-menu')
})
ipcRenderer.on('context-menu-command', (e, command) => {
if (command=="context-copy") {
if (dialogueInput.selectionStart != dialogueInput.selectionEnd) {
clipboard.writeText(dialogueInput.value.slice(dialogueInput.selectionStart, dialogueInput.selectionEnd))
}
} else if (command=="context-paste") {
if (clipboard.readText().length) {
let newString = dialogueInput.value.slice(0, dialogueInput.selectionStart) + clipboard.readText() + dialogueInput.value.slice(dialogueInput.selectionEnd, dialogueInput.value.length)
dialogueInput.value = newString
}
}
})
window.setupModal(workbenchIcon, workbenchContainer, () => window.initVoiceWorkbench())
// Info
// ====
window.setupModal(infoIcon, infoContainer)
// Patreon
// =======
// window.setupModal(patreonIcon, patreonContainer, () => {
// const data = fs.readFileSync(`${path}/patreon.txt`, "utf8") + ", minermanb"
// creditsList.innerHTML = data
// })
// Updates
// =======
app_version.innerHTML = window.appVersion
updatesVersions.innerHTML = `${window.i18n.THIS_APP_VERSION}: ${window.appVersion}`
const checkForUpdates = () => {
doFetch("http://danruta.co.uk/xvasynth_updates.txt").then(r=>r.json()).then(data => {
fs.writeFileSync(`${path}/updates.json`, JSON.stringify(data), "utf8")
checkUpdates.innerHTML = window.i18n.CHECK_FOR_UPDATES
window.showUpdates()
}).catch(() => {
checkUpdates.innerHTML = window.i18n.CANT_REACH_SERVER
})
}
window.showUpdates = () => {
window.updatesLog = fs.readFileSync(`${path}/updates.json`, "utf8")
window.updatesLog = JSON.parse(window.updatesLog)
const sortedLogVersions = Object.keys(window.updatesLog).map( a => a.split('.').map( n => +n+100000 ).join('.') ).sort()
.map( a => a.split('.').map( n => +n-100000 ).join('.') )
const appVersion = window.appVersion.replace("v", "")
const appIsUpToDate = sortedLogVersions.indexOf(appVersion)==(sortedLogVersions.length-1) || sortedLogVersions.indexOf(appVersion)==-1
if (!appIsUpToDate) {
update_nothing.style.display = "none"
update_something.style.display = "block"
updatesVersions.innerHTML = `${window.i18n.THIS_APP_VERSION}: ${appVersion}. ${window.i18n.AVAILABLE}: ${sortedLogVersions[sortedLogVersions.length-1]}`
} else {
updatesVersions.innerHTML = `${window.i18n.THIS_APP_VERSION}: ${appVersion}. ${window.i18n.UPTODATE}`
}
updatesLogList.innerHTML = ""
sortedLogVersions.reverse().forEach(version => {
const versionLabel = createElem("h2", version)
const logItem = createElem("div", versionLabel)
window.updatesLog[version].split("\n").forEach(line => {
logItem.appendChild(createElem("div", line))
})
updatesLogList.appendChild(logItem)
})
}
checkForUpdates()
window.setupModal(updatesIcon, updatesContainer)
checkUpdates.addEventListener("click", () => {
checkUpdates.innerHTML = window.i18n.CHECKING_FOR_UPDATES
checkForUpdates()
})
window.showUpdates()
// Batch generation
// ========
window.setupModal(batchIcon, batchGenerationContainer)
// Settings
// ========
window.setupModal(settingsCog, settingsContainer)
// Change Game
// ===========
window.setupModal(changeGameButton, gameSelectionContainer)
changeGameButton.addEventListener("click", () => searchGameInput.focus())
window.gameAssets = {}
window.updateGameList = (doLoadAllModels=true) => {
gameSelectionListContainer.innerHTML = ""
const fileNames = fs.readdirSync(`${window.path}/assets`)
let totalVoices = 0
let totalGames = new Set()
const itemsToSort = []
// const gameIDs = doLoadAllModels ? fileNames.filter(fn=>fn.endsWith(".json")) : Object.keys(window.games).map(gID => gID+".json")
const gameIDs = fileNames.filter(fn=>fn.endsWith(".json"))
gameIDs.forEach(gameId => {
const metadata = fs.existsSync(`${window.path}/assets/${gameId}`) ? JSON.parse(fs.readFileSync(`${window.path}/assets/${gameId}`)) : window.games[gameId.replace(".json", "")].dummyGameTheme
gameId = gameId.replace(".json", "")
metadata.gameId = gameId
const assetFile = metadata.assetFile
const gameSelection = createElem("div.gameSelection")
gameSelection.style.background = `url("assets/${assetFile}")`
const gameName = metadata.gameName
const gameSelectionContent = createElem("div.gameSelectionContent")
let numVoices = 0
const modelsPath = window.userSettings[`modelspath_${gameId}`]
if (fs.existsSync(modelsPath)) {
const files = fs.readdirSync(modelsPath)
numVoices = files.filter(fn => fn.includes(".json")).length
totalVoices += numVoices
}
if (numVoices==0) {
gameSelectionContent.style.background = "rgba(150,150,150,0.7)"
} else {
gameSelectionContent.classList.add("gameSelectionContentToHover")
totalGames.add(gameId)
}
gameSelectionContent.appendChild(createElem("div", `${numVoices} ${(numVoices>1||numVoices==0)?window.i18n.VOICE_PLURAL:window.i18n.VOICE}`))
gameSelectionContent.appendChild(createElem("div", gameName))
gameSelection.appendChild(gameSelectionContent)
window.gameAssets[gameId] = metadata
gameSelectionContent.addEventListener("click", () => {
voiceSearchInput.focus()
searchGameInput.value = ""
changeGame(metadata)
closeModal(gameSelectionContainer)
Array.from(gameSelectionListContainer.children).forEach(elem => elem.style.display="flex")
})
itemsToSort.push([numVoices, gameSelection])
const modelsDir = window.userSettings[`modelspath_${gameId}`]
if (!window.watchedModelsDirs.includes(modelsDir)) {
window.watchedModelsDirs.push(modelsDir)
try {
fs.watch(modelsDir, {recursive: false, persistent: true}, (eventType, filename) => {
if (window.userSettings.autoReloadVoices) {
if (doLoadAllModels) {
loadAllModels().then(() => changeGame(metadata))
}
}
})
} catch (e) {
// console.log(e)
}
}
})
itemsToSort.sort((a,b) => a[0]<b[0]?1:-1).forEach(([numVoices, elem]) => {
gameSelectionListContainer.appendChild(elem)
})
searchGameInput.addEventListener("keyup", (event) => {
if (event.key=="Enter") {
const voiceElems = Array.from(gameSelectionListContainer.children).filter(elem => elem.style.display=="flex")
if (voiceElems.length==1) {
voiceElems[0].children[0].click()
searchGameInput.value = ""
}
}
const voiceElems = Array.from(gameSelectionListContainer.children)
if (searchGameInput.value.length) {
voiceElems.forEach(elem => {
if (elem.children[0].children[1].innerHTML.toLowerCase().includes(searchGameInput.value)) {
elem.style.display="flex"
} else {
elem.style.display="none"
}
})
} else {
voiceElems.forEach(elem => elem.style.display="block")
}
})
searchGameInput.placeholder = window.i18n.SEARCH_N_GAMES_WITH_N2_VOICES.replace("_1", Array.from(totalGames).length).replace("_2", totalVoices)
if (doLoadAllModels) {
loadAllModels().then(() => {
// Load the last selected game
const lastGame = localStorage.getItem("lastGame")
if (lastGame) {
changeGame(JSON.parse(lastGame))
}
})
}
}
window.updateGameList()
// Embeddings
// ==========
window.setupModal(embeddingsIcon, embeddingsContainer, () => {
setTimeout(() => {
if (window.embeddingsState.isReady) {
window.embeddings_updateSize()
}
}, 100)
window.embeddings_updateSize()
window.embeddingsState.isOpen = true
if (!window.embeddingsState.ready) {
setTimeout(() => {
window.embeddingsState.ready = true
window.initEmbeddingsScene()
setTimeout(() => {
window.computeEmbsAndDimReduction(true)
}, 300)
}, 100)
}
}, () => {
window.embeddingsState.isOpen = false
})
// Arpabet
// =======
window.setupModal(arpabetIcon, arpabetContainer, () => setTimeout(()=> !window.arpabetMenuState.hasInitialised && window.refreshDictionariesList(), 100))
// Plugins
// =======
window.setupModal(pluginsIcon, pluginsContainer)
window.setupModal(style_emb_manage_btn, styleEmbeddingsContainer, window.styleEmbsModalOpenCallback)
// Other
// =====
window.setupModal(reset_what_open_btn, resetContainer)
window.setupModal(i18n_batch_metadata_open_btn, batchMetadataCSVContainer)
voiceSearchInput.addEventListener("keyup", () => {
if (event.key=="Enter") {
const voiceElems = Array.from(voiceTypeContainer.children).filter(elem => elem.style.display=="block")
if (voiceElems.length==1) {
voiceElems[0].click()
generateVoiceButton.click()
voiceSearchInput.value = ""
}
}
const voiceElems = Array.from(voiceTypeContainer.children)
if (voiceSearchInput.value.length) {
voiceElems.forEach(elem => {
if (elem.innerHTML.toLowerCase().includes(voiceSearchInput.value)) {
elem.style.display="block"
} else {
elem.style.display="none"
}
})
} else {
voiceElems.forEach(elem => elem.style.display="block")
}
})
// Splash/EULA
splashNextButton1.addEventListener("click", () => {
splash_screen1.style.display = "none"
splash_screen2.style.display = "flex"
})
EULA_closeButon.addEventListener("click", () => {
if (EULA_accept_ckbx.checked) {
closeModal(EULAContainer)
window.userSettings.EULA_accepted_2023 = true
saveUserSettings()
if (!window.totd_state.startupChecked) {
window.showTipIfEnabledAndNewDay().then(() => {
if (!serverIsUp) {
spinnerModal(serverStartingMessage)
}
})
}
if (!serverIsUp && totdContainer.style.display=="none") {
window.spinnerModal(serverStartingMessage)
}
}
})
if (!Object.keys(window.userSettings).includes("EULA_accepted_2023") || !window.userSettings.EULA_accepted_2023) {
EULAContainer.style.opacity = 0
EULAContainer.style.display = "flex"
requestAnimationFrame(() => requestAnimationFrame(() => EULAContainer.style.opacity = 1))
requestAnimationFrame(() => requestAnimationFrame(() => chromeBar.style.opacity = 1))
} else {
window.showTipIfEnabledAndNewDay().then(() => {
// If not, or the user closed the window quickly, show the server is starting message if still booting up
if (!window.serverIsUp) {
spinnerModal(serverStartingMessage)
}
})
}
// Links
document.querySelectorAll('a[href^="http"]').forEach(a => a.addEventListener("click", e => {
event.preventDefault()
shell.openExternal(a.href)
}))