mirror of
https://github.com/Anjok07/ultimatevocalremovergui.git
synced 2024-11-27 17:00:59 +01:00
7266 lines
402 KiB
Python
7266 lines
402 KiB
Python
# GUI modules
|
|
import time
|
|
#start_time = time.time()
|
|
import audioread
|
|
import gui_data.sv_ttk
|
|
import hashlib
|
|
import json
|
|
import librosa
|
|
import math
|
|
import natsort
|
|
import os
|
|
import pickle
|
|
import psutil
|
|
from pyglet import font as pyglet_font
|
|
import pyperclip
|
|
import base64
|
|
import queue
|
|
import shutil
|
|
import subprocess
|
|
import soundfile as sf
|
|
import torch
|
|
import urllib.request
|
|
import webbrowser
|
|
import wget
|
|
import traceback
|
|
import matchering as match
|
|
import tkinter as tk
|
|
import tkinter.ttk as ttk
|
|
from tkinter.font import Font
|
|
from tkinter import filedialog
|
|
from tkinter import messagebox
|
|
from collections import Counter
|
|
from __version__ import VERSION, PATCH, PATCH_MAC, PATCH_LINUX
|
|
from cryptography.fernet import Fernet
|
|
from cryptography.hazmat.primitives import hashes
|
|
from cryptography.hazmat.primitives.kdf.pbkdf2 import PBKDF2HMAC
|
|
from datetime import datetime
|
|
from gui_data.constants import *
|
|
from gui_data.app_size_values import *
|
|
from gui_data.error_handling import error_text, error_dialouge
|
|
from gui_data.old_data_check import file_check, remove_unneeded_yamls, remove_temps
|
|
from gui_data.tkinterdnd2 import TkinterDnD, DND_FILES
|
|
from lib_v5.vr_network.model_param_init import ModelParameters
|
|
from kthread import KThread
|
|
from lib_v5 import spec_utils
|
|
from pathlib import Path
|
|
from separate import (
|
|
SeperateDemucs, SeperateMDX, SeperateMDXC, SeperateVR, # Model-related
|
|
save_format, clear_gpu_cache, # Utility functions
|
|
cuda_available, mps_available, #directml_available,
|
|
)
|
|
from playsound import playsound
|
|
from typing import List
|
|
import onnx
|
|
import re
|
|
import sys
|
|
import yaml
|
|
from ml_collections import ConfigDict
|
|
from collections import Counter
|
|
|
|
# if not is_macos:
|
|
# import torch_directml
|
|
|
|
# is_choose_arch = cuda_available and directml_available
|
|
# is_opencl_only = not cuda_available and directml_available
|
|
# is_cuda_only = cuda_available and not directml_available
|
|
|
|
is_gpu_available = cuda_available or mps_available# or directml_available
|
|
|
|
# Change the current working directory to the directory
|
|
# this file sits in
|
|
if getattr(sys, 'frozen', False):
|
|
# If the application is run as a bundle, the PyInstaller bootloader
|
|
# extends the sys module by a flag frozen=True and sets the app
|
|
# path into variable _MEIPASS'.
|
|
BASE_PATH = sys._MEIPASS
|
|
else:
|
|
BASE_PATH = os.path.dirname(os.path.abspath(__file__))
|
|
|
|
os.chdir(BASE_PATH) # Change the current working directory to the base path
|
|
|
|
SPLASH_DOC = os.path.join(BASE_PATH, 'tmp', 'splash.txt')
|
|
|
|
if os.path.isfile(SPLASH_DOC):
|
|
os.remove(SPLASH_DOC)
|
|
|
|
def get_execution_time(function, name):
|
|
start = time.time()
|
|
function()
|
|
end = time.time()
|
|
time_difference = end - start
|
|
print(f'{name} Execution Time: ', time_difference)
|
|
|
|
PREVIOUS_PATCH_WIN = 'UVR_Patch_10_6_23_4_27'
|
|
|
|
is_dnd_compatible = True
|
|
banner_placement = -2
|
|
|
|
if OPERATING_SYSTEM=="Darwin":
|
|
OPEN_FILE_func = lambda input_string:subprocess.Popen(["open", input_string])
|
|
dnd_path_check = MAC_DND_CHECK
|
|
banner_placement = -8
|
|
current_patch = PATCH_MAC
|
|
is_windows = False
|
|
is_macos = True
|
|
right_click_button = '<Button-2>'
|
|
application_extension = ".dmg"
|
|
elif OPERATING_SYSTEM=="Linux":
|
|
OPEN_FILE_func = lambda input_string:subprocess.Popen(["xdg-open", input_string])
|
|
dnd_path_check = LINUX_DND_CHECK
|
|
current_patch = PATCH_LINUX
|
|
is_windows = False
|
|
is_macos = False
|
|
right_click_button = '<Button-3>'
|
|
application_extension = ".zip"
|
|
elif OPERATING_SYSTEM=="Windows":
|
|
OPEN_FILE_func = lambda input_string:os.startfile(input_string)
|
|
dnd_path_check = WINDOWS_DND_CHECK
|
|
current_patch = PATCH
|
|
is_windows = True
|
|
is_macos = False
|
|
right_click_button = '<Button-3>'
|
|
application_extension = ".exe"
|
|
|
|
def right_click_release_linux(window, top_win=None):
|
|
if OPERATING_SYSTEM=="Linux":
|
|
root.bind('<Button-1>', lambda e:window.destroy())
|
|
if top_win:
|
|
top_win.bind('<Button-1>', lambda e:window.destroy())
|
|
|
|
if not is_windows:
|
|
import ssl
|
|
ssl._create_default_https_context = ssl._create_unverified_context
|
|
else:
|
|
from ctypes import windll, wintypes
|
|
|
|
def close_process(q:queue.Queue):
|
|
def close_splash():
|
|
name = "UVR_Launcher.exe"
|
|
for process in psutil.process_iter(attrs=["name"]):
|
|
process_name = process.info.get("name")
|
|
|
|
if process_name == name:
|
|
try:
|
|
process.terminate()
|
|
q.put(f"{name} terminated.") # Push message to queue
|
|
break
|
|
except psutil.NoSuchProcess as e:
|
|
q.put(f"Error terminating {name}: {e}") # Push error to queue
|
|
|
|
try:
|
|
with open(SPLASH_DOC, 'w') as f:
|
|
f.write('1')
|
|
except:
|
|
print('No splash screen.')
|
|
|
|
thread = KThread(target=close_splash)
|
|
thread.start()
|
|
|
|
def save_data(data):
|
|
"""
|
|
Saves given data as a .pkl (pickle) file
|
|
|
|
Paramters:
|
|
data(dict):
|
|
Dictionary containing all the necessary data to save
|
|
"""
|
|
# Open data file, create it if it does not exist
|
|
with open('data.pkl', 'wb') as data_file:
|
|
pickle.dump(data, data_file)
|
|
|
|
def load_data() -> dict:
|
|
"""
|
|
Loads saved pkl file and returns the stored data
|
|
|
|
Returns(dict):
|
|
Dictionary containing all the saved data
|
|
"""
|
|
try:
|
|
with open('data.pkl', 'rb') as data_file: # Open data file
|
|
data = pickle.load(data_file)
|
|
|
|
return data
|
|
except (ValueError, FileNotFoundError):
|
|
# Data File is corrupted or not found so recreate it
|
|
|
|
save_data(data=DEFAULT_DATA)
|
|
|
|
return load_data()
|
|
|
|
def load_model_hash_data(dictionary):
|
|
'''Get the model hash dictionary'''
|
|
with open(dictionary, 'r') as d:
|
|
return json.load(d)
|
|
|
|
def font_checker(font_file):
|
|
chosen_font_name = None
|
|
chosen_font_file = None
|
|
|
|
try:
|
|
if os.path.isfile(font_file):
|
|
with open(font_file, 'r') as d:
|
|
chosen_font = json.load(d)
|
|
|
|
chosen_font_name = chosen_font["font_name"]
|
|
if chosen_font["font_file"]:
|
|
chosen_font_file = os.path.join(OTHER_FONT_PATH, chosen_font["font_file"])
|
|
chosen_font_file = chosen_font_file if os.path.isfile(chosen_font_file) else None
|
|
except Exception as e:
|
|
print(e)
|
|
|
|
chosen_font = chosen_font_name, chosen_font_file
|
|
|
|
return chosen_font
|
|
|
|
debugger = []
|
|
|
|
#--Constants--
|
|
#Models
|
|
MODELS_DIR = os.path.join(BASE_PATH, 'models')
|
|
VR_MODELS_DIR = os.path.join(MODELS_DIR, 'VR_Models')
|
|
MDX_MODELS_DIR = os.path.join(MODELS_DIR, 'MDX_Net_Models')
|
|
DEMUCS_MODELS_DIR = os.path.join(MODELS_DIR, 'Demucs_Models')
|
|
DEMUCS_NEWER_REPO_DIR = os.path.join(DEMUCS_MODELS_DIR, 'v3_v4_repo')
|
|
MDX_MIXER_PATH = os.path.join(BASE_PATH, 'lib_v5', 'mixer.ckpt')
|
|
|
|
#Cache & Parameters
|
|
VR_HASH_DIR = os.path.join(VR_MODELS_DIR, 'model_data')
|
|
VR_HASH_JSON = os.path.join(VR_MODELS_DIR, 'model_data', 'model_data.json')
|
|
MDX_HASH_DIR = os.path.join(MDX_MODELS_DIR, 'model_data')
|
|
MDX_HASH_JSON = os.path.join(MDX_HASH_DIR, 'model_data.json')
|
|
MDX_C_CONFIG_PATH = os.path.join(MDX_HASH_DIR, 'mdx_c_configs')
|
|
|
|
DEMUCS_MODEL_NAME_SELECT = os.path.join(DEMUCS_MODELS_DIR, 'model_data', 'model_name_mapper.json')
|
|
MDX_MODEL_NAME_SELECT = os.path.join(MDX_MODELS_DIR, 'model_data', 'model_name_mapper.json')
|
|
ENSEMBLE_CACHE_DIR = os.path.join(BASE_PATH, 'gui_data', 'saved_ensembles')
|
|
SETTINGS_CACHE_DIR = os.path.join(BASE_PATH, 'gui_data', 'saved_settings')
|
|
VR_PARAM_DIR = os.path.join(BASE_PATH, 'lib_v5', 'vr_network', 'modelparams')
|
|
SAMPLE_CLIP_PATH = os.path.join(BASE_PATH, 'temp_sample_clips')
|
|
ENSEMBLE_TEMP_PATH = os.path.join(BASE_PATH, 'ensemble_temps')
|
|
DOWNLOAD_MODEL_CACHE = os.path.join(BASE_PATH, 'gui_data', 'model_manual_download.json')
|
|
|
|
#CR Text
|
|
CR_TEXT = os.path.join(BASE_PATH, 'gui_data', 'cr_text.txt')
|
|
|
|
#Style
|
|
ICON_IMG_PATH = os.path.join(BASE_PATH, 'gui_data', 'img', 'GUI-Icon.ico')
|
|
if not is_windows:
|
|
MAIN_ICON_IMG_PATH = os.path.join(BASE_PATH, 'gui_data', 'img', 'GUI-Icon.png')
|
|
|
|
OWN_FONT_PATH = os.path.join(BASE_PATH, 'gui_data', 'own_font.json')
|
|
|
|
MAIN_FONT_NAME = 'Montserrat'
|
|
SEC_FONT_NAME = 'Century Gothic'
|
|
FONT_PATH = os.path.join(BASE_PATH, 'gui_data', 'fonts', 'Montserrat', 'Montserrat.ttf')#
|
|
SEC_FONT_PATH = os.path.join(BASE_PATH, 'gui_data', 'fonts', 'centurygothic', 'GOTHIC.ttf')#
|
|
OTHER_FONT_PATH = os.path.join(BASE_PATH, 'gui_data', 'fonts', 'other')#
|
|
|
|
FONT_MAPPER = {MAIN_FONT_NAME:FONT_PATH,
|
|
SEC_FONT_NAME:SEC_FONT_PATH}
|
|
|
|
#Other
|
|
COMPLETE_CHIME = os.path.join(BASE_PATH, 'gui_data', 'complete_chime.wav')
|
|
FAIL_CHIME = os.path.join(BASE_PATH, 'gui_data', 'fail_chime.wav')
|
|
CHANGE_LOG = os.path.join(BASE_PATH, 'gui_data', 'change_log.txt')
|
|
|
|
DENOISER_MODEL_PATH = os.path.join(VR_MODELS_DIR, 'UVR-DeNoise-Lite.pth')
|
|
DEVERBER_MODEL_PATH = os.path.join(VR_MODELS_DIR, 'UVR-DeEcho-DeReverb.pth')
|
|
|
|
MODEL_DATA_URLS = [VR_MODEL_DATA_LINK, MDX_MODEL_DATA_LINK, MDX_MODEL_NAME_DATA_LINK, DEMUCS_MODEL_NAME_DATA_LINK]
|
|
MODEL_DATA_FILES = [VR_HASH_JSON, MDX_HASH_JSON, MDX_MODEL_NAME_SELECT, DEMUCS_MODEL_NAME_SELECT]
|
|
|
|
file_check(os.path.join(MODELS_DIR, 'Main_Models'), VR_MODELS_DIR)
|
|
file_check(os.path.join(DEMUCS_MODELS_DIR, 'v3_repo'), DEMUCS_NEWER_REPO_DIR)
|
|
remove_unneeded_yamls(DEMUCS_MODELS_DIR)
|
|
|
|
remove_temps(ENSEMBLE_TEMP_PATH)
|
|
remove_temps(SAMPLE_CLIP_PATH)
|
|
remove_temps(os.path.join(BASE_PATH, 'img'))
|
|
|
|
if not os.path.isdir(ENSEMBLE_TEMP_PATH):
|
|
os.mkdir(ENSEMBLE_TEMP_PATH)
|
|
|
|
if not os.path.isdir(SAMPLE_CLIP_PATH):
|
|
os.mkdir(SAMPLE_CLIP_PATH)
|
|
|
|
model_hash_table = {}
|
|
data = load_data()
|
|
|
|
def drop(event, accept_mode: str = 'files'):
|
|
path = event.data
|
|
if accept_mode == 'folder':
|
|
path = path.replace('{', '').replace('}', '')
|
|
if not os.path.isdir(path):
|
|
messagebox.showerror(parent=root,
|
|
title=INVALID_FOLDER_ERROR_TEXT[0],
|
|
message=INVALID_FOLDER_ERROR_TEXT[1])
|
|
return
|
|
root.export_path_var.set(path)
|
|
elif accept_mode in ['files', FILE_1, FILE_2, FILE_1_LB, FILE_2_LB]:
|
|
path = path.replace("{", "").replace("}", "")
|
|
for dnd_file in dnd_path_check:
|
|
path = path.replace(f" {dnd_file}", f";{dnd_file}")
|
|
path = path.split(';')
|
|
path[-1] = path[-1].replace(';', '')
|
|
|
|
if accept_mode == 'files':
|
|
root.inputPaths = tuple(path)
|
|
root.process_input_selections()
|
|
root.update_inputPaths()
|
|
elif accept_mode in [FILE_1, FILE_2]:
|
|
if len(path) == 2:
|
|
root.select_audiofile(path[0])
|
|
root.select_audiofile(path[1], is_primary=False)
|
|
root.DualBatch_inputPaths = []
|
|
root.check_dual_paths()
|
|
elif len(path) == 1:
|
|
if accept_mode == FILE_1:
|
|
root.select_audiofile(path[0])
|
|
else:
|
|
root.select_audiofile(path[0], is_primary=False)
|
|
|
|
elif accept_mode in [FILE_1_LB, FILE_2_LB]:
|
|
return path
|
|
else:
|
|
return
|
|
|
|
class ModelData():
|
|
def __init__(self, model_name: str,
|
|
selected_process_method=ENSEMBLE_MODE,
|
|
is_secondary_model=False,
|
|
primary_model_primary_stem=None,
|
|
is_primary_model_primary_stem_only=False,
|
|
is_primary_model_secondary_stem_only=False,
|
|
is_pre_proc_model=False,
|
|
is_dry_check=False,
|
|
is_change_def=False,
|
|
is_get_hash_dir_only=False,
|
|
is_vocal_split_model=False):
|
|
|
|
device_set = root.device_set_var.get()
|
|
self.DENOISER_MODEL = DENOISER_MODEL_PATH
|
|
self.DEVERBER_MODEL = DEVERBER_MODEL_PATH
|
|
self.is_deverb_vocals = root.is_deverb_vocals_var.get() if os.path.isfile(DEVERBER_MODEL_PATH) else False
|
|
self.deverb_vocal_opt = DEVERB_MAPPER[root.deverb_vocal_opt_var.get()]
|
|
self.is_denoise_model = True if root.denoise_option_var.get() == DENOISE_M and os.path.isfile(DENOISER_MODEL_PATH) else False
|
|
self.is_gpu_conversion = 0 if root.is_gpu_conversion_var.get() else -1
|
|
self.is_normalization = root.is_normalization_var.get()#
|
|
self.is_use_opencl = False#True if is_opencl_only else root.is_use_opencl_var.get()
|
|
self.is_primary_stem_only = root.is_primary_stem_only_var.get()
|
|
self.is_secondary_stem_only = root.is_secondary_stem_only_var.get()
|
|
self.is_denoise = True if not root.denoise_option_var.get() == DENOISE_NONE else False
|
|
self.is_mdx_c_seg_def = root.is_mdx_c_seg_def_var.get()#
|
|
self.mdx_batch_size = 1 if root.mdx_batch_size_var.get() == DEF_OPT else int(root.mdx_batch_size_var.get())
|
|
self.mdxnet_stem_select = root.mdxnet_stems_var.get()
|
|
self.overlap = float(root.overlap_var.get()) if not root.overlap_var.get() == DEFAULT else 0.25
|
|
self.overlap_mdx = float(root.overlap_mdx_var.get()) if not root.overlap_mdx_var.get() == DEFAULT else root.overlap_mdx_var.get()
|
|
self.overlap_mdx23 = int(float(root.overlap_mdx23_var.get()))
|
|
self.semitone_shift = float(root.semitone_shift_var.get())
|
|
self.is_pitch_change = False if self.semitone_shift == 0 else True
|
|
self.is_match_frequency_pitch = root.is_match_frequency_pitch_var.get()
|
|
self.is_mdx_ckpt = False
|
|
self.is_mdx_c = False
|
|
self.is_mdx_combine_stems = root.is_mdx23_combine_stems_var.get()#
|
|
self.mdx_c_configs = None
|
|
self.mdx_model_stems = []
|
|
self.mdx_dim_f_set = None
|
|
self.mdx_dim_t_set = None
|
|
self.mdx_stem_count = 1
|
|
self.compensate = None
|
|
self.mdx_n_fft_scale_set = None
|
|
self.wav_type_set = root.wav_type_set#
|
|
self.device_set = device_set.split(':')[-1].strip() if ':' in device_set else device_set
|
|
self.mp3_bit_set = root.mp3_bit_set_var.get()
|
|
self.save_format = root.save_format_var.get()
|
|
self.is_invert_spec = root.is_invert_spec_var.get()#
|
|
self.is_mixer_mode = False#
|
|
self.demucs_stems = root.demucs_stems_var.get()
|
|
self.is_demucs_combine_stems = root.is_demucs_combine_stems_var.get()
|
|
self.demucs_source_list = []
|
|
self.demucs_stem_count = 0
|
|
self.mixer_path = MDX_MIXER_PATH
|
|
self.model_name = model_name
|
|
self.process_method = selected_process_method
|
|
self.model_status = False if self.model_name == CHOOSE_MODEL or self.model_name == NO_MODEL else True
|
|
self.primary_stem = None
|
|
self.secondary_stem = None
|
|
self.primary_stem_native = None
|
|
self.is_ensemble_mode = False
|
|
self.ensemble_primary_stem = None
|
|
self.ensemble_secondary_stem = None
|
|
self.primary_model_primary_stem = primary_model_primary_stem
|
|
self.is_secondary_model = True if is_vocal_split_model else is_secondary_model
|
|
self.secondary_model = None
|
|
self.secondary_model_scale = None
|
|
self.demucs_4_stem_added_count = 0
|
|
self.is_demucs_4_stem_secondaries = False
|
|
self.is_4_stem_ensemble = False
|
|
self.pre_proc_model = None
|
|
self.pre_proc_model_activated = False
|
|
self.is_pre_proc_model = is_pre_proc_model
|
|
self.is_dry_check = is_dry_check
|
|
self.model_samplerate = 44100
|
|
self.model_capacity = 32, 128
|
|
self.is_vr_51_model = False
|
|
self.is_demucs_pre_proc_model_inst_mix = False
|
|
self.manual_download_Button = None
|
|
self.secondary_model_4_stem = []
|
|
self.secondary_model_4_stem_scale = []
|
|
self.secondary_model_4_stem_names = []
|
|
self.secondary_model_4_stem_model_names_list = []
|
|
self.all_models = []
|
|
self.secondary_model_other = None
|
|
self.secondary_model_scale_other = None
|
|
self.secondary_model_bass = None
|
|
self.secondary_model_scale_bass = None
|
|
self.secondary_model_drums = None
|
|
self.secondary_model_scale_drums = None
|
|
self.is_multi_stem_ensemble = False
|
|
self.is_karaoke = False
|
|
self.is_bv_model = False
|
|
self.bv_model_rebalance = 0
|
|
self.is_sec_bv_rebalance = False
|
|
self.is_change_def = is_change_def
|
|
self.model_hash_dir = None
|
|
self.is_get_hash_dir_only = is_get_hash_dir_only
|
|
self.is_secondary_model_activated = False
|
|
self.vocal_split_model = None
|
|
self.is_vocal_split_model = is_vocal_split_model
|
|
self.is_vocal_split_model_activated = False
|
|
self.is_save_inst_vocal_splitter = root.is_save_inst_set_vocal_splitter_var.get()
|
|
self.is_inst_only_voc_splitter = root.check_only_selection_stem(INST_STEM_ONLY)
|
|
self.is_save_vocal_only = root.check_only_selection_stem(IS_SAVE_VOC_ONLY)
|
|
|
|
if selected_process_method == ENSEMBLE_MODE:
|
|
self.process_method, _, self.model_name = model_name.partition(ENSEMBLE_PARTITION)
|
|
self.model_and_process_tag = model_name
|
|
self.ensemble_primary_stem, self.ensemble_secondary_stem = root.return_ensemble_stems()
|
|
|
|
is_not_secondary_or_pre_proc = not is_secondary_model and not is_pre_proc_model
|
|
self.is_ensemble_mode = is_not_secondary_or_pre_proc
|
|
|
|
if root.ensemble_main_stem_var.get() == FOUR_STEM_ENSEMBLE:
|
|
self.is_4_stem_ensemble = self.is_ensemble_mode
|
|
elif root.ensemble_main_stem_var.get() == MULTI_STEM_ENSEMBLE and root.chosen_process_method_var.get() == ENSEMBLE_MODE:
|
|
self.is_multi_stem_ensemble = True
|
|
|
|
is_not_vocal_stem = self.ensemble_primary_stem != VOCAL_STEM
|
|
self.pre_proc_model_activated = root.is_demucs_pre_proc_model_activate_var.get() if is_not_vocal_stem else False
|
|
|
|
if self.process_method == VR_ARCH_TYPE:
|
|
self.is_secondary_model_activated = root.vr_is_secondary_model_activate_var.get() if not is_secondary_model else False
|
|
self.aggression_setting = float(int(root.aggression_setting_var.get())/100)
|
|
self.is_tta = root.is_tta_var.get()
|
|
self.is_post_process = root.is_post_process_var.get()
|
|
self.window_size = int(root.window_size_var.get())
|
|
self.batch_size = 1 if root.batch_size_var.get() == DEF_OPT else int(root.batch_size_var.get())
|
|
self.crop_size = int(root.crop_size_var.get())
|
|
self.is_high_end_process = 'mirroring' if root.is_high_end_process_var.get() else 'None'
|
|
self.post_process_threshold = float(root.post_process_threshold_var.get())
|
|
self.model_capacity = 32, 128
|
|
self.model_path = os.path.join(VR_MODELS_DIR, f"{self.model_name}.pth")
|
|
self.get_model_hash()
|
|
if self.model_hash:
|
|
self.model_hash_dir = os.path.join(VR_HASH_DIR, f"{self.model_hash}.json")
|
|
if is_change_def:
|
|
self.model_data = self.change_model_data()
|
|
else:
|
|
self.model_data = self.get_model_data(VR_HASH_DIR, root.vr_hash_MAPPER) if not self.model_hash == WOOD_INST_MODEL_HASH else WOOD_INST_PARAMS
|
|
if self.model_data:
|
|
vr_model_param = os.path.join(VR_PARAM_DIR, "{}.json".format(self.model_data["vr_model_param"]))
|
|
self.primary_stem = self.model_data["primary_stem"]
|
|
self.secondary_stem = secondary_stem(self.primary_stem)
|
|
self.vr_model_param = ModelParameters(vr_model_param)
|
|
self.model_samplerate = self.vr_model_param.param['sr']
|
|
self.primary_stem_native = self.primary_stem
|
|
if "nout" in self.model_data.keys() and "nout_lstm" in self.model_data.keys():
|
|
self.model_capacity = self.model_data["nout"], self.model_data["nout_lstm"]
|
|
self.is_vr_51_model = True
|
|
self.check_if_karaokee_model()
|
|
|
|
else:
|
|
self.model_status = False
|
|
|
|
if self.process_method == MDX_ARCH_TYPE:
|
|
self.is_secondary_model_activated = root.mdx_is_secondary_model_activate_var.get() if not is_secondary_model else False
|
|
self.margin = int(root.margin_var.get())
|
|
self.chunks = 0
|
|
self.mdx_segment_size = int(root.mdx_segment_size_var.get())
|
|
self.get_mdx_model_path()
|
|
self.get_model_hash()
|
|
if self.model_hash:
|
|
self.model_hash_dir = os.path.join(MDX_HASH_DIR, f"{self.model_hash}.json")
|
|
if is_change_def:
|
|
self.model_data = self.change_model_data()
|
|
else:
|
|
self.model_data = self.get_model_data(MDX_HASH_DIR, root.mdx_hash_MAPPER)
|
|
if self.model_data:
|
|
|
|
if "config_yaml" in self.model_data:
|
|
self.is_mdx_c = True
|
|
config_path = os.path.join(MDX_C_CONFIG_PATH, self.model_data["config_yaml"])
|
|
if os.path.isfile(config_path):
|
|
with open(config_path) as f:
|
|
config = ConfigDict(yaml.load(f, Loader=yaml.FullLoader))
|
|
|
|
self.mdx_c_configs = config
|
|
|
|
if self.mdx_c_configs.training.target_instrument:
|
|
# Use target_instrument as the primary stem and set 4-stem ensemble to False
|
|
target = self.mdx_c_configs.training.target_instrument
|
|
self.mdx_model_stems = [target]
|
|
self.primary_stem = target
|
|
else:
|
|
# If no specific target_instrument, use all instruments in the training config
|
|
self.mdx_model_stems = self.mdx_c_configs.training.instruments
|
|
self.mdx_stem_count = len(self.mdx_model_stems)
|
|
|
|
# Set primary stem based on stem count
|
|
if self.mdx_stem_count == 2:
|
|
self.primary_stem = self.mdx_model_stems[0]
|
|
else:
|
|
self.primary_stem = self.mdxnet_stem_select
|
|
|
|
# Update mdxnet_stem_select based on ensemble mode
|
|
if self.is_ensemble_mode:
|
|
self.mdxnet_stem_select = self.ensemble_primary_stem
|
|
else:
|
|
self.model_status = False
|
|
else:
|
|
self.compensate = self.model_data["compensate"] if root.compensate_var.get() == AUTO_SELECT else float(root.compensate_var.get())
|
|
self.mdx_dim_f_set = self.model_data["mdx_dim_f_set"]
|
|
self.mdx_dim_t_set = self.model_data["mdx_dim_t_set"]
|
|
self.mdx_n_fft_scale_set = self.model_data["mdx_n_fft_scale_set"]
|
|
self.primary_stem = self.model_data["primary_stem"]
|
|
self.primary_stem_native = self.model_data["primary_stem"]
|
|
self.check_if_karaokee_model()
|
|
|
|
self.secondary_stem = secondary_stem(self.primary_stem)
|
|
else:
|
|
self.model_status = False
|
|
|
|
if self.process_method == DEMUCS_ARCH_TYPE:
|
|
self.is_secondary_model_activated = root.demucs_is_secondary_model_activate_var.get() if not is_secondary_model else False
|
|
if not self.is_ensemble_mode:
|
|
self.pre_proc_model_activated = root.is_demucs_pre_proc_model_activate_var.get() if not root.demucs_stems_var.get() in [VOCAL_STEM, INST_STEM] else False
|
|
self.margin_demucs = int(root.margin_demucs_var.get())
|
|
self.chunks_demucs = 0
|
|
self.shifts = int(root.shifts_var.get())
|
|
self.is_split_mode = root.is_split_mode_var.get()
|
|
self.segment = root.segment_var.get()
|
|
self.is_chunk_demucs = root.is_chunk_demucs_var.get()
|
|
self.is_primary_stem_only = root.is_primary_stem_only_var.get() if self.is_ensemble_mode else root.is_primary_stem_only_Demucs_var.get()
|
|
self.is_secondary_stem_only = root.is_secondary_stem_only_var.get() if self.is_ensemble_mode else root.is_secondary_stem_only_Demucs_var.get()
|
|
self.get_demucs_model_data()
|
|
self.get_demucs_model_path()
|
|
|
|
if self.model_status:
|
|
self.model_basename = os.path.splitext(os.path.basename(self.model_path))[0]
|
|
else:
|
|
self.model_basename = None
|
|
|
|
self.pre_proc_model_activated = self.pre_proc_model_activated if not self.is_secondary_model else False
|
|
|
|
self.is_primary_model_primary_stem_only = is_primary_model_primary_stem_only
|
|
self.is_primary_model_secondary_stem_only = is_primary_model_secondary_stem_only
|
|
|
|
is_secondary_activated_and_status = self.is_secondary_model_activated and self.model_status
|
|
is_demucs = self.process_method == DEMUCS_ARCH_TYPE
|
|
is_all_stems = root.demucs_stems_var.get() == ALL_STEMS
|
|
is_valid_ensemble = not self.is_ensemble_mode and is_all_stems and is_demucs
|
|
is_multi_stem_ensemble_demucs = self.is_multi_stem_ensemble and is_demucs
|
|
|
|
if is_secondary_activated_and_status:
|
|
if is_valid_ensemble or self.is_4_stem_ensemble or is_multi_stem_ensemble_demucs:
|
|
for key in DEMUCS_4_SOURCE_LIST:
|
|
self.secondary_model_data(key)
|
|
self.secondary_model_4_stem.append(self.secondary_model)
|
|
self.secondary_model_4_stem_scale.append(self.secondary_model_scale)
|
|
self.secondary_model_4_stem_names.append(key)
|
|
|
|
self.demucs_4_stem_added_count = sum(i is not None for i in self.secondary_model_4_stem)
|
|
self.is_secondary_model_activated = any(i is not None for i in self.secondary_model_4_stem)
|
|
self.demucs_4_stem_added_count -= 1 if self.is_secondary_model_activated else 0
|
|
|
|
if self.is_secondary_model_activated:
|
|
self.secondary_model_4_stem_model_names_list = [i.model_basename if i is not None else None for i in self.secondary_model_4_stem]
|
|
self.is_demucs_4_stem_secondaries = True
|
|
else:
|
|
primary_stem = self.ensemble_primary_stem if self.is_ensemble_mode and is_demucs else self.primary_stem
|
|
self.secondary_model_data(primary_stem)
|
|
|
|
if self.process_method == DEMUCS_ARCH_TYPE and not is_secondary_model:
|
|
if self.demucs_stem_count >= 3 and self.pre_proc_model_activated:
|
|
self.pre_proc_model = root.process_determine_demucs_pre_proc_model(self.primary_stem)
|
|
self.pre_proc_model_activated = True if self.pre_proc_model else False
|
|
self.is_demucs_pre_proc_model_inst_mix = root.is_demucs_pre_proc_model_inst_mix_var.get() if self.pre_proc_model else False
|
|
|
|
if self.is_vocal_split_model and self.model_status:
|
|
self.is_secondary_model_activated = False
|
|
if self.is_bv_model:
|
|
primary = BV_VOCAL_STEM if self.primary_stem_native == VOCAL_STEM else LEAD_VOCAL_STEM
|
|
else:
|
|
primary = LEAD_VOCAL_STEM if self.primary_stem_native == VOCAL_STEM else BV_VOCAL_STEM
|
|
self.primary_stem, self.secondary_stem = primary, secondary_stem(primary)
|
|
|
|
self.vocal_splitter_model_data()
|
|
|
|
def vocal_splitter_model_data(self):
|
|
if not self.is_secondary_model and self.model_status:
|
|
self.vocal_split_model = root.process_determine_vocal_split_model()
|
|
self.is_vocal_split_model_activated = True if self.vocal_split_model else False
|
|
|
|
if self.vocal_split_model:
|
|
if self.vocal_split_model.bv_model_rebalance:
|
|
self.is_sec_bv_rebalance = True
|
|
|
|
def secondary_model_data(self, primary_stem):
|
|
secondary_model_data = root.process_determine_secondary_model(self.process_method, primary_stem, self.is_primary_stem_only, self.is_secondary_stem_only)
|
|
self.secondary_model = secondary_model_data[0]
|
|
self.secondary_model_scale = secondary_model_data[1]
|
|
self.is_secondary_model_activated = False if not self.secondary_model else True
|
|
if self.secondary_model:
|
|
self.is_secondary_model_activated = False if self.secondary_model.model_basename == self.model_basename else True
|
|
|
|
#print("self.is_secondary_model_activated: ", self.is_secondary_model_activated)
|
|
|
|
def check_if_karaokee_model(self):
|
|
if IS_KARAOKEE in self.model_data.keys():
|
|
self.is_karaoke = self.model_data[IS_KARAOKEE]
|
|
if IS_BV_MODEL in self.model_data.keys():
|
|
self.is_bv_model = self.model_data[IS_BV_MODEL]#
|
|
if IS_BV_MODEL_REBAL in self.model_data.keys() and self.is_bv_model:
|
|
self.bv_model_rebalance = self.model_data[IS_BV_MODEL_REBAL]#
|
|
|
|
def get_mdx_model_path(self):
|
|
|
|
if self.model_name.endswith(CKPT):
|
|
self.is_mdx_ckpt = True
|
|
|
|
ext = '' if self.is_mdx_ckpt else ONNX
|
|
|
|
for file_name, chosen_mdx_model in root.mdx_name_select_MAPPER.items():
|
|
if self.model_name in chosen_mdx_model:
|
|
if file_name.endswith(CKPT):
|
|
ext = ''
|
|
self.model_path = os.path.join(MDX_MODELS_DIR, f"{file_name}{ext}")
|
|
break
|
|
else:
|
|
self.model_path = os.path.join(MDX_MODELS_DIR, f"{self.model_name}{ext}")
|
|
|
|
self.mixer_path = os.path.join(MDX_MODELS_DIR, f"mixer_val.ckpt")
|
|
|
|
def get_demucs_model_path(self):
|
|
|
|
demucs_newer = self.demucs_version in {DEMUCS_V3, DEMUCS_V4}
|
|
demucs_model_dir = DEMUCS_NEWER_REPO_DIR if demucs_newer else DEMUCS_MODELS_DIR
|
|
|
|
for file_name, chosen_model in root.demucs_name_select_MAPPER.items():
|
|
if self.model_name == chosen_model:
|
|
self.model_path = os.path.join(demucs_model_dir, file_name)
|
|
break
|
|
else:
|
|
self.model_path = os.path.join(DEMUCS_NEWER_REPO_DIR, f'{self.model_name}.yaml')
|
|
|
|
def get_demucs_model_data(self):
|
|
|
|
self.demucs_version = DEMUCS_V4
|
|
|
|
for key, value in DEMUCS_VERSION_MAPPER.items():
|
|
if value in self.model_name:
|
|
self.demucs_version = key
|
|
|
|
if DEMUCS_UVR_MODEL in self.model_name:
|
|
self.demucs_source_list, self.demucs_source_map, self.demucs_stem_count = DEMUCS_2_SOURCE, DEMUCS_2_SOURCE_MAPPER, 2
|
|
else:
|
|
self.demucs_source_list, self.demucs_source_map, self.demucs_stem_count = DEMUCS_4_SOURCE, DEMUCS_4_SOURCE_MAPPER, 4
|
|
|
|
if not self.is_ensemble_mode:
|
|
self.primary_stem = PRIMARY_STEM if self.demucs_stems == ALL_STEMS else self.demucs_stems
|
|
self.secondary_stem = secondary_stem(self.primary_stem)
|
|
|
|
def get_model_data(self, model_hash_dir, hash_mapper:dict):
|
|
model_settings_json = os.path.join(model_hash_dir, f"{self.model_hash}.json")
|
|
|
|
if os.path.isfile(model_settings_json):
|
|
with open(model_settings_json, 'r') as json_file:
|
|
return json.load(json_file)
|
|
else:
|
|
for hash, settings in hash_mapper.items():
|
|
if self.model_hash in hash:
|
|
return settings
|
|
|
|
return self.get_model_data_from_popup()
|
|
|
|
def change_model_data(self):
|
|
if self.is_get_hash_dir_only:
|
|
return None
|
|
else:
|
|
return self.get_model_data_from_popup()
|
|
|
|
def get_model_data_from_popup(self):
|
|
if self.is_dry_check:
|
|
return None
|
|
|
|
if not self.is_change_def:
|
|
confirm = messagebox.askyesno(
|
|
title=UNRECOGNIZED_MODEL[0],
|
|
message=f'"{self.model_name}"{UNRECOGNIZED_MODEL[1]}',
|
|
parent=root
|
|
)
|
|
if not confirm:
|
|
return None
|
|
|
|
if self.process_method == VR_ARCH_TYPE:
|
|
root.pop_up_vr_param(self.model_hash)
|
|
return root.vr_model_params
|
|
elif self.process_method == MDX_ARCH_TYPE:
|
|
root.pop_up_mdx_model(self.model_hash, self.model_path)
|
|
return root.mdx_model_params
|
|
|
|
def get_model_hash(self):
|
|
self.model_hash = None
|
|
|
|
if not os.path.isfile(self.model_path):
|
|
self.model_status = False
|
|
self.model_hash is None
|
|
else:
|
|
if model_hash_table:
|
|
for (key, value) in model_hash_table.items():
|
|
if self.model_path == key:
|
|
self.model_hash = value
|
|
break
|
|
|
|
if not self.model_hash:
|
|
try:
|
|
with open(self.model_path, 'rb') as f:
|
|
f.seek(- 10000 * 1024, 2)
|
|
self.model_hash = hashlib.md5(f.read()).hexdigest()
|
|
except:
|
|
self.model_hash = hashlib.md5(open(self.model_path,'rb').read()).hexdigest()
|
|
|
|
table_entry = {self.model_path: self.model_hash}
|
|
model_hash_table.update(table_entry)
|
|
|
|
#print(self.model_name," - ", self.model_hash)
|
|
|
|
class Ensembler():
|
|
def __init__(self, is_manual_ensemble=False):
|
|
self.is_save_all_outputs_ensemble = root.is_save_all_outputs_ensemble_var.get()
|
|
chosen_ensemble_name = '{}'.format(root.chosen_ensemble_var.get().replace(" ", "_")) if not root.chosen_ensemble_var.get() == CHOOSE_ENSEMBLE_OPTION else 'Ensembled'
|
|
ensemble_algorithm = root.ensemble_type_var.get().partition("/")
|
|
ensemble_main_stem_pair = root.ensemble_main_stem_var.get().partition("/")
|
|
time_stamp = round(time.time())
|
|
self.audio_tool = MANUAL_ENSEMBLE
|
|
self.main_export_path = Path(root.export_path_var.get())
|
|
self.chosen_ensemble = f"_{chosen_ensemble_name}" if root.is_append_ensemble_name_var.get() else ''
|
|
ensemble_folder_name = self.main_export_path if self.is_save_all_outputs_ensemble else ENSEMBLE_TEMP_PATH
|
|
self.ensemble_folder_name = os.path.join(ensemble_folder_name, '{}_Outputs_{}'.format(chosen_ensemble_name, time_stamp))
|
|
self.is_testing_audio = f"{time_stamp}_" if root.is_testing_audio_var.get() else ''
|
|
self.primary_algorithm = ensemble_algorithm[0]
|
|
self.secondary_algorithm = ensemble_algorithm[2]
|
|
self.ensemble_primary_stem = ensemble_main_stem_pair[0]
|
|
self.ensemble_secondary_stem = ensemble_main_stem_pair[2]
|
|
self.is_normalization = root.is_normalization_var.get()
|
|
self.is_wav_ensemble = root.is_wav_ensemble_var.get()
|
|
self.wav_type_set = root.wav_type_set
|
|
self.mp3_bit_set = root.mp3_bit_set_var.get()
|
|
self.save_format = root.save_format_var.get()
|
|
if not is_manual_ensemble:
|
|
os.mkdir(self.ensemble_folder_name)
|
|
|
|
def ensemble_outputs(self, audio_file_base, export_path, stem, is_4_stem=False, is_inst_mix=False):
|
|
"""Processes the given outputs and ensembles them with the chosen algorithm"""
|
|
|
|
if is_4_stem:
|
|
algorithm = root.ensemble_type_var.get()
|
|
stem_tag = stem
|
|
else:
|
|
if is_inst_mix:
|
|
algorithm = self.secondary_algorithm
|
|
stem_tag = f"{self.ensemble_secondary_stem} {INST_STEM}"
|
|
else:
|
|
algorithm = self.primary_algorithm if stem == PRIMARY_STEM else self.secondary_algorithm
|
|
stem_tag = self.ensemble_primary_stem if stem == PRIMARY_STEM else self.ensemble_secondary_stem
|
|
|
|
stem_outputs = self.get_files_to_ensemble(folder=export_path, prefix=audio_file_base, suffix=f"_({stem_tag}).wav")
|
|
audio_file_output = f"{self.is_testing_audio}{audio_file_base}{self.chosen_ensemble}_({stem_tag})"
|
|
stem_save_path = os.path.join('{}'.format(self.main_export_path),'{}.wav'.format(audio_file_output))
|
|
|
|
#print("get_files_to_ensemble: ", stem_outputs)
|
|
|
|
if len(stem_outputs) > 1:
|
|
spec_utils.ensemble_inputs(stem_outputs, algorithm, self.is_normalization, self.wav_type_set, stem_save_path, is_wave=self.is_wav_ensemble)
|
|
save_format(stem_save_path, self.save_format, self.mp3_bit_set)
|
|
|
|
if self.is_save_all_outputs_ensemble:
|
|
for i in stem_outputs:
|
|
save_format(i, self.save_format, self.mp3_bit_set)
|
|
else:
|
|
for i in stem_outputs:
|
|
try:
|
|
os.remove(i)
|
|
except Exception as e:
|
|
print(e)
|
|
|
|
def ensemble_manual(self, audio_inputs, audio_file_base, is_bulk=False):
|
|
"""Processes the given outputs and ensembles them with the chosen algorithm"""
|
|
|
|
is_mv_sep = True
|
|
|
|
if is_bulk:
|
|
number_list = list(set([os.path.basename(i).split("_")[0] for i in audio_inputs]))
|
|
for n in number_list:
|
|
current_list = [i for i in audio_inputs if os.path.basename(i).startswith(n)]
|
|
audio_file_base = os.path.basename(current_list[0]).split('.wav')[0]
|
|
stem_testing = "instrum" if "Instrumental" in audio_file_base else "vocals"
|
|
if is_mv_sep:
|
|
audio_file_base = audio_file_base.split("_")
|
|
audio_file_base = f"{audio_file_base[1]}_{audio_file_base[2]}_{stem_testing}"
|
|
self.ensemble_manual_process(current_list, audio_file_base, is_bulk)
|
|
else:
|
|
self.ensemble_manual_process(audio_inputs, audio_file_base, is_bulk)
|
|
|
|
def ensemble_manual_process(self, audio_inputs, audio_file_base, is_bulk):
|
|
|
|
algorithm = root.choose_algorithm_var.get()
|
|
algorithm_text = "" if is_bulk else f"_({root.choose_algorithm_var.get()})"
|
|
stem_save_path = os.path.join('{}'.format(self.main_export_path),'{}{}{}.wav'.format(self.is_testing_audio, audio_file_base, algorithm_text))
|
|
spec_utils.ensemble_inputs(audio_inputs, algorithm, self.is_normalization, self.wav_type_set, stem_save_path, is_wave=self.is_wav_ensemble)
|
|
save_format(stem_save_path, self.save_format, self.mp3_bit_set)
|
|
|
|
def get_files_to_ensemble(self, folder="", prefix="", suffix=""):
|
|
"""Grab all the files to be ensembled"""
|
|
|
|
return [os.path.join(folder, i) for i in os.listdir(folder) if i.startswith(prefix) and i.endswith(suffix)]
|
|
|
|
def combine_audio(self, audio_inputs, audio_file_base):
|
|
save_format_ = lambda save_path:save_format(save_path, root.save_format_var.get(), root.mp3_bit_set_var.get())
|
|
spec_utils.combine_audio(audio_inputs,
|
|
os.path.join(self.main_export_path, f"{self.is_testing_audio}{audio_file_base}"),
|
|
self.wav_type_set,
|
|
save_format=save_format_)
|
|
|
|
class AudioTools():
|
|
def __init__(self, audio_tool):
|
|
time_stamp = round(time.time())
|
|
self.audio_tool = audio_tool
|
|
self.main_export_path = Path(root.export_path_var.get())
|
|
self.wav_type_set = root.wav_type_set
|
|
self.is_normalization = root.is_normalization_var.get()
|
|
self.is_testing_audio = f"{time_stamp}_" if root.is_testing_audio_var.get() else ''
|
|
self.save_format = lambda save_path:save_format(save_path, root.save_format_var.get(), root.mp3_bit_set_var.get())
|
|
self.align_window = TIME_WINDOW_MAPPER[root.time_window_var.get()]
|
|
self.align_intro_val = INTRO_MAPPER[root.intro_analysis_var.get()]
|
|
self.db_analysis_val = VOLUME_MAPPER[root.db_analysis_var.get()]
|
|
self.is_save_align = root.is_save_align_var.get()#
|
|
self.is_match_silence = root.is_match_silence_var.get()#
|
|
self.is_spec_match = root.is_spec_match_var.get()
|
|
|
|
self.phase_option = root.phase_option_var.get()#
|
|
self.phase_shifts = PHASE_SHIFTS_OPT[root.phase_shifts_var.get()]
|
|
|
|
def align_inputs(self, audio_inputs, audio_file_base, audio_file_2_base, command_Text, set_progress_bar):
|
|
audio_file_base = f"{self.is_testing_audio}{audio_file_base}"
|
|
audio_file_2_base = f"{self.is_testing_audio}{audio_file_2_base}"
|
|
|
|
aligned_path = os.path.join('{}'.format(self.main_export_path),'{}_(Aligned).wav'.format(audio_file_2_base))
|
|
inverted_path = os.path.join('{}'.format(self.main_export_path),'{}_(Inverted).wav'.format(audio_file_base))
|
|
|
|
spec_utils.align_audio(audio_inputs[0],
|
|
audio_inputs[1],
|
|
aligned_path,
|
|
inverted_path,
|
|
self.wav_type_set,
|
|
self.is_save_align,
|
|
command_Text,
|
|
self.save_format,
|
|
align_window=self.align_window,
|
|
align_intro_val=self.align_intro_val,
|
|
db_analysis=self.db_analysis_val,
|
|
set_progress_bar=set_progress_bar,
|
|
phase_option=self.phase_option,
|
|
phase_shifts=self.phase_shifts,
|
|
is_match_silence=self.is_match_silence,
|
|
is_spec_match=self.is_spec_match)
|
|
|
|
def match_inputs(self, audio_inputs, audio_file_base, command_Text):
|
|
|
|
target = audio_inputs[0]
|
|
reference = audio_inputs[1]
|
|
|
|
command_Text(f"Processing... ")
|
|
|
|
save_path = os.path.join('{}'.format(self.main_export_path),'{}_(Matched).wav'.format(f"{self.is_testing_audio}{audio_file_base}"))
|
|
|
|
match.process(
|
|
target=target,
|
|
reference=reference,
|
|
results=[match.save_audiofile(save_path, wav_set=self.wav_type_set),
|
|
],
|
|
)
|
|
|
|
self.save_format(save_path)
|
|
|
|
def combine_audio(self, audio_inputs, audio_file_base):
|
|
spec_utils.combine_audio(audio_inputs,
|
|
os.path.join(self.main_export_path, f"{self.is_testing_audio}{audio_file_base}"),
|
|
self.wav_type_set,
|
|
save_format=self.save_format)
|
|
|
|
def pitch_or_time_shift(self, audio_file, audio_file_base):
|
|
is_time_correction = True
|
|
rate = float(root.time_stretch_rate_var.get()) if self.audio_tool == TIME_STRETCH else float(root.pitch_rate_var.get())
|
|
is_pitch = False if self.audio_tool == TIME_STRETCH else True
|
|
if is_pitch:
|
|
is_time_correction = True if root.is_time_correction_var.get() else False
|
|
file_text = TIME_TEXT if self.audio_tool == TIME_STRETCH else PITCH_TEXT
|
|
save_path = os.path.join(self.main_export_path, f"{self.is_testing_audio}{audio_file_base}{file_text}.wav")
|
|
spec_utils.augment_audio(save_path, audio_file, rate, self.is_normalization, self.wav_type_set, self.save_format, is_pitch=is_pitch, is_time_correction=is_time_correction)
|
|
|
|
class ToolTip(object):
|
|
|
|
def __init__(self, widget):
|
|
self.widget = widget
|
|
self.tooltip = None
|
|
|
|
def showtip(self, text, is_message_box=False, is_success_message=None):#
|
|
self.hidetip()
|
|
def create_label_config():
|
|
|
|
font_size = FONT_SIZE_3 if is_message_box else FONT_SIZE_2
|
|
|
|
"""Helper function to generate label configurations."""
|
|
common_config = {
|
|
"text": text,
|
|
"relief": tk.SOLID,
|
|
"borderwidth": 1,
|
|
"font": (MAIN_FONT_NAME, f"{font_size}", "normal")
|
|
}
|
|
if is_message_box:
|
|
background_color = "#03692d" if is_success_message else "#8B0000"
|
|
return {**common_config, "background": background_color, "foreground": "#ffffff"}
|
|
else:
|
|
return {**common_config, "background": "#1C1C1C", "foreground": "#ffffff",
|
|
"highlightcolor": "#898b8e", "justify": tk.LEFT}
|
|
|
|
if is_message_box:
|
|
temp_tooltip = tk.Toplevel(self.widget)
|
|
temp_tooltip.wm_overrideredirect(True)
|
|
temp_tooltip.withdraw()
|
|
label = tk.Label(temp_tooltip, **create_label_config())
|
|
label.pack()
|
|
temp_tooltip.update() if is_windows else temp_tooltip.update_idletasks()
|
|
|
|
x = self.widget.winfo_rootx() + (self.widget.winfo_width() // 2) - (temp_tooltip.winfo_reqwidth() // 2)
|
|
y = self.widget.winfo_rooty() + self.widget.winfo_height()
|
|
|
|
temp_tooltip.destroy()
|
|
else:
|
|
x, y, _, _ = self.widget.bbox("insert")
|
|
x += self.widget.winfo_rootx() + 25
|
|
y += self.widget.winfo_rooty() + 25
|
|
|
|
# Create the actual tooltip
|
|
self.tooltip = tk.Toplevel(self.widget)
|
|
self.tooltip.wm_overrideredirect(True)
|
|
self.tooltip.wm_geometry(f"+{x}+{y}")
|
|
|
|
label_config = create_label_config()
|
|
if not is_message_box:
|
|
label_config['padx'] = 10 # horizontal padding
|
|
label_config['pady'] = 10 # vertical padding
|
|
label_config["wraplength"] = 750
|
|
label = tk.Label(self.tooltip, **label_config)
|
|
|
|
label.pack()
|
|
|
|
if is_message_box:
|
|
self.tooltip.after(3000 if type(is_success_message) is bool else 2000, self.hidetip)
|
|
|
|
def hidetip(self):
|
|
if self.tooltip:
|
|
self.tooltip.destroy()
|
|
self.tooltip = None
|
|
|
|
class ListboxBatchFrame(tk.Frame):
|
|
def __init__(self, master=None, name="Listbox", command=None, image_sel=None, img_mapper=None):
|
|
super().__init__(master)
|
|
self.master = master
|
|
|
|
self.path_list = [] # A list to keep track of the paths
|
|
self.basename_to_path = {} # A dict to map basenames to paths
|
|
|
|
self.label = tk.Label(self, text=name, font=(MAIN_FONT_NAME, f"{FONT_SIZE_5}"), foreground=FG_COLOR)
|
|
self.label.pack(pady=(10, 8)) # add padding between label and listbox
|
|
|
|
self.input_button = ttk.Button(self, text=SELECT_INPUTS, command=self.select_input) # create button for selecting files
|
|
self.input_button.pack(pady=(0, 10)) # add padding between button and next widget
|
|
|
|
self.listbox = tk.Listbox(self, activestyle='dotbox', font=(MAIN_FONT_NAME, f"{FONT_SIZE_4}"), foreground='#cdd3ce', background='#101414', exportselection=0, width=70, height=15)
|
|
self.listbox.pack(fill="both", expand=True)
|
|
|
|
self.button_frame = tk.Frame(self)
|
|
self.button_frame.pack()
|
|
|
|
self.up_button = ttk.Button(self.button_frame, image=img_mapper["up"], command=self.move_up)
|
|
self.up_button.grid(row=0, column=0)
|
|
|
|
self.down_button = ttk.Button(self.button_frame, image=img_mapper["down"], command=self.move_down)
|
|
self.down_button.grid(row=0, column=1)
|
|
|
|
if command and image_sel:
|
|
self.move_button = ttk.Button(self.button_frame, image=image_sel, command=command)
|
|
self.move_button.grid(row=0, column=2)
|
|
|
|
self.duplicate_button = ttk.Button(self.button_frame, image=img_mapper["copy"], command=self.duplicate_selected)
|
|
self.duplicate_button.grid(row=0, column=3)
|
|
|
|
self.delete_button = ttk.Button(self.button_frame, image=img_mapper["clear"], command=self.delete_selected)
|
|
self.delete_button.grid(row=0, column=4)
|
|
|
|
def delete_selected(self):
|
|
selected = self.listbox.curselection()
|
|
if selected:
|
|
basename = self.listbox.get(selected[0]).split(": ", 1)[1] # We get the actual basename here, without the index
|
|
path_to_delete = self.basename_to_path[basename] # store the path to delete
|
|
del self.basename_to_path[basename] # delete from the dict
|
|
self.path_list.remove(path_to_delete) # delete from the list
|
|
self.listbox.delete(selected)
|
|
self.update_displayed_index()
|
|
|
|
def select_input(self, inputs=None):
|
|
files = inputs if inputs else root.show_file_dialog(dialoge_type=MULTIPLE_FILE)
|
|
for file in files:
|
|
if file not in self.path_list: # only add file if it's not already in the list
|
|
basename = os.path.basename(file)
|
|
self.listbox.insert(tk.END, basename) # insert basename to the listbox
|
|
self.path_list.append(file) # append the file path to the list
|
|
self.basename_to_path[basename] = file # add to the dict
|
|
self.update_displayed_index(is_acc_dupe=False)
|
|
|
|
def duplicate_selected(self):
|
|
selected = self.listbox.curselection()
|
|
if selected:
|
|
basename = self.listbox.get(selected[0]).split(": ", 1)[1] # We get the actual basename here, without the index
|
|
path_to_duplicate = self.basename_to_path[basename] # store the path to duplicate
|
|
self.path_list.append(path_to_duplicate) # add the duplicated path to the list
|
|
self.update_displayed_index() # redraw listbox with the duplicated item
|
|
|
|
def update_displayed_index(self, inputs=None, is_acc_dupe=True):
|
|
self.basename_to_path = {} # reset the dictionary
|
|
|
|
if inputs:
|
|
self.path_list = inputs
|
|
|
|
basename_count = Counter(self.path_list) # count occurrences of each path
|
|
|
|
for i in range(len(self.path_list)):
|
|
basename = os.path.basename(self.path_list[i])
|
|
|
|
# If the path is not unique or we are adding a duplicate
|
|
if basename_count[self.path_list[i]] > 1 and is_acc_dupe:
|
|
j = 1
|
|
new_basename = f"{basename} ({j})"
|
|
while new_basename in self.basename_to_path:
|
|
j += 1
|
|
new_basename = f"{basename} ({j})"
|
|
basename = new_basename
|
|
|
|
self.basename_to_path[basename] = self.path_list[i] # update the dict with the new order
|
|
self.listbox.delete(i)
|
|
self.listbox.insert(i, f"{i + 1}: {basename}")
|
|
|
|
def move_up(self):
|
|
selected = self.listbox.curselection()
|
|
if selected and selected[0] > 0:
|
|
# Swap items in path_list
|
|
self.path_list[selected[0] - 1], self.path_list[selected[0]] = self.path_list[selected[0]], self.path_list[selected[0] - 1]
|
|
# Redraw listbox
|
|
self.update_displayed_index()
|
|
# Reselect item
|
|
self.listbox.select_set(selected[0] - 1)
|
|
|
|
def move_down(self):
|
|
selected = self.listbox.curselection()
|
|
if selected and selected[0] < self.listbox.size() - 1:
|
|
# Swap items in path_list
|
|
self.path_list[selected[0] + 1], self.path_list[selected[0]] = self.path_list[selected[0]], self.path_list[selected[0] + 1]
|
|
# Redraw listbox
|
|
self.update_displayed_index()
|
|
# Reselect item
|
|
self.listbox.select_set(selected[0] + 1)
|
|
|
|
def get_selected_path(self):
|
|
"""Returns the path associated with the selected entry."""
|
|
selected = self.listbox.curselection()
|
|
if selected:
|
|
basename = self.listbox.get(selected[0]).split(": ", 1)[1] # We get the actual basename here, without the index
|
|
path = self.basename_to_path[basename] # get the path associated with the basename
|
|
return path
|
|
return None
|
|
|
|
class ComboBoxEditableMenu(ttk.Combobox):
|
|
def __init__(self, master=None, pattern=None, default=None, width=None, is_stay_disabled=False, **kw):
|
|
|
|
if 'values' in kw:
|
|
kw['values'] = tuple(kw['values']) + (OPT_SEPARATOR, USER_INPUT)
|
|
else:
|
|
kw['values'] = (USER_INPUT)
|
|
|
|
super().__init__(master, **kw)
|
|
|
|
self.textvariable = kw.get('textvariable', tk.StringVar())
|
|
self.pattern = pattern
|
|
self.test = 1
|
|
self.tooltip = ToolTip(self)
|
|
self.is_user_input_var = tk.BooleanVar(value=False)
|
|
self.is_stay_disabled = is_stay_disabled
|
|
|
|
if isinstance(default, (str, int)):
|
|
self.default = default
|
|
else:
|
|
self.default = default[0]
|
|
|
|
self.menu_combobox_configure()
|
|
self.var_validation(is_start_up=True)
|
|
|
|
if width:
|
|
self.configure(width=width)
|
|
|
|
def menu_combobox_configure(self):
|
|
self.bind('<<ComboboxSelected>>', self.check_input)
|
|
self.bind('<Button-1>', lambda e:self.focus())
|
|
self.bind('<FocusIn>', self.focusin)
|
|
self.bind('<FocusOut>', lambda e: self.var_validation(is_focus_only=True))
|
|
|
|
if is_macos:
|
|
self.bind('<Enter>', lambda e:self.button_released())
|
|
|
|
if not self.is_stay_disabled:
|
|
self.configure(state=READ_ONLY)
|
|
|
|
def check_input(self, event=None):
|
|
if self.textvariable.get() == USER_INPUT:
|
|
self.textvariable.set('')
|
|
self.configure(state=tk.NORMAL)
|
|
self.focus()
|
|
self.selection_range(0, 0)
|
|
else:
|
|
self.var_validation()
|
|
|
|
def var_validation(self, is_focus_only=False, is_start_up=False):
|
|
if is_focus_only and not self.is_stay_disabled:
|
|
self.configure(state=READ_ONLY)
|
|
|
|
if re.fullmatch(self.pattern, self.textvariable.get()) is None:
|
|
if not is_start_up and not self.textvariable.get() in (OPT_SEPARATOR, USER_INPUT):
|
|
self.tooltip.showtip(INVALID_INPUT_E, True)
|
|
|
|
self.textvariable.set(self.default)
|
|
|
|
def button_released(self, e=None):
|
|
self.event_generate('<Button-3>')
|
|
self.event_generate('<ButtonRelease-3>')
|
|
|
|
def focusin(self, e):
|
|
self.selection_clear()
|
|
if is_macos:
|
|
self.event_generate('<Leave>')
|
|
|
|
class ComboBoxMenu(ttk.Combobox):
|
|
def __init__(self, master=None, dropdown_name=None, offset=185, is_download_menu=False, command=None, width=None, **kw):
|
|
super().__init__(master, **kw)
|
|
|
|
# Configure the combobox using the menu_combobox_configure method
|
|
self.menu_combobox_configure(is_download_menu, width=width)
|
|
|
|
# Check if both dropdown_name and 'values' are provided to update dropdown size
|
|
if dropdown_name and 'values' in kw:
|
|
self.update_dropdown_size(kw['values'], dropdown_name, offset)
|
|
|
|
if command:
|
|
self.command(command)
|
|
|
|
def menu_combobox_configure(self, is_download_menu=False, command=None, width=None):
|
|
self.bind('<FocusIn>', self.focusin)
|
|
self.bind('<MouseWheel>', lambda e:"break")
|
|
|
|
if is_macos:
|
|
self.bind('<Enter>', lambda e:self.button_released())
|
|
|
|
if not is_download_menu:
|
|
self.configure(state=READ_ONLY)
|
|
|
|
if command:
|
|
self.command(command)
|
|
|
|
if width:
|
|
self.configure(width=width)
|
|
|
|
def button_released(self, e=None):
|
|
self.event_generate('<Button-3>')
|
|
self.event_generate('<ButtonRelease-3>')
|
|
|
|
def command(self, command):
|
|
if not self.bind('<<ComboboxSelected>>'):
|
|
self.bind('<<ComboboxSelected>>', command)
|
|
|
|
def focusin(self, e):
|
|
self.selection_clear()
|
|
if is_macos:
|
|
self.event_generate('<Leave>')
|
|
|
|
def update_dropdown_size(self, option_list, dropdown_name, offset=185, command=None):
|
|
dropdown_style = f"{dropdown_name}.TCombobox"
|
|
if option_list:
|
|
max_string = max(option_list, key=len)
|
|
font = Font(font=self.cget('font'))
|
|
width_in_pixels = font.measure(max_string) - offset
|
|
width_in_pixels = 0 if width_in_pixels < 0 else width_in_pixels
|
|
else:
|
|
width_in_pixels = 0
|
|
|
|
style = ttk.Style(self)
|
|
style.configure(dropdown_style, padding=(0, 0, 0, 0), postoffset=(0, 0, width_in_pixels, 0))
|
|
self.configure(style=dropdown_style)
|
|
|
|
if command:
|
|
self.command(command)
|
|
|
|
class ThreadSafeConsole(tk.Text):
|
|
"""
|
|
Text Widget which is thread safe for tkinter
|
|
"""
|
|
|
|
def __init__(self, master, **options):
|
|
tk.Text.__init__(self, master, **options)
|
|
self.queue = queue.Queue()
|
|
self.update_me()
|
|
|
|
def write(self, line):
|
|
self.queue.put(line)
|
|
|
|
def clear(self):
|
|
self.queue.put(None)
|
|
|
|
def update_me(self):
|
|
self.configure(state=tk.NORMAL)
|
|
try:
|
|
while 1:
|
|
line = self.queue.get_nowait()
|
|
if line is None:
|
|
self.delete(1.0, tk.END)
|
|
else:
|
|
self.insert(tk.END, str(line))
|
|
self.see(tk.END)
|
|
self.update_idletasks()
|
|
except queue.Empty:
|
|
pass
|
|
self.configure(state=tk.DISABLED)
|
|
self.after(100, self.update_me)
|
|
|
|
def copy_text(self):
|
|
hightlighted_text = self.selection_get()
|
|
self.clipboard_clear()
|
|
self.clipboard_append(hightlighted_text)
|
|
|
|
def select_all_text(self):
|
|
self.tag_add('sel', '1.0', 'end')
|
|
|
|
class MainWindow(TkinterDnD.Tk if is_dnd_compatible else tk.Tk):
|
|
# --Constants--
|
|
# Layout
|
|
|
|
IMAGE_HEIGHT = IMAGE_HEIGHT
|
|
FILEPATHS_HEIGHT = FILEPATHS_HEIGHT
|
|
OPTIONS_HEIGHT = OPTIONS_HEIGHT
|
|
CONVERSIONBUTTON_HEIGHT = CONVERSIONBUTTON_HEIGHT
|
|
COMMAND_HEIGHT = COMMAND_HEIGHT
|
|
PROGRESS_HEIGHT = PROGRESS_HEIGHT
|
|
PADDING = PADDING
|
|
WIDTH = WIDTH
|
|
COL1_ROWS = 11
|
|
COL2_ROWS = 11
|
|
|
|
def __init__(self):
|
|
#Run the __init__ method on the tk.Tk class
|
|
super().__init__()
|
|
|
|
self.set_app_font()
|
|
|
|
style = ttk.Style(self)
|
|
style.map('TCombobox', selectbackground=[('focus', '#0c0c0c')], selectforeground=[('focus', 'white')])
|
|
style.configure('TCombobox', selectbackground='#0c0c0c')
|
|
#style.configure('TCheckbutton', indicatorsize=30)
|
|
|
|
# Calculate window height
|
|
height = self.IMAGE_HEIGHT + self.FILEPATHS_HEIGHT + self.OPTIONS_HEIGHT
|
|
height += self.CONVERSIONBUTTON_HEIGHT + self.COMMAND_HEIGHT + self.PROGRESS_HEIGHT
|
|
height += self.PADDING * 5 # Padding
|
|
width = self.WIDTH
|
|
self.main_window_width = width
|
|
self.main_window_height = height
|
|
|
|
# --Window Settings--
|
|
self.withdraw()
|
|
self.title('Ultimate Vocal Remover')
|
|
# Set Geometry and Center Window
|
|
self.geometry('{width}x{height}+{xpad}+{ypad}'.format(
|
|
width=self.main_window_width,
|
|
height=height,
|
|
xpad=int(self.winfo_screenwidth()/2 - width/2),
|
|
ypad=int(self.winfo_screenheight()/2 - height/2 - 30)))
|
|
|
|
self.iconbitmap(ICON_IMG_PATH) if is_windows else self.tk.call('wm', 'iconphoto', self._w, tk.PhotoImage(file=MAIN_ICON_IMG_PATH))
|
|
self.protocol("WM_DELETE_WINDOW", self.save_values)
|
|
self.resizable(False, False)
|
|
|
|
self.msg_queue = queue.Queue()
|
|
# Create a custom style that inherits from the original Combobox style.
|
|
|
|
if not is_windows:
|
|
self.update()
|
|
|
|
#Load Images
|
|
img = ImagePath(BASE_PATH)
|
|
self.logo_img = img.open_image(path=img.banner_path, size=(width, height))
|
|
self.efile_img = img.efile_img
|
|
self.stop_img = img.stop_img
|
|
self.help_img = img.help_img
|
|
self.download_img = img.download_img
|
|
self.donate_img = img.donate_img
|
|
self.key_img = img.key_img
|
|
self.credits_img = img.credits_img
|
|
|
|
self.right_img = img.right_img
|
|
self.left_img = img.left_img
|
|
self.img_mapper = {
|
|
"down":img.down_img,
|
|
"up":img.up_img,
|
|
"copy":img.copy_img,
|
|
"clear":img.clear_img
|
|
}
|
|
|
|
#Placeholders
|
|
self.error_log_var = tk.StringVar(value='')
|
|
self.vr_secondary_model_names = []
|
|
self.mdx_secondary_model_names = []
|
|
self.demucs_secondary_model_names = []
|
|
self.vr_primary_model_names = []
|
|
self.mdx_primary_model_names = []
|
|
self.demucs_primary_model_names = []
|
|
|
|
self.vr_cache_source_mapper = {}
|
|
self.mdx_cache_source_mapper = {}
|
|
self.demucs_cache_source_mapper = {}
|
|
|
|
# -Tkinter Value Holders-
|
|
|
|
try:
|
|
self.load_saved_vars(data)
|
|
except Exception as e:
|
|
self.error_log_var.set(error_text('Loading Saved Variables', e))
|
|
self.load_saved_vars(DEFAULT_DATA)
|
|
|
|
self.cached_sources_clear()
|
|
|
|
self.method_mapper = {
|
|
VR_ARCH_PM: self.vr_model_var,
|
|
MDX_ARCH_TYPE: self.mdx_net_model_var,
|
|
DEMUCS_ARCH_TYPE: self.demucs_model_var}
|
|
|
|
self.vr_secondary_model_vars = {'voc_inst_secondary_model': self.vr_voc_inst_secondary_model_var,
|
|
'other_secondary_model': self.vr_other_secondary_model_var,
|
|
'bass_secondary_model': self.vr_bass_secondary_model_var,
|
|
'drums_secondary_model': self.vr_drums_secondary_model_var,
|
|
'is_secondary_model_activate': self.vr_is_secondary_model_activate_var,
|
|
'voc_inst_secondary_model_scale': self.vr_voc_inst_secondary_model_scale_var,
|
|
'other_secondary_model_scale': self.vr_other_secondary_model_scale_var,
|
|
'bass_secondary_model_scale': self.vr_bass_secondary_model_scale_var,
|
|
'drums_secondary_model_scale': self.vr_drums_secondary_model_scale_var}
|
|
|
|
self.demucs_secondary_model_vars = {'voc_inst_secondary_model': self.demucs_voc_inst_secondary_model_var,
|
|
'other_secondary_model': self.demucs_other_secondary_model_var,
|
|
'bass_secondary_model': self.demucs_bass_secondary_model_var,
|
|
'drums_secondary_model': self.demucs_drums_secondary_model_var,
|
|
'is_secondary_model_activate': self.demucs_is_secondary_model_activate_var,
|
|
'voc_inst_secondary_model_scale': self.demucs_voc_inst_secondary_model_scale_var,
|
|
'other_secondary_model_scale': self.demucs_other_secondary_model_scale_var,
|
|
'bass_secondary_model_scale': self.demucs_bass_secondary_model_scale_var,
|
|
'drums_secondary_model_scale': self.demucs_drums_secondary_model_scale_var}
|
|
|
|
self.mdx_secondary_model_vars = {'voc_inst_secondary_model': self.mdx_voc_inst_secondary_model_var,
|
|
'other_secondary_model': self.mdx_other_secondary_model_var,
|
|
'bass_secondary_model': self.mdx_bass_secondary_model_var,
|
|
'drums_secondary_model': self.mdx_drums_secondary_model_var,
|
|
'is_secondary_model_activate': self.mdx_is_secondary_model_activate_var,
|
|
'voc_inst_secondary_model_scale': self.mdx_voc_inst_secondary_model_scale_var,
|
|
'other_secondary_model_scale': self.mdx_other_secondary_model_scale_var,
|
|
'bass_secondary_model_scale': self.mdx_bass_secondary_model_scale_var,
|
|
'drums_secondary_model_scale': self.mdx_drums_secondary_model_scale_var}
|
|
|
|
#Main Application Vars
|
|
self.progress_bar_main_var = tk.IntVar(value=0)
|
|
self.inputPathsEntry_var = tk.StringVar(value='')
|
|
self.conversion_Button_Text_var = tk.StringVar(value=START_PROCESSING)
|
|
self.chosen_ensemble_var = tk.StringVar(value=CHOOSE_ENSEMBLE_OPTION)
|
|
self.ensemble_main_stem_var = tk.StringVar(value=CHOOSE_STEM_PAIR)
|
|
self.ensemble_type_var = tk.StringVar(value=MAX_MIN)
|
|
self.save_current_settings_var = tk.StringVar(value=SELECT_SAVED_SET)
|
|
self.demucs_stems_var = tk.StringVar(value=ALL_STEMS)
|
|
self.mdxnet_stems_var = tk.StringVar(value=ALL_STEMS)
|
|
self.is_primary_stem_only_Text_var = tk.StringVar(value='')
|
|
self.is_secondary_stem_only_Text_var = tk.StringVar(value='')
|
|
self.is_primary_stem_only_Demucs_Text_var = tk.StringVar(value='')
|
|
self.is_secondary_stem_only_Demucs_Text_var = tk.StringVar(value='')
|
|
self.scaling_var = tk.DoubleVar(value=1.0)
|
|
self.active_processing_thread = None
|
|
self.verification_thread = None
|
|
self.is_menu_settings_open = False
|
|
self.is_root_defined_var = tk.BooleanVar(value=False)
|
|
self.is_check_splash = False
|
|
|
|
self.is_open_menu_advanced_vr_options = tk.BooleanVar(value=False)
|
|
self.is_open_menu_advanced_demucs_options = tk.BooleanVar(value=False)
|
|
self.is_open_menu_advanced_mdx_options = tk.BooleanVar(value=False)
|
|
self.is_open_menu_advanced_ensemble_options = tk.BooleanVar(value=False)
|
|
self.is_open_menu_view_inputs = tk.BooleanVar(value=False)
|
|
self.is_open_menu_help = tk.BooleanVar(value=False)
|
|
self.is_open_menu_error_log = tk.BooleanVar(value=False)
|
|
self.is_open_menu_advanced_align_options = tk.BooleanVar(value=False)
|
|
|
|
self.menu_advanced_vr_options_close_window = None
|
|
self.menu_advanced_demucs_options_close_window = None
|
|
self.menu_advanced_mdx_options_close_window = None
|
|
self.menu_advanced_ensemble_options_close_window = None
|
|
self.menu_help_close_window = None
|
|
self.menu_error_log_close_window = None
|
|
self.menu_view_inputs_close_window = None
|
|
self.menu_advanced_align_options_close_window = None
|
|
|
|
self.mdx_model_params = None
|
|
self.vr_model_params = None
|
|
self.current_text_box = None
|
|
self.wav_type_set = None
|
|
self.is_online_model_menu = None
|
|
self.progress_bar_var = tk.IntVar(value=0)
|
|
self.is_confirm_error_var = tk.BooleanVar(value=False)
|
|
self.clear_cache_torch = False
|
|
self.vr_hash_MAPPER = load_model_hash_data(VR_HASH_JSON)
|
|
self.mdx_hash_MAPPER = load_model_hash_data(MDX_HASH_JSON)
|
|
self.mdx_name_select_MAPPER = load_model_hash_data(MDX_MODEL_NAME_SELECT)
|
|
self.demucs_name_select_MAPPER = load_model_hash_data(DEMUCS_MODEL_NAME_SELECT)
|
|
self.is_gpu_available = is_gpu_available
|
|
self.is_process_stopped = False
|
|
self.inputs_from_dir = []
|
|
self.iteration = 0
|
|
self.true_model_count = 0
|
|
self.vr_primary_source = None
|
|
self.vr_secondary_source = None
|
|
self.mdx_primary_source = None
|
|
self.mdx_secondary_source = None
|
|
self.demucs_primary_source = None
|
|
self.demucs_secondary_source = None
|
|
self.toplevels = []
|
|
|
|
#Download Center Vars
|
|
self.online_data = {}
|
|
self.bulletin_data = INFO_UNAVAILABLE_TEXT
|
|
self.is_online = False
|
|
self.lastest_version = ''
|
|
self.model_download_demucs_var = tk.StringVar(value='')
|
|
self.model_download_mdx_var = tk.StringVar(value='')
|
|
self.model_download_vr_var = tk.StringVar(value='')
|
|
self.selected_download_var = tk.StringVar(value=NO_MODEL)
|
|
self.select_download_var = tk.StringVar(value='')
|
|
self.download_progress_info_var = tk.StringVar(value='')
|
|
self.download_progress_percent_var = tk.StringVar(value='')
|
|
self.download_progress_bar_var = tk.IntVar(value=0)
|
|
self.download_stop_var = tk.StringVar(value='')
|
|
self.app_update_status_Text_var = tk.StringVar(value=LOADING_VERSION_INFO_TEXT)
|
|
self.app_update_button_Text_var = tk.StringVar(value=CHECK_FOR_UPDATES_TEXT)
|
|
|
|
self.user_code_validation_var = tk.StringVar(value='')
|
|
self.download_link_path_var = tk.StringVar(value='')
|
|
self.download_save_path_var = tk.StringVar(value='')
|
|
self.download_update_link_var = tk.StringVar(value='')
|
|
self.download_update_path_var = tk.StringVar(value='')
|
|
self.download_demucs_models_list = []
|
|
self.download_demucs_newer_models = []
|
|
self.refresh_list_Button = None
|
|
self.stop_download_Button_DISABLE = None
|
|
self.enable_tabs = None
|
|
self.is_download_thread_active = False
|
|
self.is_process_thread_active = False
|
|
self.is_active_processing_thread = False
|
|
self.active_download_thread = None
|
|
self.pre_proc_model_toggle = None
|
|
self.change_state_lambda = None
|
|
self.file_one_sub_var = tk.StringVar(value=FILE_ONE_MAIN_LABEL)
|
|
self.file_two_sub_var = tk.StringVar(value=FILE_TWO_MAIN_LABEL)
|
|
self.cuda_device_list = GPU_DEVICE_NUM_OPTS
|
|
self.opencl_list = GPU_DEVICE_NUM_OPTS
|
|
|
|
#Model Update
|
|
self.last_found_ensembles = ENSEMBLE_OPTIONS
|
|
self.last_found_settings = ENSEMBLE_OPTIONS
|
|
self.last_found_models = ()
|
|
self.model_data_table = ()
|
|
self.ensemble_model_list = ()
|
|
self.default_change_model_list = ()
|
|
|
|
# --Widgets--
|
|
self.fill_main_frame()
|
|
self.bind_widgets()
|
|
|
|
# --Update Widgets--
|
|
self.update_available_models()
|
|
self.update_main_widget_states()
|
|
self.update_loop()
|
|
self.update_button_states()
|
|
self.download_validate_code()
|
|
self.delete_temps(is_start_up=True)
|
|
self.ensemble_listbox_Option.configure(state=tk.DISABLED)
|
|
self.command_Text.write(f'Ultimate Vocal Remover {VERSION} [{datetime.now().strftime("%Y-%m-%d %H:%M:%S")}]')
|
|
self.update_checkbox_text = lambda:self.selection_action_process_method(self.chosen_process_method_var.get())
|
|
self.check_dual_paths()
|
|
if not is_windows:
|
|
self.update_idletasks()
|
|
self.fill_gpu_list()
|
|
self.online_data_refresh(user_refresh=False, is_start_up=True)
|
|
|
|
# Menu Functions
|
|
def main_window_LABEL_SET(self, master, text):return ttk.Label(master=master, text=text, background=BG_COLOR, font=self.font_set, foreground=FG_COLOR, anchor=tk.CENTER)
|
|
def main_window_LABEL_SUB_SET(self, master, text_var):return ttk.Label(master=master, textvariable=text_var, background=BG_COLOR, font=self.font_set, foreground=FG_COLOR, anchor=tk.CENTER)
|
|
def menu_title_LABEL_SET(self, frame, text, width=35):return ttk.Label(master=frame, text=text, font=(SEC_FONT_NAME, f"{FONT_SIZE_5}", "underline"), justify="center", foreground="#13849f", width=width, anchor=tk.CENTER)
|
|
def menu_sub_LABEL_SET(self, frame, text, font_size=FONT_SIZE_2):return ttk.Label(master=frame, text=text, font=(MAIN_FONT_NAME, f"{font_size}"), foreground=FG_COLOR, anchor=tk.CENTER)
|
|
def menu_FRAME_SET(self, frame, thickness=20):return tk.Frame(frame, highlightbackground=BG_COLOR, highlightcolor=BG_COLOR, highlightthicknes=thickness)
|
|
def check_is_menu_settings_open(self):self.menu_settings() if not self.is_menu_settings_open else None
|
|
def spacer_label(self, frame): return tk.Label(frame, text='', font=(MAIN_FONT_NAME, f"{FONT_SIZE_1}"), foreground='#868687', justify="left").grid()
|
|
|
|
#Ensemble Listbox Functions
|
|
def ensemble_listbox_get_all_selected_models(self):return [self.ensemble_listbox_Option.get(i) for i in self.ensemble_listbox_Option.curselection()]
|
|
def ensemble_listbox_select_from_indexs(self, indexes):return [self.ensemble_listbox_Option.selection_set(i) for i in indexes]
|
|
def ensemble_listbox_clear_and_insert_new(self, model_ensemble_updated):return (self.ensemble_listbox_Option.delete(0, 'end'), [self.ensemble_listbox_Option.insert(tk.END, models) for models in model_ensemble_updated])
|
|
def ensemble_listbox_get_indexes_for_files(self, updated, selected):return [updated.index(model) for model in selected if model in updated]
|
|
|
|
def set_app_font(self):
|
|
chosen_font_name, chosen_font_file = font_checker(OWN_FONT_PATH)
|
|
|
|
if chosen_font_name:
|
|
gui_data.sv_ttk.set_theme("dark", chosen_font_name, 10)
|
|
if chosen_font_file:
|
|
pyglet_font.add_file(chosen_font_file)
|
|
self.font_set = Font(family=chosen_font_name, size=FONT_SIZE_F2)
|
|
self.font_entry = Font(family=chosen_font_name, size=FONT_SIZE_F2)
|
|
else:
|
|
pyglet_font.add_file(FONT_MAPPER[MAIN_FONT_NAME])
|
|
pyglet_font.add_file(FONT_MAPPER[SEC_FONT_NAME])
|
|
gui_data.sv_ttk.set_theme("dark", MAIN_FONT_NAME, 10)
|
|
self.font_set = Font(family=SEC_FONT_NAME, size=FONT_SIZE_F2)
|
|
self.font_entry = Font(family=MAIN_FONT_NAME, size=FONT_SIZE_F2)
|
|
|
|
def process_iteration(self):
|
|
self.iteration = self.iteration + 1
|
|
|
|
def assemble_model_data(self, model=None, arch_type=ENSEMBLE_MODE, is_dry_check=False, is_change_def=False, is_get_hash_dir_only=False):
|
|
|
|
if arch_type == ENSEMBLE_STEM_CHECK:
|
|
|
|
model_data = self.model_data_table
|
|
missing_models = [model.model_status for model in model_data if not model.model_status]
|
|
|
|
if missing_models or not model_data:
|
|
model_data: List[ModelData] = [ModelData(model_name, is_dry_check=is_dry_check) for model_name in self.ensemble_model_list]
|
|
self.model_data_table = model_data
|
|
|
|
if arch_type == KARAOKEE_CHECK:
|
|
model_list = []
|
|
model_data: List[ModelData] = [ModelData(model_name, is_dry_check=is_dry_check) for model_name in self.default_change_model_list]
|
|
for model in model_data:
|
|
if model.model_status and model.is_karaoke or model.is_bv_model:
|
|
model_list.append(model.model_and_process_tag)
|
|
|
|
return model_list
|
|
|
|
if arch_type == ENSEMBLE_MODE:
|
|
model_data: List[ModelData] = [ModelData(model_name) for model_name in self.ensemble_listbox_get_all_selected_models()]
|
|
if arch_type == ENSEMBLE_CHECK:
|
|
model_data: List[ModelData] = [ModelData(model, is_change_def=is_change_def, is_get_hash_dir_only=is_get_hash_dir_only)]
|
|
if arch_type == VR_ARCH_TYPE or arch_type == VR_ARCH_PM:
|
|
model_data: List[ModelData] = [ModelData(model, VR_ARCH_TYPE)]
|
|
if arch_type == MDX_ARCH_TYPE:
|
|
model_data: List[ModelData] = [ModelData(model, MDX_ARCH_TYPE)]
|
|
if arch_type == DEMUCS_ARCH_TYPE:
|
|
model_data: List[ModelData] = [ModelData(model, DEMUCS_ARCH_TYPE)]#
|
|
|
|
return model_data
|
|
|
|
def clear_cache(self, network):
|
|
|
|
if network == VR_ARCH_TYPE:
|
|
dir = VR_HASH_DIR
|
|
if network == MDX_ARCH_TYPE:
|
|
dir = MDX_HASH_DIR
|
|
|
|
for filename in os.listdir(dir):
|
|
filepath = os.path.join(dir, filename)
|
|
if filename not in ['model_data.json', 'model_name_mapper.json', 'mdx_c_configs'] and not os.path.isdir(filepath):
|
|
os.remove(filepath)
|
|
|
|
self.vr_model_var.set(CHOOSE_MODEL)
|
|
self.mdx_net_model_var.set(CHOOSE_MODEL)
|
|
self.model_data_table.clear()
|
|
self.chosen_ensemble_var.set(CHOOSE_ENSEMBLE_OPTION)
|
|
self.ensemble_main_stem_var.set(CHOOSE_STEM_PAIR)
|
|
self.ensemble_listbox_Option.configure(state=tk.DISABLED)
|
|
self.update_checkbox_text()
|
|
|
|
def thread_check(self, thread_to_check):
|
|
'''Checks if thread is alive'''
|
|
|
|
is_running = False
|
|
|
|
if type(thread_to_check) is KThread:
|
|
if thread_to_check.is_alive():
|
|
is_running = True
|
|
|
|
return is_running
|
|
|
|
# -Widget Methods--
|
|
|
|
def fill_main_frame(self):
|
|
"""Creates root window widgets"""
|
|
|
|
self.title_Label = tk.Label(master=self, image=self.logo_img, compound=tk.TOP)
|
|
self.title_Label.place(x=-2, y=banner_placement)
|
|
|
|
self.fill_filePaths_Frame()
|
|
self.fill_options_Frame()
|
|
|
|
self.conversion_Button = ttk.Button(master=self, textvariable=self.conversion_Button_Text_var, command=self.process_initialize)
|
|
self.conversion_Button.place(x=X_CONVERSION_BUTTON_1080P, y=BUTTON_Y_1080P, width=WIDTH_CONVERSION_BUTTON_1080P, height=HEIGHT_GENERIC_BUTTON_1080P,
|
|
relx=0, rely=0, relwidth=1, relheight=0)
|
|
|
|
self.conversion_Button_enable = lambda:(self.conversion_Button_Text_var.set(START_PROCESSING), self.conversion_Button.configure(state=tk.NORMAL))
|
|
self.conversion_Button_disable = lambda message:(self.conversion_Button_Text_var.set(message), self.conversion_Button.configure(state=tk.DISABLED))
|
|
|
|
self.stop_Button = ttk.Button(master=self, image=self.stop_img, command=self.confirm_stop_process)
|
|
self.stop_Button.place(x=X_STOP_BUTTON_1080P, y=BUTTON_Y_1080P, width=HEIGHT_GENERIC_BUTTON_1080P, height=HEIGHT_GENERIC_BUTTON_1080P,
|
|
relx=1, rely=0, relwidth=0, relheight=0)
|
|
self.help_hints(self.stop_Button, text=STOP_HELP)
|
|
|
|
self.settings_Button = ttk.Button(master=self, image=self.help_img, command=self.check_is_menu_settings_open)
|
|
self.settings_Button.place(x=X_SETTINGS_BUTTON_1080P, y=BUTTON_Y_1080P, width=HEIGHT_GENERIC_BUTTON_1080P, height=HEIGHT_GENERIC_BUTTON_1080P,
|
|
relx=1, rely=0, relwidth=0, relheight=0)
|
|
self.help_hints(self.settings_Button, text=SETTINGS_HELP)
|
|
|
|
self.progressbar = ttk.Progressbar(master=self, variable=self.progress_bar_main_var)
|
|
self.progressbar.place(x=X_PROGRESSBAR_1080P, y=Y_OFFSET_PROGRESS_BAR_1080P, width=WIDTH_PROGRESSBAR_1080P, height=HEIGHT_PROGRESSBAR_1080P,
|
|
relx=0, rely=0, relwidth=1, relheight=0)
|
|
|
|
# Select Music Files Option
|
|
self.console_Frame = tk.Frame(master=self, highlightbackground='#101012', highlightcolor='#101012', highlightthicknes=2)
|
|
self.console_Frame.place(x=15, y=self.IMAGE_HEIGHT + self.FILEPATHS_HEIGHT + self.OPTIONS_HEIGHT + self.CONVERSIONBUTTON_HEIGHT + self.PADDING + 5 *3, width=-30, height=self.COMMAND_HEIGHT+7,
|
|
relx=0, rely=0, relwidth=1, relheight=0)
|
|
|
|
|
|
self.command_Text = ThreadSafeConsole(master=self.console_Frame, background='#0c0c0d',fg='#898b8e', highlightcolor="#0c0c0d", font=(MAIN_FONT_NAME, FONT_SIZE_4), borderwidth=0)
|
|
self.command_Text.pack(fill=tk.BOTH, expand=1)
|
|
self.command_Text.bind(right_click_button, lambda e:self.right_click_console(e))
|
|
|
|
def fill_filePaths_Frame(self):
|
|
"""Fill Frame with neccessary widgets"""
|
|
|
|
# Select Music Files Option
|
|
self.filePaths_Frame = ttk.Frame(master=self)
|
|
self.filePaths_Frame.place(x=FILEPATHS_FRAME_X, y=FILEPATHS_FRAME_Y, width=FILEPATHS_FRAME_WIDTH, height=self.FILEPATHS_HEIGHT, relx=0, rely=0, relwidth=1, relheight=0)
|
|
|
|
self.filePaths_musicFile_Button = ttk.Button(master=self.filePaths_Frame, text=SELECT_INPUT_TEXT, command=self.input_select_filedialog)
|
|
self.filePaths_musicFile_Button.place(x=MUSICFILE_BUTTON_X, y=MUSICFILE_BUTTON_Y, width=MUSICFILE_BUTTON_WIDTH, height=MUSICFILE_BUTTON_HEIGHT, relx=0, rely=0, relwidth=0.3, relheight=0.5)
|
|
self.filePaths_musicFile_Entry = ttk.Entry(master=self.filePaths_Frame, textvariable=self.inputPathsEntry_var, font=self.font_entry, state=tk.DISABLED)
|
|
self.filePaths_musicFile_Entry.place(x=MUSICFILE_ENTRY_X, y=MUSICFILE_BUTTON_Y, width=MUSICFILE_ENTRY_WIDTH, height=MUSICFILE_ENTRY_HEIGHT, relx=0.3, rely=0, relwidth=0.7, relheight=0.5)
|
|
self.filePaths_musicFile_Open = ttk.Button(master=self.filePaths_Frame, image=self.efile_img, command=lambda:OPEN_FILE_func(os.path.dirname(self.inputPaths[0])) if self.inputPaths and os.path.isdir(os.path.dirname(self.inputPaths[0])) else self.error_dialoge(INVALID_INPUT))
|
|
self.filePaths_musicFile_Open.place(x=OPEN_BUTTON_X, y=MUSICFILE_BUTTON_Y, width=OPEN_BUTTON_WIDTH, height=MUSICFILE_ENTRY_HEIGHT, relx=0.3, rely=0, relwidth=0.7, relheight=0.5)
|
|
|
|
# Add any additional configurations or method calls here
|
|
self.filePaths_musicFile_Entry.configure(cursor="hand2")
|
|
self.help_hints(self.filePaths_musicFile_Button, text=INPUT_FOLDER_ENTRY_HELP)
|
|
self.help_hints(self.filePaths_musicFile_Entry, text=INPUT_FOLDER_ENTRY_HELP_2)
|
|
self.help_hints(self.filePaths_musicFile_Open, text=INPUT_FOLDER_BUTTON_HELP)
|
|
|
|
# Save To Option
|
|
self.filePaths_saveTo_Button = ttk.Button(master=self.filePaths_Frame, text=SELECT_OUTPUT_TEXT, command=self.export_select_filedialog)
|
|
self.filePaths_saveTo_Button.place(x=SAVETO_BUTTON_X, y=SAVETO_BUTTON_Y, width=SAVETO_BUTTON_WIDTH, height=SAVETO_BUTTON_HEIGHT, relx=0, rely=0.5, relwidth=0.3, relheight=0.5)
|
|
self.filePaths_saveTo_Entry = ttk.Entry(master=self.filePaths_Frame, textvariable=self.export_path_var, font=self.font_entry, state=tk.DISABLED)
|
|
self.filePaths_saveTo_Entry.place(x=SAVETO_ENTRY_X, y=SAVETO_BUTTON_Y, width=SAVETO_ENTRY_WIDTH, height=SAVETO_ENTRY_HEIGHT, relx=0.3, rely=0.5, relwidth=0.7, relheight=0.5)
|
|
self.filePaths_saveTo_Open = ttk.Button(master=self.filePaths_Frame, image=self.efile_img, command=lambda:OPEN_FILE_func(Path(self.export_path_var.get())) if os.path.isdir(self.export_path_var.get()) else self.error_dialoge(INVALID_EXPORT))
|
|
self.filePaths_saveTo_Open.place(x=OPEN_BUTTON_X, y=SAVETO_BUTTON_Y, width=OPEN_BUTTON_WIDTH, height=SAVETO_ENTRY_HEIGHT, relx=0.3, rely=0.5, relwidth=0.7, relheight=0.5)
|
|
self.help_hints(self.filePaths_saveTo_Button, text=OUTPUT_FOLDER_ENTRY_HELP)
|
|
self.help_hints(self.filePaths_saveTo_Open, text=OUTPUT_FOLDER_BUTTON_HELP)
|
|
|
|
def fill_options_Frame(self):
|
|
"""Fill Frame with neccessary widgets"""
|
|
|
|
self.options_Frame = ttk.Frame(master=self)
|
|
self.options_Frame.place(x=OPTIONS_FRAME_X, y=OPTIONS_FRAME_Y, width=OPTIONS_FRAME_WIDTH, height=self.OPTIONS_HEIGHT, relx=0, rely=0, relwidth=1, relheight=0)
|
|
|
|
# -Create Widgets-
|
|
|
|
## Save Format
|
|
self.wav_button = ttk.Radiobutton(master=self.options_Frame, text=WAV, variable=self.save_format_var, value=WAV)
|
|
self.wav_button.place(x=RADIOBUTTON_X_WAV, y=RADIOBUTTON_Y, width=RADIOBUTTON_WIDTH, height=RADIOBUTTON_HEIGHT, relx=0, rely=0/self.COL2_ROWS, relwidth=1/3, relheight=1/self.COL2_ROWS)
|
|
self.help_hints(self.wav_button, text=f'{FORMAT_SETTING_HELP}{WAV}')
|
|
|
|
self.flac_button = ttk.Radiobutton(master=self.options_Frame, text=FLAC, variable=self.save_format_var, value=FLAC)
|
|
self.flac_button.place(x=RADIOBUTTON_X_FLAC, y=RADIOBUTTON_Y, width=RADIOBUTTON_WIDTH, height=RADIOBUTTON_HEIGHT, relx=1/3, rely=0/self.COL2_ROWS, relwidth=1/3, relheight=1/self.COL2_ROWS)
|
|
self.help_hints(self.flac_button, text=f'{FORMAT_SETTING_HELP}{FLAC}')
|
|
|
|
self.mp3_button = ttk.Radiobutton(master=self.options_Frame, text=MP3, variable=self.save_format_var, value=MP3)
|
|
self.mp3_button.place(x=RADIOBUTTON_X_MP3, y=RADIOBUTTON_Y, width=RADIOBUTTON_WIDTH, height=RADIOBUTTON_HEIGHT, relx=2/3, rely=0/self.COL2_ROWS, relwidth=1/3, relheight=1/self.COL2_ROWS)
|
|
self.help_hints(self.mp3_button, text=f'{FORMAT_SETTING_HELP}{MP3}')
|
|
|
|
# Choose Conversion Method
|
|
self.chosen_process_method_Label = self.main_window_LABEL_SET(self.options_Frame, CHOOSE_PROC_METHOD_MAIN_LABEL)
|
|
self.chosen_process_method_Label.place(x=0, y=MAIN_ROW_Y[0], width=LEFT_ROW_WIDTH, height=LABEL_HEIGHT, relx=0, rely=2/self.COL1_ROWS, relwidth=1/3, relheight=1/self.COL1_ROWS)
|
|
self.chosen_process_method_Option = ComboBoxMenu(self.options_Frame, textvariable=self.chosen_process_method_var, values=PROCESS_METHODS, command=lambda e: self.selection_action_process_method(self.chosen_process_method_var.get(), from_widget=True, is_from_conv_menu=True))
|
|
self.chosen_process_method_Option.place(x=0, y=MAIN_ROW_Y[1], width=LEFT_ROW_WIDTH, height=OPTION_HEIGHT, relx=0, rely=3/self.COL1_ROWS, relwidth=1/3, relheight=1/self.COL1_ROWS)
|
|
#self.chosen_process_method_var.trace_add('write', lambda *args: self.update_main_widget_states())
|
|
self.help_hints(self.chosen_process_method_Label, text=CHOSEN_PROCESS_METHOD_HELP)
|
|
|
|
# Choose Settings Option
|
|
self.save_current_settings_Label = self.main_window_LABEL_SET(self.options_Frame, SELECT_SAVED_SETTINGS_MAIN_LABEL)
|
|
self.save_current_settings_Label_place = lambda:self.save_current_settings_Label.place(x=MAIN_ROW_2_X[0], y=LOW_MENU_Y[0], width=0, height=LABEL_HEIGHT, relx=2/3, rely=6/self.COL1_ROWS, relwidth=1/3, relheight=1/self.COL1_ROWS)
|
|
self.save_current_settings_Option = ComboBoxMenu(self.options_Frame, textvariable=self.save_current_settings_var, command=lambda e:self.selection_action_saved_settings(self.save_current_settings_var.get()))
|
|
self.save_current_settings_Option_place = lambda:self.save_current_settings_Option.place(x=MAIN_ROW_2_X[1], y=LOW_MENU_Y[1], width=MAIN_ROW_WIDTH, height=OPTION_HEIGHT, relx=2/3, rely=7/self.COL1_ROWS, relwidth=1/3, relheight=1/self.COL1_ROWS)
|
|
self.help_hints(self.save_current_settings_Label, text=SAVE_CURRENT_SETTINGS_HELP)
|
|
|
|
### MDX-NET ###
|
|
|
|
# Choose MDX-Net Model
|
|
self.mdx_net_model_Label = self.main_window_LABEL_SET(self.options_Frame, CHOOSE_MDX_MODEL_MAIN_LABEL)
|
|
self.mdx_net_model_Label_place = lambda:self.mdx_net_model_Label.place(x=0, y=LOW_MENU_Y[0], width=LEFT_ROW_WIDTH, height=LABEL_HEIGHT, relx=0, rely=6/self.COL1_ROWS, relwidth=1/3, relheight=1/self.COL1_ROWS)
|
|
self.mdx_net_model_Option = ComboBoxMenu(self.options_Frame, textvariable=self.mdx_net_model_var, command=lambda event: self.selection_action(event, self.mdx_net_model_var, is_mdx_net=True))
|
|
self.mdx_net_model_Option_place = lambda:self.mdx_net_model_Option.place(x=0, y=LOW_MENU_Y[1], width=LEFT_ROW_WIDTH, height=OPTION_HEIGHT, relx=0, rely=7/self.COL1_ROWS, relwidth=1/3, relheight=1/self.COL1_ROWS)
|
|
#self.mdx_net_model_var.trace_add('write', lambda *args: self.update_main_widget_states_mdx())
|
|
self.help_hints(self.mdx_net_model_Label, text=CHOOSE_MODEL_HELP)
|
|
|
|
# MDX-Overlap
|
|
self.overlap_mdx_Label = self.main_window_LABEL_SET(self.options_Frame, 'OVERLAP')
|
|
self.overlap_mdx_Label_place = lambda:self.overlap_mdx_Label.place(x=MAIN_ROW_2_X[0], y=MAIN_ROW_2_Y[0], width=0, height=LABEL_HEIGHT, relx=2/3, rely=2/self.COL1_ROWS, relwidth=1/3, relheight=1/self.COL2_ROWS)
|
|
self.overlap_mdx_Option = ComboBoxEditableMenu(self.options_Frame, values=MDX_OVERLAP, width=MENU_COMBOBOX_WIDTH, textvariable=self.overlap_mdx_var, pattern=REG_OVERLAP, default=MDX_OVERLAP)
|
|
self.overlap_mdx_Option_place = lambda:self.overlap_mdx_Option.place(x=MAIN_ROW_2_X[1], y=MAIN_ROW_2_Y[1], width=MAIN_ROW_WIDTH, height=OPTION_HEIGHT, relx=2/3, rely=3/self.COL1_ROWS, relwidth=1/3, relheight=1/self.COL2_ROWS)
|
|
|
|
# MDX23-Overlap
|
|
self.overlap_mdx23_Option = ComboBoxEditableMenu(self.options_Frame, values=MDX23_OVERLAP, width=MENU_COMBOBOX_WIDTH, textvariable=self.overlap_mdx23_var, pattern=REG_OVERLAP23, default="8")
|
|
self.overlap_mdx23_Option_place = lambda:self.overlap_mdx23_Option.place(x=MAIN_ROW_2_X[1], y=MAIN_ROW_2_Y[1], width=MAIN_ROW_WIDTH, height=OPTION_HEIGHT, relx=2/3, rely=3/self.COL1_ROWS, relwidth=1/3, relheight=1/self.COL2_ROWS)
|
|
self.help_hints(self.overlap_mdx_Label, text=MDX_OVERLAP_HELP)
|
|
|
|
# Choose MDX-Net Stems
|
|
self.mdxnet_stems_Label = self.main_window_LABEL_SET(self.options_Frame, CHOOSE_STEMS_MAIN_LABEL)
|
|
self.mdxnet_stems_Label_place = lambda:self.mdxnet_stems_Label.place(x=MAIN_ROW_X[0], y=MAIN_ROW_Y[0], width=0, height=LABEL_HEIGHT, relx=1/3, rely=2/self.COL2_ROWS, relwidth=1/3, relheight=1/self.COL2_ROWS)
|
|
|
|
self.mdxnet_stems_Option = ComboBoxMenu(self.options_Frame, textvariable=self.mdxnet_stems_var)
|
|
self.mdxnet_stems_Option_place = lambda:self.mdxnet_stems_Option.place(x=MAIN_ROW_X[1], y=MAIN_ROW_Y[1], width=MAIN_ROW_WIDTH, height=OPTION_HEIGHT, relx=1/3, rely=3/self.COL2_ROWS, relwidth=1/3, relheight=1/self.COL2_ROWS)
|
|
self.help_hints(self.mdxnet_stems_Label, text=DEMUCS_STEMS_HELP)
|
|
|
|
# MDX-Segment Size
|
|
self.mdx_segment_size_Label = self.main_window_LABEL_SET(self.options_Frame, SEGMENT_MDX_MAIN_LABEL)
|
|
self.mdx_segment_size_Label_place = lambda:self.mdx_segment_size_Label.place(x=MAIN_ROW_X[0], y=MAIN_ROW_Y[0], width=0, height=LABEL_HEIGHT, relx=1/3, rely=2/self.COL1_ROWS, relwidth=1/3, relheight=1/self.COL2_ROWS)
|
|
self.mdx_segment_size_Option = ComboBoxEditableMenu(self.options_Frame, values=MDX_SEGMENTS, width=MENU_COMBOBOX_WIDTH, textvariable=self.mdx_segment_size_var, pattern=REG_MDX_SEG, default="256")#
|
|
self.mdx_segment_size_Option_place = lambda:self.mdx_segment_size_Option.place(x=MAIN_ROW_X[1], y=MAIN_ROW_Y[1], width=MAIN_ROW_WIDTH, height=OPTION_HEIGHT, relx=1/3, rely=3/self.COL1_ROWS, relwidth=1/3, relheight=1/self.COL2_ROWS)
|
|
self.help_hints(self.mdx_segment_size_Label, text=MDX_SEGMENT_SIZE_HELP)
|
|
|
|
### VR ARCH ###
|
|
|
|
# Choose VR Model
|
|
self.vr_model_Label = self.main_window_LABEL_SET(self.options_Frame, SELECT_VR_MODEL_MAIN_LABEL)
|
|
self.vr_model_Label_place = lambda:self.vr_model_Label.place(x=0, y=LOW_MENU_Y[0], width=LEFT_ROW_WIDTH, height=LABEL_HEIGHT, relx=0, rely=6/self.COL1_ROWS, relwidth=1/3, relheight=1/self.COL1_ROWS)
|
|
self.vr_model_Option = ComboBoxMenu(self.options_Frame, textvariable=self.vr_model_var, command=lambda event: self.selection_action(event, self.vr_model_var))
|
|
self.vr_model_Option_place = lambda:self.vr_model_Option.place(x=0, y=LOW_MENU_Y[1], width=LEFT_ROW_WIDTH, height=OPTION_HEIGHT, relx=0, rely=7/self.COL1_ROWS, relwidth=1/3, relheight=1/self.COL1_ROWS)
|
|
self.help_hints(self.vr_model_Label, text=CHOOSE_MODEL_HELP)
|
|
|
|
# Aggression Setting
|
|
self.aggression_setting_Label = self.main_window_LABEL_SET(self.options_Frame, AGGRESSION_SETTING_MAIN_LABEL)
|
|
self.aggression_setting_Label_place = lambda:self.aggression_setting_Label.place(x=MAIN_ROW_2_X[0], y=MAIN_ROW_2_Y[0], width=0, height=LABEL_HEIGHT, relx=2/3, rely=2/self.COL2_ROWS, relwidth=1/3, relheight=1/self.COL2_ROWS)
|
|
self.aggression_setting_Option = ComboBoxEditableMenu(self.options_Frame, values=VR_AGGRESSION, textvariable=self.aggression_setting_var, pattern=REG_AGGRESSION, default=VR_AGGRESSION[5])#
|
|
self.aggression_setting_Option_place = lambda:self.aggression_setting_Option.place(x=MAIN_ROW_2_X[1], y=MAIN_ROW_2_Y[1], width=MAIN_ROW_WIDTH, height=OPTION_HEIGHT, relx=2/3, rely=3/self.COL2_ROWS, relwidth=1/3, relheight=1/self.COL2_ROWS)
|
|
self.help_hints(self.aggression_setting_Label, text=AGGRESSION_SETTING_HELP)
|
|
|
|
# Window Size
|
|
self.window_size_Label = self.main_window_LABEL_SET(self.options_Frame, WINDOW_SIZE_MAIN_LABEL)
|
|
self.window_size_Label_place = lambda:self.window_size_Label.place(x=MAIN_ROW_X[0], y=MAIN_ROW_Y[0], width=0, height=LABEL_HEIGHT, relx=1/3, rely=2/self.COL2_ROWS, relwidth=1/3, relheight=1/self.COL2_ROWS)
|
|
self.window_size_Option = ComboBoxEditableMenu(self.options_Frame, values=VR_WINDOW, textvariable=self.window_size_var, pattern=REG_WINDOW, default=VR_WINDOW[1])#
|
|
self.window_size_Option_place = lambda:self.window_size_Option.place(x=MAIN_ROW_X[1], y=MAIN_ROW_Y[1], width=MAIN_ROW_WIDTH, height=OPTION_HEIGHT, relx=1/3, rely=3/self.COL2_ROWS, relwidth=1/3, relheight=1/self.COL2_ROWS)
|
|
self.help_hints(self.window_size_Label, text=WINDOW_SIZE_HELP)
|
|
|
|
### DEMUCS ###
|
|
|
|
# Choose Demucs Models
|
|
self.demucs_model_Label = self.main_window_LABEL_SET(self.options_Frame, CHOOSE_DEMUCS_MODEL_MAIN_LABEL)
|
|
self.demucs_model_Label_place = lambda:self.demucs_model_Label.place(x=0, y=LOW_MENU_Y[0], width=LEFT_ROW_WIDTH, height=LABEL_HEIGHT, relx=0, rely=6/self.COL1_ROWS, relwidth=1/3, relheight=1/self.COL1_ROWS)
|
|
self.demucs_model_Option = ComboBoxMenu(self.options_Frame, textvariable=self.demucs_model_var, command=lambda event: self.selection_action(event, self.demucs_model_var))
|
|
self.demucs_model_Option_place = lambda:self.demucs_model_Option.place(x=0, y=LOW_MENU_Y[1], width=LEFT_ROW_WIDTH, height=OPTION_HEIGHT, relx=0, rely=7/self.COL1_ROWS, relwidth=1/3, relheight=1/self.COL1_ROWS)
|
|
self.help_hints(self.demucs_model_Label, text=CHOOSE_MODEL_HELP)
|
|
|
|
# Choose Demucs Stems
|
|
self.demucs_stems_Label = self.main_window_LABEL_SET(self.options_Frame, CHOOSE_STEMS_MAIN_LABEL)
|
|
self.demucs_stems_Label_place = lambda:self.demucs_stems_Label.place(x=MAIN_ROW_X[0], y=MAIN_ROW_Y[0], width=0, height=LABEL_HEIGHT, relx=1/3, rely=2/self.COL2_ROWS, relwidth=1/3, relheight=1/self.COL2_ROWS)
|
|
self.demucs_stems_Option = ComboBoxMenu(self.options_Frame, textvariable=self.demucs_stems_var)
|
|
self.demucs_stems_Option_place = lambda:self.demucs_stems_Option.place(x=MAIN_ROW_X[1], y=MAIN_ROW_Y[1], width=MAIN_ROW_WIDTH, height=OPTION_HEIGHT, relx=1/3, rely=3/self.COL2_ROWS, relwidth=1/3, relheight=1/self.COL2_ROWS)
|
|
self.help_hints(self.demucs_stems_Label, text=DEMUCS_STEMS_HELP)
|
|
|
|
# Demucs-Segment
|
|
self.segment_Label = self.main_window_LABEL_SET(self.options_Frame, CHOOSE_SEGMENT_MAIN_LABEL)
|
|
self.segment_Label_place = lambda:self.segment_Label.place(x=MAIN_ROW_2_X[0], y=MAIN_ROW_2_Y[0], width=0, height=LABEL_HEIGHT, relx=2/3, rely=2/self.COL2_ROWS, relwidth=1/3, relheight=1/self.COL2_ROWS)
|
|
self.segment_Option = ComboBoxEditableMenu(self.options_Frame, values=DEMUCS_SEGMENTS, textvariable=self.segment_var, pattern=REG_SEGMENTS, default=DEMUCS_SEGMENTS)#
|
|
self.segment_Option_place = lambda:self.segment_Option.place(x=MAIN_ROW_2_X[1], y=MAIN_ROW_2_Y[1], width=MAIN_ROW_WIDTH, height=OPTION_HEIGHT, relx=2/3, rely=3/self.COL2_ROWS, relwidth=1/3, relheight=1/self.COL2_ROWS)
|
|
self.help_hints(self.segment_Label, text=SEGMENT_HELP)
|
|
|
|
# Stem A
|
|
self.is_primary_stem_only_Demucs_Option = ttk.Checkbutton(master=self.options_Frame, textvariable=self.is_primary_stem_only_Demucs_Text_var, variable=self.is_primary_stem_only_Demucs_var, command=lambda:self.is_primary_stem_only_Demucs_Option_toggle())
|
|
self.is_primary_stem_only_Demucs_Option_place = lambda:self.is_primary_stem_only_Demucs_Option.place(x=CHECK_BOX_X, y=CHECK_BOX_Y, width=CHECK_BOX_WIDTH, height=CHECK_BOX_HEIGHT, relx=1/3, rely=6/self.COL2_ROWS, relwidth=1/3, relheight=1/self.COL2_ROWS)
|
|
self.is_primary_stem_only_Demucs_Option_toggle = lambda:self.is_secondary_stem_only_Demucs_var.set(False) if self.is_primary_stem_only_Demucs_var.get() else self.is_secondary_stem_only_Demucs_Option.configure(state=tk.NORMAL)
|
|
self.help_hints(self.is_primary_stem_only_Demucs_Option, text=SAVE_STEM_ONLY_HELP)
|
|
|
|
# Stem B
|
|
self.is_secondary_stem_only_Demucs_Option = ttk.Checkbutton(master=self.options_Frame, textvariable=self.is_secondary_stem_only_Demucs_Text_var, variable=self.is_secondary_stem_only_Demucs_var, command=lambda:self.is_secondary_stem_only_Demucs_Option_toggle())
|
|
self.is_secondary_stem_only_Demucs_Option_place = lambda:self.is_secondary_stem_only_Demucs_Option.place(x=CHECK_BOX_X, y=CHECK_BOX_Y, width=CHECK_BOX_WIDTH, height=CHECK_BOX_HEIGHT, relx=1/3, rely=7/self.COL2_ROWS, relwidth=1/3, relheight=1/self.COL2_ROWS)
|
|
self.is_secondary_stem_only_Demucs_Option_toggle = lambda:self.is_primary_stem_only_Demucs_var.set(False) if self.is_secondary_stem_only_Demucs_var.get() else self.is_primary_stem_only_Demucs_Option.configure(state=tk.NORMAL)
|
|
self.is_stem_only_Demucs_Options_Enable = lambda:(self.is_primary_stem_only_Demucs_Option.configure(state=tk.NORMAL), self.is_secondary_stem_only_Demucs_Option.configure(state=tk.NORMAL))
|
|
self.help_hints(self.is_secondary_stem_only_Demucs_Option, text=SAVE_STEM_ONLY_HELP)
|
|
|
|
### ENSEMBLE MODE ###
|
|
|
|
# Ensemble Mode
|
|
self.chosen_ensemble_Label = self.main_window_LABEL_SET(self.options_Frame, ENSEMBLE_OPTIONS_MAIN_LABEL)
|
|
self.chosen_ensemble_Label_place = lambda:self.chosen_ensemble_Label.place(x=0, y=LOW_MENU_Y[0], width=LEFT_ROW_WIDTH, height=LABEL_HEIGHT, relx=0, rely=6/self.COL1_ROWS, relwidth=1/3, relheight=1/self.COL1_ROWS)
|
|
self.chosen_ensemble_Option = ComboBoxMenu(self.options_Frame, textvariable=self.chosen_ensemble_var, command=lambda e:self.selection_action_chosen_ensemble(self.chosen_ensemble_var.get()))
|
|
self.chosen_ensemble_Option_place = lambda:self.chosen_ensemble_Option.place(x=0, y=LOW_MENU_Y[1], width=LEFT_ROW_WIDTH, height=OPTION_HEIGHT, relx=0, rely=7/self.COL1_ROWS, relwidth=1/3, relheight=1/self.COL1_ROWS)
|
|
self.help_hints(self.chosen_ensemble_Label, text=CHOSEN_ENSEMBLE_HELP)
|
|
|
|
# Ensemble Main Stems
|
|
self.ensemble_main_stem_Label = self.main_window_LABEL_SET(self.options_Frame, CHOOSE_MAIN_PAIR_MAIN_LABEL)
|
|
self.ensemble_main_stem_Label_place = lambda:self.ensemble_main_stem_Label.place(x=MAIN_ROW_X[0], y=MAIN_ROW_Y[0], width=0, height=LABEL_HEIGHT, relx=1/3, rely=2/self.COL1_ROWS, relwidth=1/3, relheight=1/self.COL1_ROWS)
|
|
self.ensemble_main_stem_Option = ComboBoxMenu(self.options_Frame, textvariable=self.ensemble_main_stem_var, values=ENSEMBLE_MAIN_STEM, command=lambda e: self.selection_action_ensemble_stems(self.ensemble_main_stem_var.get()))
|
|
self.ensemble_main_stem_Option_place = lambda:self.ensemble_main_stem_Option.place(x=MAIN_ROW_X[1], y=MAIN_ROW_Y[1], width=MAIN_ROW_WIDTH, height=OPTION_HEIGHT, relx=1/3, rely=3/self.COL1_ROWS, relwidth=1/3, relheight=1/self.COL1_ROWS)
|
|
self.help_hints(self.ensemble_main_stem_Label, text=ENSEMBLE_MAIN_STEM_HELP)
|
|
|
|
# Ensemble Algorithm
|
|
self.ensemble_type_Label = self.main_window_LABEL_SET(self.options_Frame, CHOOSE_ENSEMBLE_ALGORITHM_MAIN_LABEL)
|
|
self.ensemble_type_Label_place = lambda:self.ensemble_type_Label.place(x=MAIN_ROW_2_X[0], y=MAIN_ROW_2_Y[0], width=0, height=LABEL_HEIGHT, relx=2/3, rely=2/11, relwidth=1/3, relheight=1/self.COL1_ROWS)
|
|
self.ensemble_type_Option = ComboBoxMenu(self.options_Frame, textvariable=self.ensemble_type_var, values=ENSEMBLE_TYPE)
|
|
self.ensemble_type_Option_place = lambda:self.ensemble_type_Option.place(x=MAIN_ROW_2_X[1], y=MAIN_ROW_2_Y[1], width=MAIN_ROW_WIDTH, height=OPTION_HEIGHT,relx=2/3, rely=3/11, relwidth=1/3, relheight=1/self.COL1_ROWS)
|
|
self.help_hints(self.ensemble_type_Label, text=ENSEMBLE_TYPE_HELP)
|
|
|
|
# Select Music Files Option
|
|
|
|
# Ensemble Save Ensemble Outputs
|
|
self.ensemble_listbox_Label = self.main_window_LABEL_SET(self.options_Frame, AVAILABLE_MODELS_MAIN_LABEL)
|
|
self.ensemble_listbox_Label_place = lambda:self.ensemble_listbox_Label.place(x=MAIN_ROW_2_X[0], y=MAIN_ROW_2_Y[1], width=0, height=LABEL_HEIGHT, relx=2/3, rely=5/11, relwidth=1/3, relheight=1/self.COL1_ROWS)
|
|
self.ensemble_listbox_Frame = tk.Frame(self.options_Frame, highlightbackground='#04332c', highlightcolor='#04332c', highlightthicknes=1)
|
|
self.ensemble_listbox_Option = tk.Listbox(self.ensemble_listbox_Frame, selectmode=tk.MULTIPLE, activestyle='dotbox', font=(MAIN_FONT_NAME, f"{FONT_SIZE_1}"), background='#070708', exportselection=0, relief=tk.SOLID, borderwidth=0)
|
|
self.ensemble_listbox_scroll = ttk.Scrollbar(self.options_Frame, orient=tk.VERTICAL)
|
|
self.ensemble_listbox_Option.config(yscrollcommand=self.ensemble_listbox_scroll.set)
|
|
self.ensemble_listbox_scroll.configure(command=self.ensemble_listbox_Option.yview)
|
|
self.ensemble_listbox_Option_place = lambda: (self.ensemble_listbox_Frame.place(x=ENSEMBLE_LISTBOX_FRAME_X, y=ENSEMBLE_LISTBOX_FRAME_Y, width=ENSEMBLE_LISTBOX_FRAME_WIDTH, height=ENSEMBLE_LISTBOX_FRAME_HEIGHT, relx=2/3, rely=6/11, relwidth=1/3, relheight=1/self.COL1_ROWS),
|
|
self.ensemble_listbox_scroll.place(x=ENSEMBLE_LISTBOX_SCROLL_X, y=ENSEMBLE_LISTBOX_SCROLL_Y, width=ENSEMBLE_LISTBOX_SCROLL_WIDTH, height=ENSEMBLE_LISTBOX_SCROLL_HEIGHT, relx=2/3, rely=6/11, relwidth=1/10, relheight=1/self.COL1_ROWS))
|
|
self.ensemble_listbox_Option_pack = lambda:self.ensemble_listbox_Option.pack(fill=tk.BOTH, expand=1)
|
|
self.help_hints(self.ensemble_listbox_Label, text=ENSEMBLE_LISTBOX_HELP)
|
|
|
|
### AUDIO TOOLS ###
|
|
|
|
# Chosen Audio Tool
|
|
self.chosen_audio_tool_Label = self.main_window_LABEL_SET(self.options_Frame, CHOOSE_AUDIO_TOOLS_MAIN_LABEL)
|
|
self.chosen_audio_tool_Label_place = lambda:self.chosen_audio_tool_Label.place(x=0, y=LOW_MENU_Y[0], width=LEFT_ROW_WIDTH, height=LABEL_HEIGHT, relx=0, rely=6/self.COL1_ROWS, relwidth=1/3, relheight=1/self.COL1_ROWS)
|
|
self.chosen_audio_tool_Option = ComboBoxMenu(self.options_Frame, textvariable=self.chosen_audio_tool_var, values=AUDIO_TOOL_OPTIONS, command=lambda e: self.update_main_widget_states())
|
|
self.chosen_audio_tool_Option_place = lambda:self.chosen_audio_tool_Option.place(x=0, y=LOW_MENU_Y[1], width=LEFT_ROW_WIDTH, height=OPTION_HEIGHT, relx=0, rely=7/self.COL1_ROWS, relwidth=1/3, relheight=1/self.COL1_ROWS)
|
|
self.help_hints(self.chosen_audio_tool_Label, text=AUDIO_TOOLS_HELP)
|
|
|
|
# Choose Agorithim
|
|
self.choose_algorithm_Label = self.main_window_LABEL_SET(self.options_Frame, CHOOSE_MANUAL_ALGORITHM_MAIN_LABEL)
|
|
self.choose_algorithm_Label_place = lambda:self.choose_algorithm_Label.place(x=MAIN_ROW_X[0], y=MAIN_ROW_Y[0], width=0, height=LABEL_HEIGHT, relx=1/3, rely=2/self.COL1_ROWS, relwidth=1/3, relheight=1/self.COL2_ROWS)
|
|
self.choose_algorithm_Option = ComboBoxMenu(self.options_Frame, textvariable=self.choose_algorithm_var, values=MANUAL_ENSEMBLE_OPTIONS)
|
|
self.choose_algorithm_Option_place = lambda:self.choose_algorithm_Option.place(x=MAIN_ROW_X[1], y=MAIN_ROW_Y[1], width=MAIN_ROW_WIDTH, height=OPTION_HEIGHT, relx=1/3, rely=3/self.COL1_ROWS, relwidth=1/3, relheight=1/self.COL2_ROWS)
|
|
#self.help_hints(self.mdx_segment_size_Label, text=MDX_SEGMENT_SIZE_HELP)
|
|
|
|
|
|
# Time Stretch
|
|
self.time_stretch_rate_Label = self.main_window_LABEL_SET(self.options_Frame, CHOOSE_RATE_MAIN_LABEL)
|
|
self.time_stretch_rate_Label_place = lambda:self.time_stretch_rate_Label.place(x=MAIN_ROW_X[0], y=MAIN_ROW_Y[0], width=0, height=LABEL_HEIGHT, relx=1/3, rely=2/self.COL1_ROWS, relwidth=1/3, relheight=1/self.COL2_ROWS)
|
|
self.time_stretch_rate_Option = ComboBoxEditableMenu(self.options_Frame, values=TIME_PITCH, textvariable=self.time_stretch_rate_var, pattern=REG_TIME, default=TIME_PITCH)#
|
|
self.time_stretch_rate_Option_place = lambda:self.time_stretch_rate_Option.place(x=MAIN_ROW_X[1], y=MAIN_ROW_Y[1], width=MAIN_ROW_WIDTH, height=OPTION_HEIGHT, relx=1/3, rely=3/self.COL1_ROWS, relwidth=1/3, relheight=1/self.COL2_ROWS)
|
|
#self.help_hints(self.mdx_segment_size_Label, text=MDX_SEGMENT_SIZE_HELP)
|
|
|
|
# Pitch Rate
|
|
self.pitch_rate_Label = self.main_window_LABEL_SET(self.options_Frame, CHOOSE_SEMITONES_MAIN_LABEL)
|
|
self.pitch_rate_Label_place = lambda:self.pitch_rate_Label.place(x=MAIN_ROW_X[0], y=MAIN_ROW_Y[0], width=0, height=LABEL_HEIGHT, relx=1/3, rely=2/self.COL1_ROWS, relwidth=1/3, relheight=1/self.COL2_ROWS)
|
|
self.pitch_rate_Option = ComboBoxEditableMenu(self.options_Frame, values=TIME_PITCH, textvariable=self.pitch_rate_var, pattern=REG_PITCH, default=TIME_PITCH)#
|
|
self.pitch_rate_Option_place = lambda:self.pitch_rate_Option.place(x=MAIN_ROW_X[1], y=MAIN_ROW_Y[1], width=MAIN_ROW_WIDTH, height=OPTION_HEIGHT, relx=1/3, rely=3/self.COL1_ROWS, relwidth=1/3, relheight=1/self.COL2_ROWS)
|
|
|
|
# Is Time Correction
|
|
self.is_time_correction_Option = ttk.Checkbutton(master=self.options_Frame, text=TIME_CORRECTION_TEXT, variable=self.is_time_correction_var)
|
|
self.is_time_correction_Option_place = lambda:self.is_time_correction_Option.place(x=CHECK_BOX_X, y=CHECK_BOX_Y, width=CHECK_BOX_WIDTH, height=CHECK_BOX_HEIGHT, relx=1/3, rely=5/self.COL2_ROWS, relwidth=1/3, relheight=1/self.COL2_ROWS)
|
|
self.help_hints(self.is_time_correction_Option, text=IS_TIME_CORRECTION_HELP)
|
|
|
|
# Is Wav Ensemble
|
|
self.is_wav_ensemble_Option = ttk.Checkbutton(master=self.options_Frame, text=ENSEMBLE_WAVFORMS_TEXT, variable=self.is_wav_ensemble_var)
|
|
self.is_wav_ensemble_Option_place = lambda:self.is_wav_ensemble_Option.place(x=CHECK_BOX_X, y=CHECK_BOX_Y, width=CHECK_BOX_WIDTH, height=CHECK_BOX_HEIGHT, relx=1/3, rely=5/self.COL2_ROWS, relwidth=1/3, relheight=1/self.COL2_ROWS)
|
|
self.help_hints(self.is_wav_ensemble_Option, text=IS_WAV_ENSEMBLE_HELP)
|
|
|
|
## ALIGN TOOL ##
|
|
|
|
# Track 1
|
|
self.fileOne_Label = self.main_window_LABEL_SUB_SET(self.options_Frame, self.file_one_sub_var)
|
|
self.fileOne_Label_place = lambda: self.fileOne_Label.place(x=FILEONE_LABEL_X, y=LABEL_Y, width=FILEONE_LABEL_WIDTH, height=LABEL_HEIGHT, relx=1/3, rely=2/self.COL1_ROWS, relwidth=1/3, relheight=1/self.COL2_ROWS)
|
|
|
|
self.fileOne_Entry = ttk.Entry(master=self.options_Frame, textvariable=self.fileOneEntry_var, font=self.font_entry, state=tk.DISABLED)
|
|
self.fileOne_Entry_place = lambda: self.fileOne_Entry.place(x=SUB_ENT_ROW_X, y=ENTRY_Y, width=ENTRY_WIDTH, height=OPTION_HEIGHT, relx=1/3, rely=3/self.COL1_ROWS, relwidth=1/3, relheight=1/self.COL2_ROWS)
|
|
self.help_hints(self.fileOne_Entry, text=INPUT_SEC_FIELDS_HELP)
|
|
self.fileOne_Entry.configure(cursor="hand2")
|
|
|
|
self.fileOne_Open = ttk.Button(master=self.options_Frame, image=self.efile_img, command=lambda:OPEN_FILE_func(os.path.dirname(self.fileOneEntry_Full_var.get())))
|
|
self.fileOne_Open_place = lambda:self.fileOne_Open.place(x=ENTRY_OPEN_BUTT_X_OFF, y=ENTRY_Y, width=ENTRY_OPEN_BUTT_WIDTH, height=OPTION_HEIGHT, relx=1/3, rely=3/self.COL1_ROWS, relwidth=1/3, relheight=1/self.COL2_ROWS)#OPEN_FILE_func(Path(self.export_path_var.get())) if os.path.isdir(self.export_path_var.get()) else self.error_dialoge(INVALID_EXPORT))
|
|
self.help_hints(self.fileOne_Open, text=INPUT_FOLDER_BUTTON_HELP)
|
|
|
|
# Track 2
|
|
self.fileTwo_Label = self.main_window_LABEL_SUB_SET(self.options_Frame, self.file_two_sub_var)
|
|
self.fileTwo_Label_place = lambda: self.fileTwo_Label.place(x=FILETWO_LABEL_X, y=LABEL_Y, width=FILETWO_LABEL_WIDTH, height=LABEL_HEIGHT, relx=1/3, rely=4.5/self.COL1_ROWS, relwidth=1/3, relheight=1/self.COL2_ROWS)
|
|
|
|
self.fileTwo_Entry = ttk.Entry(master=self.options_Frame, textvariable=self.fileTwoEntry_var, font=self.font_entry, state=tk.DISABLED)
|
|
self.fileTwo_Entry_place = lambda:self.fileTwo_Entry.place(x=SUB_ENT_ROW_X, y=ENTRY_Y, width=ENTRY_WIDTH, height=OPTION_HEIGHT, relx=1/3, rely=5.5/self.COL1_ROWS, relwidth=1/3, relheight=1/self.COL2_ROWS)
|
|
self.help_hints(self.fileTwo_Entry, text=INPUT_SEC_FIELDS_HELP)
|
|
self.fileTwo_Entry.configure(cursor="hand2")
|
|
|
|
self.fileTwo_Open = ttk.Button(master=self.options_Frame, image=self.efile_img, command=lambda:OPEN_FILE_func(os.path.dirname(self.fileTwoEntry_Full_var.get())))
|
|
self.fileTwo_Open_place = lambda:self.fileTwo_Open.place(x=ENTRY_OPEN_BUTT_X_OFF, y=ENTRY_Y, width=ENTRY_OPEN_BUTT_WIDTH, height=OPTION_HEIGHT, relx=1/3, rely=5.5/self.COL1_ROWS, relwidth=1/3, relheight=1/self.COL2_ROWS)
|
|
self.help_hints(self.fileTwo_Open, text=INPUT_FOLDER_BUTTON_HELP)
|
|
|
|
# Time Window
|
|
self.time_window_Label = self.main_window_LABEL_SET(self.options_Frame, TIME_WINDOW_MAIN_LABEL)
|
|
self.time_window_Label_place = lambda: self.time_window_Label.place(x=TIME_WINDOW_LABEL_X, y=LABEL_Y, width=TIME_WINDOW_LABEL_WIDTH, height=LABEL_HEIGHT, relx=1/3, rely=7.37/self.COL1_ROWS, relwidth=1/3, relheight=1/self.COL2_ROWS)
|
|
self.time_window_Option = ComboBoxMenu(self.options_Frame, textvariable=self.time_window_var, values=tuple(TIME_WINDOW_MAPPER.keys()))
|
|
self.time_window_Option_place = lambda: self.time_window_Option.place(x=SUB_ENT_ROW_X, y=ENTRY_Y, width=OPTION_WIDTH, height=OPTION_HEIGHT, relx=1/3, rely=8.37/self.COL1_ROWS, relwidth=1/3, relheight=1/self.COL2_ROWS)
|
|
self.help_hints(self.time_window_Label, text=TIME_WINDOW_ALIGN_HELP)
|
|
|
|
# Align Shifts
|
|
self.intro_analysis_Label = self.main_window_LABEL_SET(self.options_Frame, INTRO_ANALYSIS_MAIN_LABEL)
|
|
self.intro_analysis_Label_place = lambda: self.intro_analysis_Label.place(x=INTRO_ANALYSIS_LABEL_X, y=LABEL_Y, width=INTRO_ANALYSIS_LABEL_WIDTH, height=LABEL_HEIGHT, relx=2/3, rely=7.37/self.COL1_ROWS, relwidth=1/3, relheight=1/self.COL2_ROWS)
|
|
self.intro_analysis_Option = ComboBoxMenu(self.options_Frame, textvariable=self.intro_analysis_var, values=tuple(INTRO_MAPPER.keys()))
|
|
self.intro_analysis_Option_place = lambda: self.intro_analysis_Option.place(x=INTRO_ANALYSIS_OPTION_X, y=ENTRY_Y, width=OPTION_WIDTH, height=OPTION_HEIGHT, relx=2/3, rely=8.37/self.COL1_ROWS, relwidth=1/3, relheight=1/self.COL2_ROWS)
|
|
self.help_hints(self.intro_analysis_Label, text=INTRO_ANALYSIS_ALIGN_HELP)
|
|
|
|
# Volume Adjustment
|
|
self.db_analysis_Label = self.main_window_LABEL_SET(self.options_Frame, VOLUME_ADJUSTMENT_MAIN_LABEL)
|
|
self.db_analysis_Label_place = lambda: self.db_analysis_Label.place(x=DB_ANALYSIS_LABEL_X, y=LABEL_Y, width=DB_ANALYSIS_LABEL_WIDTH, height=LABEL_HEIGHT, relx=2/3, rely=7.37/self.COL1_ROWS, relwidth=1/3, relheight=1/self.COL2_ROWS)
|
|
self.db_analysis_Option = ComboBoxMenu(self.options_Frame, textvariable=self.db_analysis_var, values=tuple(VOLUME_MAPPER.keys()))
|
|
self.db_analysis_Option_place = lambda: self.db_analysis_Option.place(x=DB_ANALYSIS_OPTION_X, y=ENTRY_Y, width=OPTION_WIDTH, height=OPTION_HEIGHT, relx=2/3, rely=8.37/self.COL1_ROWS, relwidth=1/3, relheight=1/self.COL2_ROWS)
|
|
self.help_hints(self.db_analysis_Label, text=VOLUME_ANALYSIS_ALIGN_HELP)
|
|
|
|
# Wav-Type
|
|
self.wav_type_set_Label = self.main_window_LABEL_SET(self.options_Frame, WAVE_TYPE_TEXT)
|
|
self.wav_type_set_Label_place = lambda: self.wav_type_set_Label.place(x=WAV_TYPE_SET_LABEL_X, y=LABEL_Y, width=WAV_TYPE_SET_LABEL_WIDTH, height=LABEL_HEIGHT, relx=1/3, rely=7.37/self.COL1_ROWS, relwidth=1/3, relheight=1/self.COL2_ROWS)
|
|
self.wav_type_set_Option = ComboBoxMenu(self.options_Frame, textvariable=self.wav_type_set_var, values=WAV_TYPE)
|
|
self.wav_type_set_Option_place = lambda: self.wav_type_set_Option.place(x=SUB_ENT_ROW_X, y=ENTRY_Y, width=OPTION_WIDTH, height=OPTION_HEIGHT, relx=1/3, rely=8.37/self.COL1_ROWS, relwidth=1/3, relheight=1/self.COL2_ROWS)
|
|
|
|
### SHARED SETTINGS ###
|
|
|
|
# GPU Selection
|
|
self.is_gpu_conversion_Option = ttk.Checkbutton(master=self.options_Frame, text=GPU_CONVERSION_MAIN_LABEL, variable=self.is_gpu_conversion_var)
|
|
self.is_gpu_conversion_Option_place = lambda:self.is_gpu_conversion_Option.place(x=CHECK_BOX_X, y=CHECK_BOX_Y, width=CHECK_BOX_WIDTH, height=CHECK_BOX_HEIGHT, relx=1/3, rely=5/self.COL2_ROWS, relwidth=1/3, relheight=1/self.COL2_ROWS)
|
|
self.is_gpu_conversion_Disable = lambda:(self.is_gpu_conversion_Option.configure(state=tk.DISABLED), self.is_gpu_conversion_var.set(False))
|
|
self.is_gpu_conversion_Enable = lambda:self.is_gpu_conversion_Option.configure(state=tk.NORMAL)
|
|
self.help_hints(self.is_gpu_conversion_Option, text=IS_GPU_CONVERSION_HELP)
|
|
|
|
# Vocal Only
|
|
self.is_primary_stem_only_Option = ttk.Checkbutton(master=self.options_Frame, textvariable=self.is_primary_stem_only_Text_var, variable=self.is_primary_stem_only_var, command=lambda:self.is_primary_stem_only_Option_toggle())
|
|
self.is_primary_stem_only_Option_place = lambda:self.is_primary_stem_only_Option.place(x=CHECK_BOX_X, y=CHECK_BOX_Y, width=CHECK_BOX_WIDTH, height=CHECK_BOX_HEIGHT, relx=1/3, rely=6/self.COL2_ROWS, relwidth=1/3, relheight=1/self.COL2_ROWS)
|
|
self.is_primary_stem_only_Option_toggle = lambda:self.is_secondary_stem_only_var.set(False) if self.is_primary_stem_only_var.get() else self.is_secondary_stem_only_Option.configure(state=tk.NORMAL)
|
|
self.help_hints(self.is_primary_stem_only_Option, text=SAVE_STEM_ONLY_HELP)
|
|
|
|
# Instrumental Only
|
|
self.is_secondary_stem_only_Option = ttk.Checkbutton(master=self.options_Frame, textvariable=self.is_secondary_stem_only_Text_var, variable=self.is_secondary_stem_only_var, command=lambda:self.is_secondary_stem_only_Option_toggle())
|
|
self.is_secondary_stem_only_Option_place = lambda:self.is_secondary_stem_only_Option.place(x=CHECK_BOX_X, y=CHECK_BOX_Y, width=CHECK_BOX_WIDTH, height=CHECK_BOX_HEIGHT, relx=1/3, rely=7/self.COL2_ROWS, relwidth=1/3, relheight=1/self.COL2_ROWS)
|
|
self.is_secondary_stem_only_Option_toggle = lambda:self.is_primary_stem_only_var.set(False) if self.is_secondary_stem_only_var.get() else self.is_primary_stem_only_Option.configure(state=tk.NORMAL)
|
|
self.is_stem_only_Options_Enable = lambda:(self.is_primary_stem_only_Option.configure(state=tk.NORMAL), self.is_secondary_stem_only_Option.configure(state=tk.NORMAL))
|
|
self.help_hints(self.is_secondary_stem_only_Option, text=SAVE_STEM_ONLY_HELP)
|
|
|
|
# Sample Mode
|
|
self.model_sample_mode_Option = ttk.Checkbutton(master=self.options_Frame, textvariable=self.model_sample_mode_duration_checkbox_var, variable=self.model_sample_mode_var)#f'Sample ({self.model_sample_mode_duration_var.get()} Seconds)'
|
|
self.model_sample_mode_Option_place = lambda rely=8:self.model_sample_mode_Option.place(x=CHECK_BOX_X, y=CHECK_BOX_Y, width=CHECK_BOX_WIDTH, height=CHECK_BOX_HEIGHT, relx=1/3, rely=rely/self.COL2_ROWS, relwidth=1/3, relheight=1/self.COL2_ROWS)
|
|
self.help_hints(self.model_sample_mode_Option, text=MODEL_SAMPLE_MODE_HELP)
|
|
|
|
self.GUI_LIST = (self.vr_model_Label,
|
|
self.vr_model_Option,
|
|
self.aggression_setting_Label,
|
|
self.aggression_setting_Option,
|
|
self.window_size_Label,
|
|
self.window_size_Option,
|
|
self.demucs_model_Label,
|
|
self.demucs_model_Option,
|
|
self.demucs_stems_Label,
|
|
self.demucs_stems_Option,
|
|
self.segment_Label,
|
|
self.segment_Option,
|
|
self.mdx_net_model_Label,
|
|
self.mdx_net_model_Option,
|
|
self.overlap_mdx_Label,
|
|
self.overlap_mdx_Option,
|
|
self.overlap_mdx23_Option,
|
|
self.mdxnet_stems_Label,
|
|
self.mdxnet_stems_Option,
|
|
self.mdx_segment_size_Label,
|
|
self.mdx_segment_size_Option,
|
|
self.chosen_ensemble_Label,
|
|
self.chosen_ensemble_Option,
|
|
self.save_current_settings_Label,
|
|
self.save_current_settings_Option,
|
|
self.ensemble_main_stem_Label,
|
|
self.ensemble_main_stem_Option,
|
|
self.ensemble_type_Label,
|
|
self.ensemble_type_Option,
|
|
self.ensemble_listbox_Label,
|
|
self.ensemble_listbox_Frame,
|
|
self.ensemble_listbox_Option,
|
|
self.ensemble_listbox_scroll,
|
|
self.chosen_audio_tool_Label,
|
|
self.chosen_audio_tool_Option,
|
|
self.choose_algorithm_Label,
|
|
self.choose_algorithm_Option,
|
|
self.time_stretch_rate_Label,
|
|
self.time_stretch_rate_Option,
|
|
self.wav_type_set_Label,
|
|
self.wav_type_set_Option,
|
|
self.pitch_rate_Label,
|
|
self.pitch_rate_Option,
|
|
self.fileOne_Label,
|
|
self.fileOne_Entry,
|
|
self.fileOne_Open,
|
|
self.fileTwo_Label,
|
|
self.fileTwo_Entry,
|
|
self.fileTwo_Open,
|
|
self.intro_analysis_Label,
|
|
self.intro_analysis_Option,
|
|
self.time_window_Label,
|
|
self.time_window_Option,
|
|
self.db_analysis_Label,
|
|
self.db_analysis_Option,
|
|
self.is_gpu_conversion_Option,
|
|
self.is_primary_stem_only_Option,
|
|
self.is_secondary_stem_only_Option,
|
|
self.is_primary_stem_only_Demucs_Option,
|
|
self.is_secondary_stem_only_Demucs_Option,
|
|
self.model_sample_mode_Option,
|
|
self.is_time_correction_Option,
|
|
self.is_wav_ensemble_Option)
|
|
|
|
REFRESH_VARS = (self.mdx_net_model_var,
|
|
self.vr_model_var,
|
|
self.demucs_model_var,
|
|
# self.demucs_stems_var,
|
|
# self.mdxnet_stems_var,
|
|
self.is_chunk_demucs_var,
|
|
self.is_chunk_mdxnet_var,
|
|
# self.is_primary_stem_only_Demucs_var,
|
|
# self.is_secondary_stem_only_Demucs_var,
|
|
# self.is_primary_stem_only_var,
|
|
# self.is_secondary_stem_only_var,
|
|
self.model_download_demucs_var,
|
|
self.model_download_mdx_var,
|
|
self.model_download_vr_var,
|
|
self.select_download_var,
|
|
# self.is_primary_stem_only_Demucs_Text_var,
|
|
# self.is_secondary_stem_only_Demucs_Text_var,
|
|
self.chosen_process_method_var,
|
|
self.ensemble_main_stem_var)
|
|
|
|
# Change States
|
|
for var in REFRESH_VARS:
|
|
var.trace_add('write', lambda *args: self.update_button_states())
|
|
|
|
def combo_box_selection_clear(self, frame:tk.Frame):
|
|
for option in frame.winfo_children():
|
|
if type(option) is ttk.Combobox or type(option) is ComboBoxEditableMenu:
|
|
option.selection_clear()
|
|
|
|
def focus_out_widgets(self, all_widgets, frame):
|
|
for option in all_widgets:
|
|
if not type(option) is ComboBoxEditableMenu:
|
|
option.bind('<Button-1>', lambda e:(option.focus(), self.combo_box_selection_clear(frame)))
|
|
|
|
def bind_widgets(self):
|
|
"""Bind widgets to the drag & drop mechanic"""
|
|
|
|
self.chosen_audio_tool_align = tk.BooleanVar(value=True)
|
|
other_items = [self.options_Frame, self.filePaths_Frame, self.title_Label, self.progressbar, self.conversion_Button, self.settings_Button, self.stop_Button, self.command_Text]
|
|
all_widgets = self.options_Frame.winfo_children() + self.filePaths_Frame.winfo_children() + other_items
|
|
self.focus_out_widgets(all_widgets, self.options_Frame)
|
|
|
|
if is_dnd_compatible:
|
|
self.filePaths_saveTo_Button.drop_target_register(DND_FILES)
|
|
self.filePaths_saveTo_Entry.drop_target_register(DND_FILES)
|
|
self.drop_target_register(DND_FILES)
|
|
self.dnd_bind('<<Drop>>', lambda e: drop(e, accept_mode='files'))
|
|
self.filePaths_saveTo_Button.dnd_bind('<<Drop>>', lambda e: drop(e, accept_mode='folder'))
|
|
self.filePaths_saveTo_Entry.dnd_bind('<<Drop>>', lambda e: drop(e, accept_mode='folder'))
|
|
|
|
self.fileOne_Entry.drop_target_register(DND_FILES)
|
|
self.fileTwo_Entry.drop_target_register(DND_FILES)
|
|
self.fileOne_Entry.dnd_bind('<<Drop>>', lambda e: drop(e, accept_mode=FILE_1))
|
|
self.fileTwo_Entry.dnd_bind('<<Drop>>', lambda e: drop(e, accept_mode=FILE_2))
|
|
|
|
self.ensemble_listbox_Option.bind('<<ListboxSelect>>', lambda e: self.chosen_ensemble_var.set(CHOOSE_ENSEMBLE_OPTION))
|
|
self.options_Frame.bind(right_click_button, lambda e:(self.right_click_menu_popup(e, main_menu=True), self.options_Frame.focus()))
|
|
self.filePaths_musicFile_Entry.bind(right_click_button, lambda e:(self.input_right_click_menu(e), self.filePaths_musicFile_Entry.focus()))
|
|
self.filePaths_musicFile_Entry.bind('<Button-1>', lambda e:(self.check_is_menu_open(INPUTS_MENU), self.filePaths_musicFile_Entry.focus()))
|
|
|
|
self.fileOne_Entry.bind('<Button-1>', lambda e:self.menu_batch_dual())
|
|
self.fileTwo_Entry.bind('<Button-1>', lambda e:self.menu_batch_dual())
|
|
self.fileOne_Entry.bind(right_click_button, lambda e:self.input_dual_right_click_menu(e, is_primary=True))
|
|
self.fileTwo_Entry.bind(right_click_button, lambda e:self.input_dual_right_click_menu(e, is_primary=False))
|
|
if not is_macos:
|
|
self.bind("<Configure>", self.adjust_toplevel_positions)
|
|
|
|
def auto_save(self):
|
|
try:
|
|
self.save_values(app_close=False, is_auto_save=True)
|
|
except Exception as e:
|
|
print(e)
|
|
|
|
#--Input/Export Methods--
|
|
|
|
def linux_filebox_fix(self, is_on=True):
|
|
fg_color_set = '#575757' if is_on else "#F6F6F7"
|
|
style = ttk.Style(self)
|
|
style.configure('TButton', foreground='#F6F6F7')
|
|
style.configure('TCheckbutton', foreground='#F6F6F7')
|
|
style.configure('TCombobox', foreground='#F6F6F7')
|
|
style.configure('TEntry', foreground='#F6F6F7')
|
|
style.configure('TLabel', foreground='#F6F6F7')
|
|
style.configure('TMenubutton', foreground='#F6F6F7')
|
|
style.configure('TRadiobutton', foreground='#F6F6F7')
|
|
gui_data.sv_ttk.set_theme("dark", MAIN_FONT_NAME, 10, fg_color_set=fg_color_set)
|
|
|
|
def show_file_dialog(self, text='Select Audio files', dialoge_type=None):
|
|
parent_win = root
|
|
is_linux = not is_windows and not is_macos
|
|
|
|
if is_linux:
|
|
self.linux_filebox_fix()
|
|
top = tk.Toplevel(root)
|
|
top.withdraw()
|
|
top.protocol("WM_DELETE_WINDOW", lambda: None)
|
|
parent_win = top
|
|
|
|
if dialoge_type == MULTIPLE_FILE:
|
|
filenames = filedialog.askopenfilenames(parent=parent_win,
|
|
title=text)
|
|
elif dialoge_type == MAIN_MULTIPLE_FILE:
|
|
filenames = filedialog.askopenfilenames(parent=parent_win,
|
|
title=text,
|
|
initialfile='',
|
|
initialdir=self.lastDir)
|
|
elif dialoge_type == SINGLE_FILE:
|
|
filenames = filedialog.askopenfilename(parent=parent_win,
|
|
title=text)
|
|
elif dialoge_type == CHOOSE_EXPORT_FIR:
|
|
filenames = filedialog.askdirectory(
|
|
parent=parent_win,
|
|
title=f'Select Folder',)
|
|
|
|
if is_linux:
|
|
print("Is Linux")
|
|
self.linux_filebox_fix(False)
|
|
top.destroy()
|
|
|
|
return filenames
|
|
|
|
def input_select_filedialog(self):
|
|
"""Make user select music files"""
|
|
|
|
if self.lastDir is not None:
|
|
if not os.path.isdir(self.lastDir):
|
|
self.lastDir = None
|
|
|
|
paths = self.show_file_dialog(dialoge_type=MAIN_MULTIPLE_FILE)
|
|
|
|
if paths: # Path selected
|
|
self.inputPaths = paths
|
|
|
|
self.process_input_selections()
|
|
self.update_inputPaths()
|
|
|
|
def export_select_filedialog(self):
|
|
"""Make user select a folder to export the converted files in"""
|
|
|
|
export_path = None
|
|
|
|
path = self.show_file_dialog(dialoge_type=CHOOSE_EXPORT_FIR)
|
|
|
|
if path: # Path selected
|
|
self.export_path_var.set(path)
|
|
export_path = self.export_path_var.get()
|
|
|
|
return export_path
|
|
|
|
def update_inputPaths(self):
|
|
"""Update the music file entry"""
|
|
|
|
if self.inputPaths:
|
|
if len(self.inputPaths) == 1:
|
|
text = self.inputPaths[0]
|
|
else:
|
|
count = len(self.inputPaths) - 1
|
|
file_text = 'file' if len(self.inputPaths) == 2 else 'files'
|
|
text = f"{self.inputPaths[0]}, +{count} {file_text}"
|
|
else:
|
|
# Empty Selection
|
|
text = ''
|
|
|
|
self.inputPathsEntry_var.set(text)
|
|
|
|
def select_audiofile(self, path=None, is_primary=True):
|
|
"""Make user select music files"""
|
|
|
|
vars = {
|
|
True: (self.fileOneEntry_Full_var, self.fileOneEntry_var, self.fileTwoEntry_Full_var, self.fileTwoEntry_var),
|
|
False: (self.fileTwoEntry_Full_var, self.fileTwoEntry_var, self.fileOneEntry_Full_var, self.fileOneEntry_var)
|
|
}
|
|
|
|
file_path_var, file_basename_var, file_path_2_var, file_basename_2_var = vars[is_primary]
|
|
|
|
if not path:
|
|
path = self.show_file_dialog(text='Select Audio file', dialoge_type=SINGLE_FILE)
|
|
|
|
if path: # Path selected
|
|
file_path_var.set(path)
|
|
file_basename_var.set(os.path.basename(path))
|
|
|
|
if BATCH_MODE_DUAL in file_path_2_var.get():
|
|
file_path_2_var.set("")
|
|
file_basename_2_var.set("")
|
|
|
|
self.DualBatch_inputPaths = []
|
|
self.check_dual_paths()
|
|
|
|
#--Utility Methods--
|
|
|
|
def restart(self):
|
|
"""Restart the application after asking for confirmation"""
|
|
|
|
confirm = messagebox.askyesno(parent=root,
|
|
title=CONFIRM_RESTART_TEXT[0],
|
|
message=CONFIRM_RESTART_TEXT[1])
|
|
|
|
if confirm:
|
|
self.save_values(app_close=True, is_restart=True)
|
|
|
|
def delete_temps(self, is_start_up=False):
|
|
"""Deletes temp files"""
|
|
|
|
DIRECTORIES = (BASE_PATH, VR_MODELS_DIR, MDX_MODELS_DIR, DEMUCS_MODELS_DIR, DEMUCS_NEWER_REPO_DIR)
|
|
EXTENSIONS = (('.aes', '.txt', '.tmp'))
|
|
|
|
try:
|
|
if os.path.isfile(f"{current_patch}{application_extension}"):
|
|
os.remove(f"{current_patch}{application_extension}")
|
|
|
|
if not is_start_up:
|
|
if os.path.isfile(SPLASH_DOC):
|
|
os.remove(SPLASH_DOC)
|
|
|
|
for dir in DIRECTORIES:
|
|
for temp_file in os.listdir(dir):
|
|
if temp_file.endswith(EXTENSIONS):
|
|
if os.path.isfile(os.path.join(dir, temp_file)):
|
|
os.remove(os.path.join(dir, temp_file))
|
|
except Exception as e:
|
|
self.error_log_var.set(error_text(TEMP_FILE_DELETION_TEXT, e))
|
|
|
|
def get_files_from_dir(self, directory, ext, is_mdxnet=False):
|
|
"""Gets files from specified directory that ends with specified extention"""
|
|
|
|
return tuple(
|
|
x if is_mdxnet and x.endswith(CKPT) else os.path.splitext(x)[0]
|
|
for x in os.listdir(directory)
|
|
if x.endswith(ext)
|
|
)
|
|
|
|
def return_ensemble_stems(self, is_primary=False):
|
|
"""Grabs and returns the chosen ensemble stems."""
|
|
|
|
ensemble_stem = self.ensemble_main_stem_var.get().partition("/")
|
|
|
|
if is_primary:
|
|
return ensemble_stem[0]
|
|
else:
|
|
return ensemble_stem[0], ensemble_stem[2]
|
|
|
|
def message_box(self, message):
|
|
"""Template for confirmation box"""
|
|
|
|
confirm = messagebox.askyesno(title=message[0],
|
|
message=message[1],
|
|
parent=root)
|
|
|
|
return confirm
|
|
|
|
def error_dialoge(self, message):
|
|
"""Template for messagebox that informs user of error"""
|
|
|
|
messagebox.showerror(master=self,
|
|
title=message[0],
|
|
message=message[1],
|
|
parent=root)
|
|
|
|
def model_list(self, primary_stem: str, secondary_stem: str, is_4_stem_check=False, is_multi_stem=False, is_dry_check=False, is_no_demucs=False, is_check_vocal_split=False):
|
|
|
|
stem_check = self.assemble_model_data(arch_type=ENSEMBLE_STEM_CHECK, is_dry_check=is_dry_check)
|
|
|
|
def matches_stem(model: ModelData):
|
|
primary_match = model.primary_stem in {primary_stem, secondary_stem}
|
|
mdx_stem_match = primary_stem in model.mdx_model_stems and model.mdx_stem_count <= 2
|
|
return primary_match or mdx_stem_match if is_no_demucs else primary_match or primary_stem in model.mdx_model_stems
|
|
|
|
result = []
|
|
|
|
for model in stem_check:
|
|
if is_multi_stem:
|
|
result.append(model.model_and_process_tag)
|
|
elif is_4_stem_check and (model.demucs_stem_count == 4 or model.mdx_stem_count == 4):
|
|
result.append(model.model_and_process_tag)
|
|
elif matches_stem(model) or (not is_no_demucs and primary_stem.lower() in model.demucs_source_list):
|
|
if is_check_vocal_split:
|
|
model_name = None if model.is_karaoke or not model.vocal_split_model else model.model_basename
|
|
else:
|
|
model_name = model.model_and_process_tag
|
|
|
|
result.append(model_name)
|
|
|
|
return result
|
|
|
|
def help_hints(self, widget, text):
|
|
toolTip = ToolTip(widget)
|
|
def enter(event):
|
|
if self.help_hints_var.get():
|
|
toolTip.showtip(text)
|
|
def leave(event):
|
|
toolTip.hidetip()
|
|
widget.bind('<Enter>', enter)
|
|
widget.bind('<Leave>', leave)
|
|
widget.bind(right_click_button, lambda e:copy_help_hint(e))
|
|
|
|
def copy_help_hint(event):
|
|
if self.help_hints_var.get():
|
|
right_click_menu = tk.Menu(self, font=(MAIN_FONT_NAME, FONT_SIZE_1), tearoff=0)
|
|
right_click_menu.add_command(label='Copy Help Hint Text', command=right_click_menu_copy_hint)
|
|
|
|
try:
|
|
right_click_menu.tk_popup(event.x_root,event.y_root)
|
|
right_click_release_linux(right_click_menu)
|
|
finally:
|
|
right_click_menu.grab_release()
|
|
else:
|
|
if widget.winfo_toplevel() == root:
|
|
self.right_click_menu_popup(event, main_menu=True)
|
|
|
|
def right_click_menu_copy_hint():
|
|
pyperclip.copy(text)
|
|
|
|
def check_is_menu_open(self, menu):
|
|
try:
|
|
menu_mapping = {
|
|
VR_OPTION: (self.is_open_menu_advanced_vr_options, self.menu_advanced_vr_options, self.menu_advanced_vr_options_close_window),
|
|
DEMUCS_OPTION: (self.is_open_menu_advanced_demucs_options, self.menu_advanced_demucs_options, self.menu_advanced_demucs_options_close_window),
|
|
MDX_OPTION: (self.is_open_menu_advanced_mdx_options, self.menu_advanced_mdx_options, self.menu_advanced_mdx_options_close_window),
|
|
ENSEMBLE_OPTION: (self.is_open_menu_advanced_ensemble_options, self.menu_advanced_ensemble_options, self.menu_advanced_ensemble_options_close_window),
|
|
HELP_OPTION: (self.is_open_menu_help, self.menu_help, self.menu_help_close_window),
|
|
ERROR_OPTION: (self.is_open_menu_error_log, self.menu_error_log, self.menu_error_log_close_window),
|
|
INPUTS_MENU: (self.is_open_menu_view_inputs, self.menu_view_inputs, self.menu_view_inputs_close_window),
|
|
ALIGNMENT_TOOL: (self.is_open_menu_advanced_align_options, self.menu_advanced_align_options, self.menu_advanced_align_options_close_window)
|
|
}
|
|
|
|
is_open, open_method, close_method = menu_mapping.get(menu, (None, None, None))
|
|
if is_open and is_open.get():
|
|
close_method()
|
|
open_method()
|
|
except Exception as e:
|
|
self.error_log_var.set("{}".format(error_text(menu, e)))
|
|
|
|
def input_right_click_menu(self, event):
|
|
|
|
right_click_menu = tk.Menu(self, font=(MAIN_FONT_NAME, FONT_SIZE_1), tearoff=0)
|
|
right_click_menu.add_command(label='See All Inputs', command=lambda:self.check_is_menu_open(INPUTS_MENU))
|
|
|
|
try:
|
|
right_click_menu.tk_popup(event.x_root,event.y_root)
|
|
right_click_release_linux(right_click_menu)
|
|
finally:
|
|
right_click_menu.grab_release()
|
|
|
|
def input_dual_right_click_menu(self, event, is_primary:bool):
|
|
input_path = self.fileOneEntry_Full_var.get() if is_primary else self.fileTwoEntry_Full_var.get()
|
|
right_click_menu = tk.Menu(self, font=(MAIN_FONT_NAME, FONT_SIZE_1), tearoff=0)
|
|
right_click_menu.add_command(label=CHOOSE_INPUT_TEXT, command=lambda:self.select_audiofile(is_primary=is_primary))
|
|
if input_path and os.path.isdir(os.path.dirname(input_path)):
|
|
right_click_menu.add_command(label=OPEN_INPUT_DIR_TEXT, command=lambda:OPEN_FILE_func(os.path.dirname(input_path)))
|
|
right_click_menu.add_command(label=BATCH_PROCESS_MENU_TEXT, command=self.menu_batch_dual)
|
|
|
|
try:
|
|
right_click_menu.tk_popup(event.x_root,event.y_root)
|
|
right_click_release_linux(right_click_menu)
|
|
finally:
|
|
right_click_menu.grab_release()
|
|
|
|
def cached_sources_clear(self):
|
|
|
|
self.vr_cache_source_mapper = {}
|
|
self.mdx_cache_source_mapper = {}
|
|
self.demucs_cache_source_mapper = {}
|
|
|
|
def cached_source_callback(self, process_method, model_name=None):
|
|
|
|
model, sources = None, None
|
|
|
|
if process_method == VR_ARCH_TYPE:
|
|
mapper = self.vr_cache_source_mapper
|
|
if process_method == MDX_ARCH_TYPE:
|
|
mapper = self.mdx_cache_source_mapper
|
|
if process_method == DEMUCS_ARCH_TYPE:
|
|
mapper = self.demucs_cache_source_mapper
|
|
|
|
for key, value in mapper.items():
|
|
if model_name in key:
|
|
model = key
|
|
sources = value
|
|
|
|
return model, sources
|
|
|
|
def cached_model_source_holder(self, process_method, sources, model_name=None):
|
|
|
|
if process_method == VR_ARCH_TYPE:
|
|
self.vr_cache_source_mapper = {**self.vr_cache_source_mapper, **{model_name: sources}}
|
|
if process_method == MDX_ARCH_TYPE:
|
|
self.mdx_cache_source_mapper = {**self.mdx_cache_source_mapper, **{model_name: sources}}
|
|
if process_method == DEMUCS_ARCH_TYPE:
|
|
self.demucs_cache_source_mapper = {**self.demucs_cache_source_mapper, **{model_name: sources}}
|
|
|
|
def cached_source_model_list_check(self, model_list: List[ModelData]):
|
|
|
|
model: ModelData
|
|
primary_model_names = lambda process_method:[model.model_basename if model.process_method == process_method else None for model in model_list]
|
|
secondary_model_names = lambda process_method:[model.secondary_model.model_basename if model.is_secondary_model_activated and model.process_method == process_method else None for model in model_list]
|
|
|
|
self.vr_primary_model_names = primary_model_names(VR_ARCH_TYPE)
|
|
self.mdx_primary_model_names = primary_model_names(MDX_ARCH_TYPE)
|
|
self.demucs_primary_model_names = primary_model_names(DEMUCS_ARCH_TYPE)
|
|
self.vr_secondary_model_names = secondary_model_names(VR_ARCH_TYPE)
|
|
self.mdx_secondary_model_names = secondary_model_names(MDX_ARCH_TYPE)
|
|
self.demucs_secondary_model_names = [model.secondary_model.model_basename if model.is_secondary_model_activated and model.process_method == DEMUCS_ARCH_TYPE and not model.secondary_model is None else None for model in model_list]
|
|
self.demucs_pre_proc_model_name = [model.pre_proc_model.model_basename if model.pre_proc_model else None for model in model_list]#list(dict.fromkeys())
|
|
|
|
for model in model_list:
|
|
if model.process_method == DEMUCS_ARCH_TYPE and model.is_demucs_4_stem_secondaries:
|
|
if not model.is_4_stem_ensemble:
|
|
self.demucs_secondary_model_names = model.secondary_model_4_stem_model_names_list
|
|
break
|
|
else:
|
|
for i in model.secondary_model_4_stem_model_names_list:
|
|
self.demucs_secondary_model_names.append(i)
|
|
|
|
self.all_models = self.vr_primary_model_names + self.mdx_primary_model_names + self.demucs_primary_model_names + self.vr_secondary_model_names + self.mdx_secondary_model_names + self.demucs_secondary_model_names + self.demucs_pre_proc_model_name
|
|
|
|
def verify_audio(self, audio_file, is_process=True, sample_path=None):
|
|
is_good = False
|
|
error_data = ''
|
|
|
|
if not type(audio_file) is tuple:
|
|
audio_file = [audio_file]
|
|
|
|
for i in audio_file:
|
|
if os.path.isfile(i):
|
|
try:
|
|
librosa.load(i, duration=3, mono=False, sr=44100) if not type(sample_path) is str else self.create_sample(i, sample_path)
|
|
is_good = True
|
|
except Exception as e:
|
|
error_name = f'{type(e).__name__}'
|
|
traceback_text = ''.join(traceback.format_tb(e.__traceback__))
|
|
message = f'{error_name}: "{e}"\n{traceback_text}"'
|
|
if is_process:
|
|
audio_base_name = os.path.basename(i)
|
|
self.error_log_var.set(f'{ERROR_LOADING_FILE_TEXT[0]}:\n\n\"{audio_base_name}\"\n\n{ERROR_LOADING_FILE_TEXT[1]}:\n\n{message}')
|
|
else:
|
|
error_data = AUDIO_VERIFICATION_CHECK(i, message)
|
|
|
|
if is_process:
|
|
return is_good
|
|
else:
|
|
return is_good, error_data
|
|
|
|
def create_sample(self, audio_file, sample_path=SAMPLE_CLIP_PATH):
|
|
try:
|
|
with audioread.audio_open(audio_file) as f:
|
|
track_length = int(f.duration)
|
|
except Exception as e:
|
|
print('Audioread failed to get duration. Trying Librosa...')
|
|
y, sr = librosa.load(audio_file, mono=False, sr=44100)
|
|
track_length = int(librosa.get_duration(y=y, sr=sr))
|
|
|
|
clip_duration = int(self.model_sample_mode_duration_var.get())
|
|
|
|
if track_length >= clip_duration:
|
|
offset_cut = track_length//3
|
|
off_cut = offset_cut + track_length
|
|
if not off_cut >= clip_duration:
|
|
offset_cut = 0
|
|
name_apped = f'{clip_duration}_second_'
|
|
else:
|
|
offset_cut, clip_duration = 0, track_length
|
|
name_apped = ''
|
|
|
|
sample = librosa.load(audio_file, offset=offset_cut, duration=clip_duration, mono=False, sr=44100)[0].T
|
|
audio_sample = os.path.join(sample_path, f'{os.path.splitext(os.path.basename(audio_file))[0]}_{name_apped}sample.wav')
|
|
sf.write(audio_sample, sample, 44100)
|
|
|
|
return audio_sample
|
|
|
|
#--Right Click Menu Pop-Ups--
|
|
|
|
def right_click_select_settings_sub(self, parent_menu, process_method):
|
|
saved_settings_sub_menu = tk.Menu(parent_menu, font=(MAIN_FONT_NAME, FONT_SIZE_1), tearoff=False)
|
|
settings_options = self.last_found_settings + tuple(SAVE_SET_OPTIONS)
|
|
|
|
for settings_options in settings_options:
|
|
settings_options = settings_options.replace("_", " ")
|
|
saved_settings_sub_menu.add_command(label=settings_options, command=lambda o=settings_options:self.selection_action_saved_settings(o, process_method=process_method))
|
|
|
|
saved_settings_sub_menu.insert_separator(len(self.last_found_settings))
|
|
|
|
return saved_settings_sub_menu
|
|
|
|
def right_click_menu_popup(self, event, text_box=False, main_menu=False):
|
|
|
|
def add_text_edit_options(menu):
|
|
"""Add options related to text editing."""
|
|
menu.add_command(label='Copy', command=self.right_click_menu_copy)
|
|
menu.add_command(label='Paste', command=lambda: self.right_click_menu_paste(text_box=text_box))
|
|
menu.add_command(label='Delete', command=lambda: self.right_click_menu_delete(text_box=text_box))
|
|
|
|
def add_advanced_settings_options(menu, settings_mapper, var_mapper):
|
|
"""Add advanced settings options to the menu."""
|
|
current_method = self.chosen_process_method_var.get()
|
|
|
|
if current_method in settings_mapper and (var_mapper[current_method] or (current_method == DEMUCS_ARCH_TYPE and self.is_demucs_pre_proc_model_activate_var.get())):
|
|
menu.add_cascade(label='Select Saved Settings', menu=saved_settings_sub_load_for_menu)
|
|
menu.add_separator()
|
|
for method, option in settings_mapper.items():
|
|
if method != ENSEMBLE_MODE or current_method == ENSEMBLE_MODE:
|
|
menu.add_command(label=f'Advanced {method} Settings', command=option)
|
|
elif current_method in settings_mapper:
|
|
menu.add_command(label=f'Advanced {current_method} Settings', command=settings_mapper[current_method])
|
|
|
|
# Create the right-click menu
|
|
right_click_menu = tk.Menu(self, font=(MAIN_FONT_NAME, FONT_SIZE_1), tearoff=0)
|
|
|
|
# Mappings
|
|
settings_mapper = {
|
|
ENSEMBLE_MODE: lambda: self.check_is_menu_open(ENSEMBLE_OPTION),
|
|
VR_ARCH_PM: lambda: self.check_is_menu_open(VR_OPTION),
|
|
MDX_ARCH_TYPE: lambda: self.check_is_menu_open(MDX_OPTION),
|
|
DEMUCS_ARCH_TYPE: lambda: self.check_is_menu_open(DEMUCS_OPTION)
|
|
}
|
|
|
|
var_mapper = {
|
|
ENSEMBLE_MODE: True,
|
|
VR_ARCH_PM: self.vr_is_secondary_model_activate_var.get(),
|
|
MDX_ARCH_TYPE: self.mdx_is_secondary_model_activate_var.get(),
|
|
DEMUCS_ARCH_TYPE: self.demucs_is_secondary_model_activate_var.get()
|
|
}
|
|
|
|
# Submenu for saved settings
|
|
saved_settings_sub_load_for_menu = tk.Menu(right_click_menu, font=(MAIN_FONT_NAME, FONT_SIZE_1), tearoff=False)
|
|
for label, arch_type in [(VR_ARCH_SETTING_LOAD, VR_ARCH_PM), (MDX_SETTING_LOAD, MDX_ARCH_TYPE), (DEMUCS_SETTING_LOAD, DEMUCS_ARCH_TYPE), (ALL_ARCH_SETTING_LOAD, None)]:
|
|
submenu = self.right_click_select_settings_sub(saved_settings_sub_load_for_menu, arch_type)
|
|
saved_settings_sub_load_for_menu.add_cascade(label=label, menu=submenu)
|
|
|
|
if not main_menu:
|
|
add_text_edit_options(right_click_menu)
|
|
else:
|
|
if self.chosen_process_method_var.get() == AUDIO_TOOLS and self.chosen_audio_tool_var.get() == ALIGN_INPUTS:
|
|
right_click_menu.add_command(label='Advanced Align Tool Settings', command=lambda: self.check_is_menu_open(ALIGNMENT_TOOL))
|
|
else:
|
|
add_advanced_settings_options(right_click_menu, settings_mapper, var_mapper)
|
|
|
|
# Additional Settings and Help Hints
|
|
if not self.is_menu_settings_open:
|
|
right_click_menu.add_command(label='Additional Settings', command=lambda: self.menu_settings(select_tab_2=True))
|
|
|
|
help_hints_label = 'Enable' if not self.help_hints_var.get() else 'Disable'
|
|
right_click_menu.add_command(label=f'{help_hints_label} Help Hints', command=lambda: self.help_hints_var.set(not self.help_hints_var.get()))
|
|
|
|
if self.error_log_var.get():
|
|
right_click_menu.add_command(label='Error Log', command=lambda: self.check_is_menu_open(ERROR_OPTION))
|
|
|
|
try:
|
|
right_click_menu.tk_popup(event.x_root, event.y_root)
|
|
right_click_release_linux(right_click_menu)
|
|
finally:
|
|
right_click_menu.grab_release()
|
|
|
|
def right_click_menu_copy(self):
|
|
hightlighted_text = self.current_text_box.selection_get()
|
|
self.clipboard_clear()
|
|
self.clipboard_append(hightlighted_text)
|
|
|
|
def right_click_menu_paste(self, text_box=False):
|
|
clipboard = self.clipboard_get()
|
|
self.right_click_menu_delete(text_box=True) if text_box else self.right_click_menu_delete()
|
|
self.current_text_box.insert(self.current_text_box.index(tk.INSERT), clipboard)
|
|
|
|
def right_click_menu_delete(self, text_box=False):
|
|
if text_box:
|
|
try:
|
|
s0 = self.current_text_box.index("sel.first")
|
|
s1 = self.current_text_box.index("sel.last")
|
|
self.current_text_box.tag_configure('highlight')
|
|
self.current_text_box.tag_add("highlight", s0, s1)
|
|
start_indexes = self.current_text_box.tag_ranges("highlight")[0::2]
|
|
end_indexes = self.current_text_box.tag_ranges("highlight")[1::2]
|
|
|
|
for start, end in zip(start_indexes, end_indexes):
|
|
self.current_text_box.tag_remove("highlight", start, end)
|
|
|
|
for start, end in zip(start_indexes, end_indexes):
|
|
self.current_text_box.delete(start, end)
|
|
except Exception as e:
|
|
print('RIGHT-CLICK-DELETE ERROR: \n', e)
|
|
else:
|
|
self.current_text_box.delete(0, tk.END)
|
|
|
|
def right_click_console(self, event):
|
|
right_click_menu = tk.Menu(self, font=(MAIN_FONT_NAME, FONT_SIZE_1), tearoff=0)
|
|
right_click_menu.add_command(label='Copy', command=self.command_Text.copy_text)
|
|
right_click_menu.add_command(label='Select All', command=self.command_Text.select_all_text)
|
|
|
|
try:
|
|
right_click_menu.tk_popup(event.x_root,event.y_root)
|
|
right_click_release_linux(right_click_menu)
|
|
finally:
|
|
right_click_menu.grab_release()
|
|
|
|
#--Secondary Window Methods--
|
|
|
|
def vocal_splitter_Button_opt(self, top_window, frame, pady, width=15):
|
|
vocal_splitter_Button = ttk.Button(frame, text=VOCAL_SPLITTER_OPTIONS_TEXT, command=lambda:self.pop_up_set_vocal_splitter(top_window), width=width)#
|
|
vocal_splitter_Button.grid(pady=pady)
|
|
|
|
def adjust_toplevel_positions(self, event):
|
|
# Copy the list to avoid modifying while iterating
|
|
for toplevel in self.toplevels.copy():
|
|
# Check if the toplevel window is still alive
|
|
if not toplevel.winfo_exists():
|
|
self.toplevels.remove(toplevel)
|
|
else:
|
|
menu_offset_x = (root.winfo_width() - toplevel.winfo_width()) // 2
|
|
menu_offset_y = (root.winfo_height() - toplevel.winfo_height()) // 2
|
|
toplevel.geometry("+%d+%d" % (root.winfo_x() + menu_offset_x, root.winfo_y() + menu_offset_y))
|
|
|
|
def menu_placement(self, window: tk.Toplevel, title, pop_up=False, is_help_hints=False, close_function=None, frame_list=None, top_window=None):
|
|
"""Prepares and centers each secondary window relative to the main window"""
|
|
|
|
top_window = top_window if top_window else root
|
|
window.withdraw()
|
|
window.resizable(False, False)
|
|
window.wm_transient(top_window)
|
|
window.title(title)
|
|
window.iconbitmap(ICON_IMG_PATH) if is_windows else self.tk.call('wm', 'iconphoto', window._w, tk.PhotoImage(file=MAIN_ICON_IMG_PATH))
|
|
|
|
root_location_x = root.winfo_x()
|
|
root_location_y = root.winfo_y()
|
|
root_x = root.winfo_width()
|
|
root_y = root.winfo_height()
|
|
window.update() if is_windows else window.update_idletasks()
|
|
sub_menu_x = window.winfo_reqwidth()
|
|
sub_menu_y = window.winfo_reqheight()
|
|
menu_offset_x = (root_x - sub_menu_x) // 2
|
|
menu_offset_y = (root_y - sub_menu_y) // 2
|
|
window.geometry("+%d+%d" %(root_location_x+menu_offset_x, root_location_y+menu_offset_y))
|
|
|
|
window.deiconify()
|
|
window.configure(bg=BG_COLOR)
|
|
|
|
if not is_macos:
|
|
self.toplevels.append(window)
|
|
|
|
def right_click_menu(event):
|
|
help_hints_label = 'Enable' if self.help_hints_var.get() == False else 'Disable'
|
|
help_hints_bool = True if self.help_hints_var.get() == False else False
|
|
right_click_menu = tk.Menu(self, font=(MAIN_FONT_NAME, FONT_SIZE_1), tearoff=0)
|
|
if is_help_hints:
|
|
right_click_menu.add_command(label=f'{help_hints_label} Help Hints', command=lambda:self.help_hints_var.set(help_hints_bool))
|
|
right_click_menu.add_command(label='Exit Window', command=close_function)
|
|
|
|
try:
|
|
right_click_menu.tk_popup(event.x_root,event.y_root)
|
|
right_click_release_linux(right_click_menu, window)
|
|
finally:
|
|
right_click_menu.grab_release()
|
|
|
|
if close_function:
|
|
window.bind(right_click_button, lambda e:right_click_menu(e))
|
|
|
|
if frame_list:
|
|
for frame in frame_list:
|
|
#self.adjust_widget_widths(frame)
|
|
self.focus_out_widgets(frame.winfo_children() + [frame], frame)
|
|
|
|
if pop_up:
|
|
window.attributes('-topmost', 'true') if OPERATING_SYSTEM == "Linux" else None
|
|
window.grab_set()
|
|
root.wait_window(window)
|
|
|
|
def adjust_widget_widths(self, frame):
|
|
|
|
def resize_widget(widgets):
|
|
max_width = max(wid.winfo_width() for wid in widgets)
|
|
for wid in widgets:
|
|
if isinstance(wid, (tk.Button, ttk.Combobox)):
|
|
# For widgets where width represents characters, not pixels
|
|
wid.configure(width=int(max_width / wid.winfo_pixels('1c')))
|
|
else:
|
|
# For widgets where width represents pixels
|
|
wid.configure(width=max_width)
|
|
|
|
resize_widget([widget for widget in frame.winfo_children() if isinstance(widget, tk.Button)])
|
|
resize_widget([widget for widget in frame.winfo_children() if isinstance(widget, ttk.Combobox)])
|
|
|
|
def menu_move_tab(notebook: ttk.Notebook, tab_text, new_position):
|
|
# Get the tab ID
|
|
tab_id = None
|
|
for tab in notebook.tabs():
|
|
if notebook.tab(tab, "text") == tab_text:
|
|
tab_id = tab
|
|
break
|
|
|
|
if tab_id is None:
|
|
print(f"No tab named '{tab_text}'")
|
|
return
|
|
|
|
# remove the tab
|
|
notebook.forget(tab_id)
|
|
|
|
# add it back in new position
|
|
notebook.insert(new_position, tab_id)
|
|
|
|
def menu_tab_control(self, toplevel, ai_network_vars, is_demucs=False, is_mdxnet=False):
|
|
"""Prepares the tabs setup for some windows"""
|
|
|
|
tabControl = ttk.Notebook(toplevel)
|
|
|
|
tab1 = ttk.Frame(tabControl)
|
|
tab2 = ttk.Frame(tabControl)
|
|
|
|
tabControl.add(tab1, text=SETTINGS_GUIDE_TEXT)
|
|
tabControl.add(tab2, text=SECONDARY_MODEL_TEXT)
|
|
|
|
tab1.grid_rowconfigure(0, weight=1)
|
|
tab1.grid_columnconfigure(0, weight=1)
|
|
|
|
tab2.grid_rowconfigure(0, weight=1)
|
|
tab2.grid_columnconfigure(0, weight=1)
|
|
|
|
if is_demucs or is_mdxnet:
|
|
tab3 = ttk.Frame(tabControl)
|
|
tabControl.add(tab3, text=PREPROCESS_MODEL_CHOOSE_TEXT if is_demucs else MDX23C_ONLY_OPTIONS_TEXT)
|
|
tab3.grid_rowconfigure(0, weight=1)
|
|
tab3.grid_columnconfigure(0, weight=1)
|
|
|
|
tabControl.pack(expand=1, fill=tk.BOTH)
|
|
|
|
self.tab2_loaded = False
|
|
self.tab3_loaded = False
|
|
|
|
def on_tab_selected(event):
|
|
# Check if it's tab2 (by tab id or tab title) and if it hasn't been loaded before
|
|
load_screen = False
|
|
if event.widget.tab('current', option='text') == 'Secondary Model' and not self.tab2_loaded:
|
|
tab = tab2
|
|
self.tab2_loaded = True
|
|
tab_load = lambda:self.menu_secondary_model(tab, ai_network_vars)
|
|
load_screen = True
|
|
elif event.widget.tab('current', option='text') == PREPROCESS_MODEL_CHOOSE_TEXT and not self.tab3_loaded:
|
|
tab = tab3
|
|
self.tab3_loaded = True
|
|
tab_load = lambda:self.menu_preproc_model(tab)
|
|
load_screen = True
|
|
|
|
if load_screen:
|
|
# Step 1: Add "Loading..." label
|
|
loading_label = ttk.Label(tab, text="Updating model lists...", font=Font(family=MAIN_FONT_NAME, size=14))
|
|
loading_label.place(relx=0.5, rely=0.5, anchor=tk.CENTER) # Assuming you want to center it
|
|
|
|
# Step 2: Update the UI to show the label
|
|
tab.update_idletasks()
|
|
|
|
# Load the content
|
|
tab_load()
|
|
|
|
# Step 3: Remove or update the "Loading..." label
|
|
loading_label.destroy() # Remove the label. Or you can update its text if desired.
|
|
|
|
#self.on_tab_changed(tabControl)
|
|
|
|
tabControl.bind("<<NotebookTabChanged>>", on_tab_selected)
|
|
|
|
if is_demucs or is_mdxnet:
|
|
return tab1, tab3
|
|
else:
|
|
return tab1
|
|
|
|
def menu_view_inputs(self):
|
|
|
|
menu_view_inputs_top = tk.Toplevel(root)
|
|
|
|
self.is_open_menu_view_inputs.set(True)
|
|
self.menu_view_inputs_close_window = lambda:close_window()
|
|
menu_view_inputs_top.protocol("WM_DELETE_WINDOW", self.menu_view_inputs_close_window)
|
|
|
|
input_length_var = tk.StringVar(value='')
|
|
input_info_text_var = tk.StringVar(value='')
|
|
is_widen_box_var = tk.BooleanVar(value=False)
|
|
is_play_file_var = tk.BooleanVar(value=False)
|
|
varification_text_var = tk.StringVar(value=VERIFY_INPUTS_TEXT)
|
|
|
|
reset_list = lambda:(input_files_listbox_Option.delete(0, 'end'), [input_files_listbox_Option.insert(tk.END, inputs) for inputs in self.inputPaths])
|
|
audio_input_total = lambda:input_length_var.set(f'{AUDIO_INPUT_TOTAL_TEXT}: {len(self.inputPaths)}')
|
|
audio_input_total()
|
|
|
|
def list_diff(list1, list2): return list(set(list1).symmetric_difference(set(list2)))
|
|
|
|
def list_to_string(list1): return '\n'.join(''.join(sub) for sub in list1)
|
|
|
|
def close_window():
|
|
self.verification_thread.kill() if self.thread_check(self.verification_thread) else None
|
|
self.is_open_menu_view_inputs.set(False)
|
|
menu_view_inputs_top.destroy()
|
|
|
|
def drag_n_drop(e):
|
|
input_info_text_var.set('')
|
|
drop(e, accept_mode='files')
|
|
reset_list()
|
|
audio_input_total()
|
|
|
|
def selected_files(is_remove=False):
|
|
if not self.thread_check(self.active_processing_thread):
|
|
items_list = [input_files_listbox_Option.get(i) for i in input_files_listbox_Option.curselection()]
|
|
inputPaths = list(self.inputPaths)# if is_remove else items_list
|
|
if is_remove:
|
|
[inputPaths.remove(i) for i in items_list if items_list]
|
|
else:
|
|
[inputPaths.remove(i) for i in self.inputPaths if i not in items_list]
|
|
removed_files = list_diff(self.inputPaths, inputPaths)
|
|
[input_files_listbox_Option.delete(input_files_listbox_Option.get(0, tk.END).index(i)) for i in removed_files]
|
|
starting_len = len(self.inputPaths)
|
|
self.inputPaths = tuple(inputPaths)
|
|
self.update_inputPaths()
|
|
audio_input_total()
|
|
input_info_text_var.set(f'{starting_len - len(self.inputPaths)} input(s) removed.')
|
|
else:
|
|
input_info_text_var.set('You cannot remove inputs during an active process.')
|
|
|
|
def box_size():
|
|
input_info_text_var.set('')
|
|
input_files_listbox_Option.config(width=230, height=25) if is_widen_box_var.get() else input_files_listbox_Option.config(width=110, height=17)
|
|
self.menu_placement(menu_view_inputs_top, 'Selected Inputs', pop_up=True)
|
|
|
|
def input_options(is_select_inputs=True):
|
|
input_info_text_var.set('')
|
|
if is_select_inputs:
|
|
self.input_select_filedialog()
|
|
else:
|
|
self.inputPaths = ()
|
|
reset_list()
|
|
self.update_inputPaths()
|
|
audio_input_total()
|
|
|
|
def pop_open_file_path(is_play_file=False):
|
|
if self.inputPaths:
|
|
track_selected = self.inputPaths[input_files_listbox_Option.index(tk.ACTIVE)]
|
|
if os.path.isfile(track_selected):
|
|
OPEN_FILE_func(track_selected if is_play_file else os.path.dirname(track_selected))
|
|
|
|
def get_export_dir():
|
|
if os.path.isdir(self.export_path_var.get()):
|
|
export_dir = self.export_path_var.get()
|
|
else:
|
|
export_dir = self.export_select_filedialog()
|
|
|
|
return export_dir
|
|
|
|
def verify_audio(is_create_samples=False):
|
|
inputPaths = list(self.inputPaths)
|
|
iterated_list = self.inputPaths if not is_create_samples else [input_files_listbox_Option.get(i) for i in input_files_listbox_Option.curselection()]
|
|
removed_files = []
|
|
export_dir = None
|
|
total_audio_count, current_file = len(iterated_list), 0
|
|
if iterated_list:
|
|
for i in iterated_list:
|
|
current_file += 1
|
|
input_info_text_var.set(f'{SAMPLE_BEGIN if is_create_samples else VERIFY_BEGIN}{current_file}/{total_audio_count}')
|
|
if is_create_samples:
|
|
export_dir = get_export_dir()
|
|
if not export_dir:
|
|
input_info_text_var.set(f'No export directory selected.')
|
|
return
|
|
is_good, error_data = self.verify_audio(i, is_process=False, sample_path=export_dir)
|
|
if not is_good:
|
|
inputPaths.remove(i)
|
|
removed_files.append(error_data)#sample = self.create_sample(i)
|
|
|
|
varification_text_var.set(VERIFY_INPUTS_TEXT)
|
|
input_files_listbox_Option.configure(state=tk.NORMAL)
|
|
|
|
if removed_files:
|
|
input_info_text_var.set(f'{len(removed_files)} {BROKEN_OR_INCOM_TEXT}')
|
|
error_text = ''
|
|
for i in removed_files:
|
|
error_text += i
|
|
removed_files = list_diff(self.inputPaths, inputPaths)
|
|
[input_files_listbox_Option.delete(input_files_listbox_Option.get(0, tk.END).index(i)) for i in removed_files]
|
|
self.error_log_var.set(REMOVED_FILES(list_to_string(removed_files), error_text))
|
|
self.inputPaths = tuple(inputPaths)
|
|
self.update_inputPaths()
|
|
else:
|
|
input_info_text_var.set(f'No errors found!')
|
|
|
|
audio_input_total()
|
|
else:
|
|
input_info_text_var.set(f'{NO_FILES_TEXT} {SELECTED_VER if is_create_samples else DETECTED_VER}')
|
|
varification_text_var.set(VERIFY_INPUTS_TEXT)
|
|
input_files_listbox_Option.configure(state=tk.NORMAL)
|
|
return
|
|
|
|
audio_input_total()
|
|
|
|
def verify_audio_start_thread(is_create_samples=False):
|
|
|
|
if not self.thread_check(self.active_processing_thread):
|
|
if not self.thread_check(self.verification_thread):
|
|
varification_text_var.set('Stop Progress')
|
|
input_files_listbox_Option.configure(state=tk.DISABLED)
|
|
self.verification_thread = KThread(target=lambda:verify_audio(is_create_samples=is_create_samples))
|
|
self.verification_thread.start()
|
|
else:
|
|
input_files_listbox_Option.configure(state=tk.NORMAL)
|
|
varification_text_var.set(VERIFY_INPUTS_TEXT)
|
|
input_info_text_var.set('Process Stopped')
|
|
self.verification_thread.kill()
|
|
else:
|
|
input_info_text_var.set('You cannot verify inputs during an active process.')
|
|
|
|
def right_click_menu(event):
|
|
right_click_menu = tk.Menu(self, font=(MAIN_FONT_NAME, FONT_SIZE_1), tearoff=0)
|
|
right_click_menu.add_command(label='Remove Selected Items Only', command=lambda:selected_files(is_remove=True))
|
|
right_click_menu.add_command(label='Keep Selected Items Only', command=lambda:selected_files(is_remove=False))
|
|
right_click_menu.add_command(label='Clear All Input(s)', command=lambda:input_options(is_select_inputs=False))
|
|
right_click_menu.add_separator()
|
|
right_click_menu_sub = tk.Menu(right_click_menu, font=(MAIN_FONT_NAME, FONT_SIZE_1), tearoff=False)
|
|
right_click_menu.add_command(label='Verify and Create Samples of Selected Inputs', command=lambda:verify_audio_start_thread(is_create_samples=True))
|
|
right_click_menu.add_cascade(label='Preferred Double Click Action', menu=right_click_menu_sub)
|
|
if is_play_file_var.get():
|
|
right_click_menu_sub.add_command(label='Enable: Open Audio File Directory', command=lambda:(input_files_listbox_Option.bind('<Double-Button>', lambda e:pop_open_file_path()), is_play_file_var.set(False)))
|
|
else:
|
|
right_click_menu_sub.add_command(label='Enable: Open Audio File', command=lambda:(input_files_listbox_Option.bind('<Double-Button>', lambda e:pop_open_file_path(is_play_file=True)), is_play_file_var.set(True)))
|
|
|
|
try:
|
|
right_click_menu.tk_popup(event.x_root,event.y_root)
|
|
right_click_release_linux(right_click_menu, menu_view_inputs_top)
|
|
finally:
|
|
right_click_menu.grab_release()
|
|
|
|
menu_view_inputs_Frame = self.menu_FRAME_SET(menu_view_inputs_top)
|
|
menu_view_inputs_Frame.grid(row=0)
|
|
|
|
self.main_window_LABEL_SET(menu_view_inputs_Frame, SELECTED_INPUTS).grid(row=0,column=0,padx=0,pady=MENU_PADDING_1)
|
|
tk.Label(menu_view_inputs_Frame, textvariable=input_length_var, font=(MAIN_FONT_NAME, f"{FONT_SIZE_1}"), foreground=FG_COLOR).grid(row=1, column=0, padx=0, pady=MENU_PADDING_1)
|
|
if not OPERATING_SYSTEM == "Linux":
|
|
ttk.Button(menu_view_inputs_Frame, text=SELECT_INPUTS, command=lambda:input_options()).grid(row=2,column=0,padx=0,pady=MENU_PADDING_2)
|
|
input_files_listbox_Option = tk.Listbox(menu_view_inputs_Frame, selectmode=tk.EXTENDED, activestyle='dotbox', font=(MAIN_FONT_NAME, f"{FONT_SIZE_1}"), background='#101414', exportselection=0, width=110, height=17, relief=tk.SOLID, borderwidth=0)
|
|
input_files_listbox_vertical_scroll = ttk.Scrollbar(menu_view_inputs_Frame, orient=tk.VERTICAL)
|
|
input_files_listbox_Option.config(yscrollcommand=input_files_listbox_vertical_scroll.set)
|
|
input_files_listbox_vertical_scroll.configure(command=input_files_listbox_Option.yview)
|
|
input_files_listbox_Option.grid(row=4, sticky=tk.W)
|
|
input_files_listbox_vertical_scroll.grid(row=4, column=1, sticky=tk.NS)
|
|
|
|
tk.Label(menu_view_inputs_Frame, textvariable=input_info_text_var, font=(MAIN_FONT_NAME, f"{FONT_SIZE_1}"), foreground=FG_COLOR).grid(row=5, column=0, padx=0, pady=0)
|
|
ttk.Checkbutton(menu_view_inputs_Frame, text=WIDEN_BOX, variable=is_widen_box_var, command=lambda:box_size()).grid(row=6,column=0,padx=0,pady=0)
|
|
verify_audio_Button = ttk.Button(menu_view_inputs_Frame, textvariable=varification_text_var, command=lambda:verify_audio_start_thread())
|
|
verify_audio_Button.grid(row=7,column=0,padx=0,pady=MENU_PADDING_1)
|
|
ttk.Button(menu_view_inputs_Frame, text=CLOSE_WINDOW, command=lambda:menu_view_inputs_top.destroy()).grid(row=8,column=0,padx=0,pady=MENU_PADDING_1)
|
|
|
|
if is_dnd_compatible:
|
|
menu_view_inputs_top.drop_target_register(DND_FILES)
|
|
menu_view_inputs_top.dnd_bind('<<Drop>>', lambda e: drag_n_drop(e))
|
|
input_files_listbox_Option.bind(right_click_button, lambda e:right_click_menu(e))
|
|
input_files_listbox_Option.bind('<Double-Button>', lambda e:pop_open_file_path())
|
|
input_files_listbox_Option.bind('<Delete>', lambda e:selected_files(is_remove=True))
|
|
input_files_listbox_Option.bind('<BackSpace>', lambda e:selected_files(is_remove=False))
|
|
|
|
reset_list()
|
|
|
|
self.menu_placement(menu_view_inputs_top, 'Selected Inputs', pop_up=True)
|
|
|
|
def menu_batch_dual(self):
|
|
menu_batch_dual_top = tk.Toplevel(root)
|
|
|
|
def drag_n_drop(event, accept_mode):
|
|
listbox = left_frame if accept_mode == FILE_1_LB else right_frame
|
|
paths = drop(event, accept_mode)
|
|
for item in paths:
|
|
if item not in listbox.path_list: # only add file if it's not already in the list
|
|
basename = os.path.basename(item)
|
|
listbox.listbox.insert(tk.END, basename) # insert basename to the listbox
|
|
listbox.path_list.append(item) # append the file path to the list
|
|
listbox.update_displayed_index()
|
|
|
|
def move_entry(is_primary=True):
|
|
if is_primary:
|
|
selected_frame, other_frame = left_frame, right_frame
|
|
else:
|
|
selected_frame, other_frame = right_frame, left_frame
|
|
|
|
selected = selected_frame.listbox.curselection()
|
|
|
|
if selected:
|
|
basename = selected_frame.listbox.get(selected[0]).split(': ', 1)[1] # remove displayed index
|
|
|
|
if basename in other_frame.basename_to_path:
|
|
return
|
|
|
|
path = selected_frame.basename_to_path[basename] # Get the actual path
|
|
|
|
selected_frame.listbox.delete(selected)
|
|
other_frame.listbox.insert(tk.END, basename)
|
|
|
|
selected_frame.path_list.remove(path)
|
|
del selected_frame.basename_to_path[basename]
|
|
|
|
other_frame.path_list.append(path)
|
|
other_frame.basename_to_path[basename] = path
|
|
|
|
selected_frame.update_displayed_index()
|
|
other_frame.update_displayed_index()
|
|
|
|
def open_selected_path(lb, is_play_file=False):
|
|
selected_frame = left_frame if lb == FILE_1_LB else right_frame
|
|
selected_path = selected_frame.get_selected_path()
|
|
|
|
if selected_path:
|
|
if os.path.isfile(selected_path):
|
|
OPEN_FILE_func(selected_path if is_play_file else os.path.dirname(selected_path))
|
|
|
|
def clear_all_data(lb):
|
|
selected_frame = left_frame if lb == FILE_1_LB else right_frame
|
|
selected_frame.listbox.delete(0, "end")
|
|
selected_frame.path_list.clear()
|
|
selected_frame.basename_to_path.clear()
|
|
|
|
def clear_all(event, lb):
|
|
selected_frame = left_frame if lb == FILE_1_LB else right_frame
|
|
selected = selected_frame.listbox.curselection()
|
|
|
|
right_click_menu = tk.Menu(self, font=(MAIN_FONT_NAME, FONT_SIZE_1), tearoff=0)
|
|
if selected:
|
|
right_click_menu.add_command(label='Open Location', command=lambda:open_selected_path(lb))
|
|
right_click_menu.add_command(label='Open File', command=lambda:open_selected_path(lb, is_play_file=True))
|
|
right_click_menu.add_command(label='Clear All', command=lambda:clear_all_data(lb))
|
|
|
|
try:
|
|
right_click_menu.tk_popup(event.x_root,event.y_root)
|
|
right_click_release_linux(right_click_menu, menu_batch_dual_top)
|
|
finally:
|
|
right_click_menu.grab_release()
|
|
|
|
def gather_input_list():
|
|
left_paths = list(left_frame.basename_to_path.values())
|
|
right_paths = list(right_frame.basename_to_path.values())
|
|
|
|
clear_all_data(FILE_1_LB)
|
|
clear_all_data(FILE_2_LB)
|
|
|
|
if left_paths and right_paths:
|
|
left_frame.select_input(left_paths)
|
|
right_frame.select_input(right_paths)
|
|
|
|
self.DualBatch_inputPaths = list(zip(left_paths, right_paths))
|
|
self.check_dual_paths()
|
|
menu_batch_dual_top.destroy()
|
|
|
|
menu_view_inputs_Frame = self.menu_FRAME_SET(menu_batch_dual_top)
|
|
menu_view_inputs_Frame.grid(row=0)
|
|
|
|
left_frame = ListboxBatchFrame(menu_view_inputs_Frame, self.file_one_sub_var.get().title(), move_entry, self.right_img, self.img_mapper)
|
|
left_frame.grid(row=0, column=0, sticky="nsew", padx=(0, 5))
|
|
|
|
right_frame = ListboxBatchFrame(menu_view_inputs_Frame, self.file_two_sub_var.get().title(), lambda:move_entry(False), self.left_img, self.img_mapper)
|
|
right_frame.grid(row=0, column=1, sticky="nsew", padx=(5, 0))
|
|
|
|
left_frame.listbox.drop_target_register(DND_FILES)
|
|
right_frame.listbox.drop_target_register(DND_FILES)
|
|
left_frame.listbox.dnd_bind('<<Drop>>', lambda e: drag_n_drop(e, FILE_1_LB))
|
|
right_frame.listbox.dnd_bind('<<Drop>>', lambda e: drag_n_drop(e, FILE_2_LB))
|
|
left_frame.listbox.dnd_bind(right_click_button, lambda e: clear_all(e, FILE_1_LB))
|
|
right_frame.listbox.dnd_bind(right_click_button, lambda e: clear_all(e, FILE_2_LB))
|
|
|
|
menu_view_inputs_bottom_Frame = self.menu_FRAME_SET(menu_batch_dual_top)
|
|
menu_view_inputs_bottom_Frame.grid(row=1)
|
|
|
|
confirm_btn = ttk.Button(menu_view_inputs_bottom_Frame, text=CONFIRM_ENTRIES, command=gather_input_list)
|
|
confirm_btn.grid(pady=MENU_PADDING_1)
|
|
|
|
close_btn = ttk.Button(menu_view_inputs_bottom_Frame, text=CLOSE_WINDOW, command=lambda:menu_batch_dual_top.destroy())
|
|
close_btn.grid(pady=MENU_PADDING_1)
|
|
|
|
if self.check_dual_paths():
|
|
left_frame_pane = [i[0] for i in self.DualBatch_inputPaths]
|
|
right_frame_pane = [i[1] for i in self.DualBatch_inputPaths]
|
|
left_frame.update_displayed_index(left_frame_pane)
|
|
right_frame.update_displayed_index(right_frame_pane)
|
|
self.check_dual_paths()
|
|
|
|
self.menu_placement(menu_batch_dual_top, DUAL_AUDIO_PROCESSING, pop_up=True)
|
|
|
|
def check_dual_paths(self, is_fill_menu=False):
|
|
|
|
if self.DualBatch_inputPaths:
|
|
first_paths = tuple(self.DualBatch_inputPaths)
|
|
first_paths_len = len(first_paths)
|
|
first_paths = first_paths[0]
|
|
|
|
if first_paths_len == 1:
|
|
file1_base_text = os.path.basename(first_paths[0])
|
|
file2_base_text = os.path.basename(first_paths[1])
|
|
else:
|
|
first_paths_len = first_paths_len - 1
|
|
file1_base_text = f"{os.path.basename(first_paths[0])}, +{first_paths_len} file(s){BATCH_MODE_DUAL}"
|
|
file2_base_text = f"{os.path.basename(first_paths[1])}, +{first_paths_len} file(s){BATCH_MODE_DUAL}"
|
|
|
|
self.fileOneEntry_var.set(file1_base_text)
|
|
self.fileOneEntry_Full_var.set(f"{first_paths[0]}")
|
|
self.fileTwoEntry_var.set(file2_base_text)
|
|
self.fileTwoEntry_Full_var.set(f"{first_paths[1]}")
|
|
else:
|
|
if is_fill_menu:
|
|
file_one = self.fileOneEntry_Full_var.get()
|
|
file_two = self.fileTwoEntry_Full_var.get()
|
|
|
|
if file_one and file_two and BATCH_MODE_DUAL not in file_one and BATCH_MODE_DUAL not in file_two:
|
|
self.DualBatch_inputPaths = [(file_one, file_two)]
|
|
else:
|
|
if BATCH_MODE_DUAL in self.fileOneEntry_var.get():
|
|
self.fileOneEntry_var.set("")
|
|
self.fileOneEntry_Full_var.set("")
|
|
if BATCH_MODE_DUAL in self.fileTwoEntry_var.get():
|
|
self.fileTwoEntry_var.set("")
|
|
self.fileTwoEntry_Full_var.set("")
|
|
|
|
return self.DualBatch_inputPaths
|
|
|
|
def fill_gpu_list(self):
|
|
try:
|
|
if cuda_available:
|
|
self.cuda_device_list = [f"{torch.cuda.get_device_properties(i).name}:{i}" for i in range(torch.cuda.device_count())]
|
|
self.cuda_device_list.insert(0, DEFAULT)
|
|
#print(self.cuda_device_list)
|
|
|
|
# if directml_available:
|
|
# self.opencl_list = [f"{torch_directml.device_name(i)}:{i}" for i in range(torch_directml.device_count())]
|
|
# self.opencl_list.insert(0, DEFAULT)
|
|
except Exception as e:
|
|
print(e)
|
|
|
|
# if is_cuda_only:
|
|
# self.is_use_opencl_var.set(False)
|
|
|
|
check_gpu_list = self.cuda_device_list#self.opencl_list if is_opencl_only or self.is_use_opencl_var.get() else self.cuda_device_list
|
|
if not self.device_set_var.get() in check_gpu_list:
|
|
self.device_set_var.set(DEFAULT)
|
|
|
|
def loop_gpu_list(self, option_menu:ComboBoxMenu, menu_name, option_list):
|
|
option_menu['values'] = option_list
|
|
option_menu.update_dropdown_size(option_list, menu_name)
|
|
|
|
def menu_settings(self, select_tab_2=False, select_tab_3=False):#**
|
|
"""Open Settings and Download Center"""
|
|
|
|
settings_menu = tk.Toplevel()
|
|
|
|
option_var = tk.StringVar(value=SELECT_SAVED_SETTING)
|
|
self.is_menu_settings_open = True
|
|
|
|
tabControl = ttk.Notebook(settings_menu)
|
|
|
|
tab1 = ttk.Frame(tabControl)
|
|
tab2 = ttk.Frame(tabControl)
|
|
tab3 = ttk.Frame(tabControl)
|
|
|
|
tabControl.add(tab1, text = SETTINGS_GUIDE_TEXT)
|
|
tabControl.add(tab2, text = ADDITIONAL_SETTINGS_TEXT)
|
|
tabControl.add(tab3, text = DOWNLOAD_CENTER_TEXT)
|
|
|
|
tabControl.pack(expand = 1, fill ="both")
|
|
|
|
tab1.grid_rowconfigure(0, weight=1)
|
|
tab1.grid_columnconfigure(0, weight=1)
|
|
|
|
tab2.grid_rowconfigure(0, weight=1)
|
|
tab2.grid_columnconfigure(0, weight=1)
|
|
|
|
tab3.grid_rowconfigure(0, weight=1)
|
|
tab3.grid_columnconfigure(0, weight=1)
|
|
|
|
self.disable_tabs = lambda:(tabControl.tab(0, state="disabled"), tabControl.tab(1, state="disabled"))
|
|
self.enable_tabs = lambda:(tabControl.tab(0, state="normal"), tabControl.tab(1, state="normal"))
|
|
self.main_menu_var = tk.StringVar(value=CHOOSE_ADVANCED_MENU_TEXT)
|
|
|
|
self.download_progress_bar_var.set(0)
|
|
self.download_progress_info_var.set('')
|
|
self.download_progress_percent_var.set('')
|
|
|
|
def set_vars_for_sample_mode(event):
|
|
value = int(float(event))
|
|
value = round(value / 5) * 5
|
|
self.model_sample_mode_duration_var.set(value)
|
|
self.model_sample_mode_duration_checkbox_var.set(SAMPLE_MODE_CHECKBOX(value))
|
|
self.model_sample_mode_duration_label_var.set(f'{value} {SECONDS_TEXT}')
|
|
|
|
#Settings Tab 1
|
|
settings_menu_main_Frame = self.menu_FRAME_SET(tab1)
|
|
settings_menu_main_Frame.grid(row=0)
|
|
settings_title_Label = self.menu_title_LABEL_SET(settings_menu_main_Frame, GENERAL_MENU_TEXT)
|
|
settings_title_Label.grid(pady=MENU_PADDING_2)
|
|
|
|
select_Label = self.menu_sub_LABEL_SET(settings_menu_main_Frame, ADDITIONAL_MENUS_INFORMATION_TEXT)
|
|
select_Label.grid(pady=MENU_PADDING_1)
|
|
|
|
select_Option = ComboBoxMenu(settings_menu_main_Frame, textvariable=self.main_menu_var, values=OPTION_LIST, width=GEN_SETTINGS_WIDTH+3)
|
|
select_Option.update_dropdown_size(OPTION_LIST, 'menuchoose', command=lambda e:(self.check_is_menu_open(self.main_menu_var.get()), close_window()))
|
|
select_Option.grid(pady=MENU_PADDING_1)
|
|
|
|
help_hints_Option = ttk.Checkbutton(settings_menu_main_Frame, text=ENABLE_HELP_HINTS_TEXT, variable=self.help_hints_var, width=HELP_HINT_CHECKBOX_WIDTH)
|
|
help_hints_Option.grid(pady=MENU_PADDING_1)
|
|
|
|
open_app_dir_Button = ttk.Button(settings_menu_main_Frame, text=OPEN_APPLICATION_DIRECTORY_TEXT, command=lambda:OPEN_FILE_func(BASE_PATH), width=SETTINGS_BUT_WIDTH)
|
|
open_app_dir_Button.grid(pady=MENU_PADDING_1)
|
|
|
|
reset_all_app_settings_Button = ttk.Button(settings_menu_main_Frame, text=RESET_ALL_SETTINGS_TO_DEFAULT_TEXT, command=lambda:self.load_to_default_confirm(), width=SETTINGS_BUT_WIDTH)#pop_up_change_model_defaults
|
|
reset_all_app_settings_Button.grid(pady=MENU_PADDING_1)
|
|
|
|
if is_windows:
|
|
restart_app_Button = ttk.Button(settings_menu_main_Frame, text=RESTART_APPLICATION_TEXT, command=lambda:self.restart())
|
|
restart_app_Button.grid(pady=MENU_PADDING_1)
|
|
|
|
delete_your_settings_Label = self.menu_title_LABEL_SET(settings_menu_main_Frame, DELETE_USER_SAVED_SETTING_TEXT)
|
|
delete_your_settings_Label.grid(pady=MENU_PADDING_2)
|
|
self.help_hints(delete_your_settings_Label, text=DELETE_YOUR_SETTINGS_HELP)
|
|
|
|
delete_your_settings_Option = ComboBoxMenu(settings_menu_main_Frame, textvariable=option_var, width=GEN_SETTINGS_WIDTH+3)
|
|
delete_your_settings_Option.grid(padx=20,pady=MENU_PADDING_1)
|
|
self.deletion_list_fill(delete_your_settings_Option, option_var, SETTINGS_CACHE_DIR, SELECT_SAVED_SETTING, menu_name='deletesetting')
|
|
|
|
app_update_Label = self.menu_title_LABEL_SET(settings_menu_main_Frame, APPLICATION_UPDATES_TEXT)
|
|
app_update_Label.grid(pady=MENU_PADDING_2)
|
|
|
|
self.app_update_button = ttk.Button(settings_menu_main_Frame, textvariable=self.app_update_button_Text_var, width=SETTINGS_BUT_WIDTH-2, command=lambda:self.pop_up_update_confirmation())
|
|
self.app_update_button.grid(pady=MENU_PADDING_1)
|
|
|
|
self.app_update_status_Label = tk.Label(settings_menu_main_Frame, textvariable=self.app_update_status_Text_var, padx=3, pady=3, font=(MAIN_FONT_NAME, f"{FONT_SIZE_4}"), width=UPDATE_LABEL_WIDTH, justify="center", relief="ridge", fg="#13849f")
|
|
self.app_update_status_Label.grid(pady=20)
|
|
|
|
donate_Button = ttk.Button(settings_menu_main_Frame, image=self.donate_img, command=lambda:webbrowser.open_new_tab(DONATE_LINK_BMAC))
|
|
donate_Button.grid(pady=MENU_PADDING_2)
|
|
self.help_hints(donate_Button, text=DONATE_HELP)
|
|
|
|
close_settings_win_Button = ttk.Button(settings_menu_main_Frame, text=CLOSE_WINDOW, command=lambda:close_window())
|
|
close_settings_win_Button.grid(pady=MENU_PADDING_1)
|
|
|
|
#Settings Tab 2
|
|
settings_menu_format_Frame = self.menu_FRAME_SET(tab2)
|
|
settings_menu_format_Frame.grid(row=0)
|
|
|
|
audio_format_title_Label = self.menu_title_LABEL_SET(settings_menu_format_Frame, AUDIO_FORMAT_SETTINGS_TEXT, width=20)
|
|
audio_format_title_Label.grid(pady=MENU_PADDING_2)
|
|
|
|
wav_type_set_Label = self.menu_sub_LABEL_SET(settings_menu_format_Frame, WAV_TYPE_TEXT)
|
|
wav_type_set_Label.grid(pady=MENU_PADDING_1)
|
|
|
|
wav_type_set_Option = ComboBoxMenu(settings_menu_format_Frame, textvariable=self.wav_type_set_var, values=WAV_TYPE, width=HELP_HINT_CHECKBOX_WIDTH)
|
|
wav_type_set_Option.grid(padx=20,pady=MENU_PADDING_1)
|
|
|
|
mp3_bit_set_Label = self.menu_sub_LABEL_SET(settings_menu_format_Frame, MP3_BITRATE_TEXT)
|
|
mp3_bit_set_Label.grid(pady=MENU_PADDING_1)
|
|
|
|
mp3_bit_set_Option = ComboBoxMenu(settings_menu_format_Frame, textvariable=self.mp3_bit_set_var, values=MP3_BIT_RATES, width=HELP_HINT_CHECKBOX_WIDTH)
|
|
mp3_bit_set_Option.grid(padx=20,pady=MENU_PADDING_1)
|
|
|
|
audio_format_title_Label = self.menu_title_LABEL_SET(settings_menu_format_Frame, GENERAL_PROCESS_SETTINGS_TEXT)
|
|
audio_format_title_Label.grid(pady=MENU_PADDING_2)
|
|
|
|
is_testing_audio_Option = ttk.Checkbutton(settings_menu_format_Frame, text=SETTINGS_TEST_MODE_TEXT, width=GEN_SETTINGS_WIDTH, variable=self.is_testing_audio_var)
|
|
is_testing_audio_Option.grid()
|
|
self.help_hints(is_testing_audio_Option, text=IS_TESTING_AUDIO_HELP)
|
|
|
|
is_add_model_name_Option = ttk.Checkbutton(settings_menu_format_Frame, text=MODEL_TEST_MODE_TEXT, width=GEN_SETTINGS_WIDTH, variable=self.is_add_model_name_var)
|
|
is_add_model_name_Option.grid()
|
|
self.help_hints(is_add_model_name_Option, text=IS_MODEL_TESTING_AUDIO_HELP)
|
|
|
|
is_create_model_folder_Option = ttk.Checkbutton(settings_menu_format_Frame, text=GENERATE_MODEL_FOLDER_TEXT, width=GEN_SETTINGS_WIDTH, variable=self.is_create_model_folder_var)
|
|
is_create_model_folder_Option.grid()
|
|
self.help_hints(is_create_model_folder_Option, text=IS_CREATE_MODEL_FOLDER_HELP)
|
|
|
|
is_accept_any_input_Option = ttk.Checkbutton(settings_menu_format_Frame, text=ACCEPT_ANY_INPUT_TEXT, width=GEN_SETTINGS_WIDTH, variable=self.is_accept_any_input_var)
|
|
is_accept_any_input_Option.grid()
|
|
self.help_hints(is_accept_any_input_Option, text=IS_ACCEPT_ANY_INPUT_HELP)
|
|
|
|
is_task_complete_Option = ttk.Checkbutton(settings_menu_format_Frame, text=NOTIFICATION_CHIMES_TEXT, width=GEN_SETTINGS_WIDTH, variable=self.is_task_complete_var)
|
|
is_task_complete_Option.grid()
|
|
self.help_hints(is_task_complete_Option, text=IS_TASK_COMPLETE_HELP)
|
|
|
|
is_normalization_Option = ttk.Checkbutton(settings_menu_format_Frame, text=NORMALIZE_OUTPUT_TEXT, width=GEN_SETTINGS_WIDTH, variable=self.is_normalization_var)
|
|
is_normalization_Option.grid()
|
|
self.help_hints(is_normalization_Option, text=IS_NORMALIZATION_HELP)
|
|
|
|
change_model_default_Button = ttk.Button(settings_menu_format_Frame, text=CHANGE_MODEL_DEFAULTS_TEXT, command=lambda:self.pop_up_change_model_defaults(settings_menu), width=SETTINGS_BUT_WIDTH-2)#
|
|
change_model_default_Button.grid(pady=MENU_PADDING_4)
|
|
|
|
#if not is_choose_arch:
|
|
self.vocal_splitter_Button_opt(settings_menu, settings_menu_format_Frame, width=SETTINGS_BUT_WIDTH-2, pady=MENU_PADDING_4)
|
|
|
|
if not is_macos and self.is_gpu_available:
|
|
gpu_list_options = lambda:self.loop_gpu_list(device_set_Option, 'gpudevice', self.cuda_device_list)#self.opencl_list if is_opencl_only or self.is_use_opencl_var.get() else self.cuda_device_list)
|
|
device_set_Label = self.menu_title_LABEL_SET(settings_menu_format_Frame, CUDA_NUM_TEXT)
|
|
device_set_Label.grid(pady=MENU_PADDING_2)
|
|
|
|
device_set_Option = ComboBoxMenu(settings_menu_format_Frame, textvariable=self.device_set_var, values=GPU_DEVICE_NUM_OPTS, width=GEN_SETTINGS_WIDTH+1)
|
|
device_set_Option.grid(padx=20,pady=MENU_PADDING_1)
|
|
gpu_list_options()
|
|
self.help_hints(device_set_Label, text=IS_CUDA_SELECT_HELP)
|
|
|
|
# if is_choose_arch:
|
|
# is_use_opencl_Option = ttk.Checkbutton(settings_menu_format_Frame,
|
|
# text=USE_OPENCL_TEXT,
|
|
# width=9,
|
|
# variable=self.is_use_opencl_var,
|
|
# command=lambda:(gpu_list_options(), self.device_set_var.set(DEFAULT)))
|
|
# is_use_opencl_Option.grid()
|
|
# self.help_hints(is_use_opencl_Option, text=IS_NORMALIZATION_HELP)
|
|
|
|
model_sample_mode_Label = self.menu_title_LABEL_SET(settings_menu_format_Frame, MODEL_SAMPLE_MODE_SETTINGS_TEXT)
|
|
model_sample_mode_Label.grid(pady=MENU_PADDING_2)
|
|
|
|
model_sample_mode_duration_Label = self.menu_sub_LABEL_SET(settings_menu_format_Frame, SAMPLE_CLIP_DURATION_TEXT)
|
|
model_sample_mode_duration_Label.grid(pady=MENU_PADDING_1)
|
|
|
|
tk.Label(settings_menu_format_Frame, textvariable=self.model_sample_mode_duration_label_var, font=(MAIN_FONT_NAME, f"{FONT_SIZE_1}"), foreground=FG_COLOR).grid(pady=2)
|
|
model_sample_mode_duration_Option = ttk.Scale(settings_menu_format_Frame, variable=self.model_sample_mode_duration_var, from_=5, to=120, command=set_vars_for_sample_mode, orient='horizontal')
|
|
model_sample_mode_duration_Option.grid(pady=2)
|
|
|
|
#Settings Tab 3
|
|
settings_menu_download_center_Frame = self.menu_FRAME_SET(tab3)
|
|
settings_menu_download_center_Frame.grid(row=0)
|
|
|
|
download_center_title_Label = self.menu_title_LABEL_SET(settings_menu_download_center_Frame, APPLICATION_DOWNLOAD_CENTER_TEXT)
|
|
download_center_title_Label.grid(padx=20,pady=MENU_PADDING_2)
|
|
|
|
select_download_Label = self.menu_sub_LABEL_SET(settings_menu_download_center_Frame, SELECT_DOWNLOAD_TEXT)
|
|
select_download_Label.grid(pady=MENU_PADDING_2)
|
|
|
|
self.model_download_vr_Button = ttk.Radiobutton(settings_menu_download_center_Frame, text='VR Arch', width=8, variable=self.select_download_var, value='VR Arc', command=lambda:self.download_list_state())
|
|
self.model_download_vr_Button.grid(pady=MENU_PADDING_1)
|
|
self.model_download_vr_Option = ComboBoxMenu(settings_menu_download_center_Frame, textvariable=self.model_download_vr_var, width=READ_ONLY_COMBO_WIDTH)
|
|
self.model_download_vr_Option.grid(pady=MENU_PADDING_1)
|
|
|
|
self.model_download_mdx_Button = ttk.Radiobutton(settings_menu_download_center_Frame, text='MDX-Net', width=8, variable=self.select_download_var, value='MDX-Net', command=lambda:self.download_list_state())
|
|
self.model_download_mdx_Button.grid(pady=MENU_PADDING_1)
|
|
self.model_download_mdx_Option = ComboBoxMenu(settings_menu_download_center_Frame, textvariable=self.model_download_mdx_var, width=READ_ONLY_COMBO_WIDTH)
|
|
self.model_download_mdx_Option.grid(pady=MENU_PADDING_1)
|
|
|
|
self.model_download_demucs_Button = ttk.Radiobutton(settings_menu_download_center_Frame, text='Demucs', width=8, variable=self.select_download_var, value='Demucs', command=lambda:self.download_list_state())
|
|
self.model_download_demucs_Button.grid(pady=MENU_PADDING_1)
|
|
self.model_download_demucs_Option = ComboBoxMenu(settings_menu_download_center_Frame, textvariable=self.model_download_demucs_var, width=READ_ONLY_COMBO_WIDTH)
|
|
self.model_download_demucs_Option.grid(pady=MENU_PADDING_1)
|
|
|
|
self.download_Button = ttk.Button(settings_menu_download_center_Frame, image=self.download_img, command=lambda:self.download_item())#, command=download_model)
|
|
self.download_Button.grid(pady=MENU_PADDING_1)
|
|
|
|
self.download_progress_info_Label = tk.Label(settings_menu_download_center_Frame, textvariable=self.download_progress_info_var, font=(MAIN_FONT_NAME, f"{FONT_SIZE_2}"), foreground=FG_COLOR, borderwidth=0)
|
|
self.download_progress_info_Label.grid(pady=MENU_PADDING_1)
|
|
|
|
self.download_progress_percent_Label = tk.Label(settings_menu_download_center_Frame, textvariable=self.download_progress_percent_var, font=(MAIN_FONT_NAME, f"{FONT_SIZE_2}"), wraplength=350, foreground=FG_COLOR)
|
|
self.download_progress_percent_Label.grid(pady=MENU_PADDING_1)
|
|
|
|
self.download_progress_bar_Progressbar = ttk.Progressbar(settings_menu_download_center_Frame, variable=self.download_progress_bar_var)
|
|
self.download_progress_bar_Progressbar.grid(pady=MENU_PADDING_1)
|
|
|
|
self.stop_download_Button = ttk.Button(settings_menu_download_center_Frame, textvariable=self.download_stop_var, width=15, command=lambda:self.download_post_action(DOWNLOAD_STOPPED))
|
|
self.stop_download_Button.grid(pady=MENU_PADDING_1)
|
|
self.stop_download_Button_DISABLE = lambda:(self.download_stop_var.set(""), self.stop_download_Button.configure(state=tk.DISABLED))
|
|
self.stop_download_Button_ENABLE = lambda:(self.download_stop_var.set(STOP_DOWNLOAD_TEXT), self.stop_download_Button.configure(state=tk.NORMAL))
|
|
|
|
self.refresh_list_Button = ttk.Button(settings_menu_download_center_Frame, text=REFRESH_LIST_TEXT, command=lambda:self.online_data_refresh(refresh_list_Button=True))#, command=refresh_list)
|
|
self.refresh_list_Button.grid(pady=MENU_PADDING_1)
|
|
|
|
self.download_key_Button = ttk.Button(settings_menu_download_center_Frame, image=self.key_img, command=lambda:self.pop_up_user_code_input())
|
|
self.download_key_Button.grid(pady=MENU_PADDING_1)
|
|
|
|
self.manual_download_Button = ttk.Button(settings_menu_download_center_Frame, text=TRY_MANUAL_DOWNLOAD_TEXT, command=self.menu_manual_downloads)
|
|
self.manual_download_Button.grid(pady=MENU_PADDING_1)
|
|
|
|
self.download_center_Buttons = (self.model_download_vr_Button,
|
|
self.model_download_mdx_Button,
|
|
self.model_download_demucs_Button,
|
|
self.download_Button,
|
|
self.download_key_Button)
|
|
|
|
self.download_lists = (self.model_download_vr_Option,
|
|
self.model_download_mdx_Option,
|
|
self.model_download_demucs_Option)
|
|
|
|
self.download_list_vars = (self.model_download_vr_var,
|
|
self.model_download_mdx_var,
|
|
self.model_download_demucs_var)
|
|
|
|
self.online_data_refresh()
|
|
|
|
self.menu_placement(settings_menu, SETTINGS_GUIDE_TEXT, is_help_hints=True, close_function=lambda:close_window())
|
|
|
|
if select_tab_2:
|
|
tabControl.select(tab2)
|
|
settings_menu.update_idletasks()
|
|
|
|
if select_tab_3:
|
|
tabControl.select(tab3)
|
|
settings_menu.update_idletasks()
|
|
|
|
def close_window():
|
|
self.active_download_thread.terminate() if self.thread_check(self.active_download_thread) else None
|
|
self.is_menu_settings_open = False
|
|
self.select_download_var.set('')
|
|
settings_menu.destroy()
|
|
|
|
#self.update_checkbox_text()
|
|
settings_menu.protocol("WM_DELETE_WINDOW", close_window)
|
|
|
|
def menu_advanced_vr_options(self):#**
|
|
"""Open Advanced VR Options"""
|
|
|
|
vr_opt = tk.Toplevel()
|
|
|
|
tab1 = self.menu_tab_control(vr_opt, self.vr_secondary_model_vars)
|
|
|
|
self.is_open_menu_advanced_vr_options.set(True)
|
|
self.menu_advanced_vr_options_close_window = lambda:(self.is_open_menu_advanced_vr_options.set(False), vr_opt.destroy())
|
|
vr_opt.protocol("WM_DELETE_WINDOW", self.menu_advanced_vr_options_close_window)
|
|
|
|
toggle_post_process = lambda:self.post_process_threshold_Option.configure(state=READ_ONLY) if self.is_post_process_var.get() else self.post_process_threshold_Option.configure(state=tk.DISABLED)
|
|
|
|
vr_opt_frame = self.menu_FRAME_SET(tab1)
|
|
vr_opt_frame.grid(pady=0 if not self.chosen_process_method_var.get() == VR_ARCH_PM else 70)
|
|
|
|
vr_title = self.menu_title_LABEL_SET(vr_opt_frame, ADVANCED_VR_OPTIONS_TEXT)
|
|
vr_title.grid(padx=25, pady=MENU_PADDING_2)
|
|
|
|
if not self.chosen_process_method_var.get() == VR_ARCH_PM:
|
|
window_size_Label = self.menu_sub_LABEL_SET(vr_opt_frame, WINDOW_SIZE_TEXT)
|
|
window_size_Label.grid(pady=MENU_PADDING_1)
|
|
window_size_Option = ComboBoxEditableMenu(vr_opt_frame, values=VR_WINDOW, width=MENU_COMBOBOX_WIDTH, textvariable=self.window_size_var, pattern=REG_WINDOW, default=VR_WINDOW[1])#
|
|
window_size_Option.grid(pady=MENU_PADDING_1)
|
|
self.help_hints(window_size_Label, text=WINDOW_SIZE_HELP)
|
|
|
|
aggression_setting_Label = self.menu_sub_LABEL_SET(vr_opt_frame, AGGRESSION_SETTING_TEXT)
|
|
aggression_setting_Label.grid(pady=MENU_PADDING_1)
|
|
aggression_setting_Option = ComboBoxEditableMenu(vr_opt_frame, values=VR_AGGRESSION, width=MENU_COMBOBOX_WIDTH, textvariable=self.aggression_setting_var, pattern=REG_AGGRESSION, default=VR_AGGRESSION[5])#
|
|
aggression_setting_Option.grid(pady=MENU_PADDING_1)
|
|
self.help_hints(aggression_setting_Label, text=AGGRESSION_SETTING_HELP)
|
|
|
|
self.batch_size_Label = self.menu_sub_LABEL_SET(vr_opt_frame, BATCH_SIZE_TEXT)
|
|
self.batch_size_Label.grid(pady=MENU_PADDING_1)
|
|
self.batch_size_Option = ComboBoxEditableMenu(vr_opt_frame, values=BATCH_SIZE, width=MENU_COMBOBOX_WIDTH, textvariable=self.batch_size_var, pattern=REG_BATCHES, default=BATCH_SIZE)#
|
|
self.batch_size_Option.grid(pady=MENU_PADDING_1)
|
|
self.help_hints(self.batch_size_Label, text=BATCH_SIZE_HELP)
|
|
|
|
self.post_process_threshold_Label = self.menu_sub_LABEL_SET(vr_opt_frame, POST_PROCESS_THRESHOLD_TEXT)
|
|
self.post_process_threshold_Label.grid(pady=MENU_PADDING_1)
|
|
self.post_process_threshold_Option = ComboBoxEditableMenu(vr_opt_frame, values=POST_PROCESSES_THREASHOLD_VALUES, width=MENU_COMBOBOX_WIDTH, textvariable=self.post_process_threshold_var, pattern=REG_THES_POSTPORCESS, default=POST_PROCESSES_THREASHOLD_VALUES[1])#
|
|
self.post_process_threshold_Option.grid(pady=MENU_PADDING_1)
|
|
self.help_hints(self.post_process_threshold_Label, text=POST_PROCESS_THREASHOLD_HELP)
|
|
|
|
self.is_tta_Option = ttk.Checkbutton(vr_opt_frame, text=ENABLE_TTA_TEXT, width=VR_CHECKBOXS_WIDTH, variable=self.is_tta_var)
|
|
self.is_tta_Option.grid(pady=0)
|
|
self.help_hints(self.is_tta_Option, text=IS_TTA_HELP)
|
|
|
|
self.is_post_process_Option = ttk.Checkbutton(vr_opt_frame, text=POST_PROCESS_TEXT, width=VR_CHECKBOXS_WIDTH, variable=self.is_post_process_var, command=toggle_post_process)
|
|
self.is_post_process_Option.grid(pady=0)
|
|
self.help_hints(self.is_post_process_Option, text=IS_POST_PROCESS_HELP)
|
|
|
|
self.is_high_end_process_Option = ttk.Checkbutton(vr_opt_frame, text=HIGHEND_PROCESS_TEXT, width=VR_CHECKBOXS_WIDTH, variable=self.is_high_end_process_var)
|
|
self.is_high_end_process_Option.grid(pady=0)
|
|
self.help_hints(self.is_high_end_process_Option, text=IS_HIGH_END_PROCESS_HELP)
|
|
|
|
self.vocal_splitter_Button_opt(vr_opt, vr_opt_frame, pady=MENU_PADDING_1, width=VR_BUT_WIDTH)
|
|
|
|
self.vr_clear_cache_Button = ttk.Button(vr_opt_frame, text=CLEAR_AUTOSET_CACHE_TEXT, command=lambda:self.clear_cache(VR_ARCH_TYPE), width=VR_BUT_WIDTH)
|
|
self.vr_clear_cache_Button.grid(pady=MENU_PADDING_1)
|
|
self.help_hints(self.vr_clear_cache_Button, text=CLEAR_CACHE_HELP)
|
|
|
|
self.open_vr_model_dir_Button = ttk.Button(vr_opt_frame, text=OPEN_MODELS_FOLDER_TEXT, command=lambda:OPEN_FILE_func(VR_MODELS_DIR), width=VR_BUT_WIDTH)
|
|
self.open_vr_model_dir_Button.grid(pady=MENU_PADDING_1)
|
|
|
|
self.vr_return_Button=ttk.Button(vr_opt_frame, text=BACK_TO_MAIN_MENU, command=lambda:(self.menu_advanced_vr_options_close_window(), self.check_is_menu_settings_open()))
|
|
self.vr_return_Button.grid(pady=MENU_PADDING_1)
|
|
|
|
self.vr_close_Button = ttk.Button(vr_opt_frame, text=CLOSE_WINDOW, command=lambda:self.menu_advanced_vr_options_close_window())
|
|
self.vr_close_Button.grid(pady=MENU_PADDING_1)
|
|
|
|
toggle_post_process()
|
|
|
|
frame_list = [vr_opt_frame]
|
|
self.menu_placement(vr_opt, ADVANCED_VR_OPTIONS_TEXT, is_help_hints=True, close_function=self.menu_advanced_vr_options_close_window, frame_list=frame_list)
|
|
|
|
def menu_advanced_demucs_options(self):#**
|
|
"""Open Advanced Demucs Options"""
|
|
|
|
demuc_opt = tk.Toplevel()
|
|
|
|
self.is_open_menu_advanced_demucs_options.set(True)
|
|
self.menu_advanced_demucs_options_close_window = lambda:(self.is_open_menu_advanced_demucs_options.set(False), demuc_opt.destroy())
|
|
demuc_opt.protocol("WM_DELETE_WINDOW", self.menu_advanced_demucs_options_close_window)
|
|
|
|
tab1, tab3 = self.menu_tab_control(demuc_opt, self.demucs_secondary_model_vars, is_demucs=True)
|
|
|
|
demucs_frame = self.menu_FRAME_SET(tab1)
|
|
demucs_frame.grid(pady=0 if not self.chosen_process_method_var.get() == DEMUCS_ARCH_TYPE else 55)
|
|
|
|
demucs_pre_model_frame = self.menu_FRAME_SET(tab3)
|
|
demucs_pre_model_frame.grid(row=0)
|
|
|
|
demucs_title_Label = self.menu_title_LABEL_SET(demucs_frame, ADVANCED_DEMUCS_OPTIONS_TEXT)
|
|
demucs_title_Label.grid(pady=MENU_PADDING_2)
|
|
|
|
if not self.chosen_process_method_var.get() == DEMUCS_ARCH_TYPE:
|
|
segment_Label = self.menu_sub_LABEL_SET(demucs_frame, SEGMENTS_TEXT)
|
|
segment_Label.grid(pady=MENU_PADDING_2)
|
|
segment_Option = ComboBoxEditableMenu(demucs_frame, values=DEMUCS_SEGMENTS, width=MENU_COMBOBOX_WIDTH, textvariable=self.segment_var, pattern=REG_SEGMENTS, default=DEMUCS_SEGMENTS)#
|
|
segment_Option.grid()
|
|
self.help_hints(segment_Label, text=SEGMENT_HELP)
|
|
|
|
self.shifts_Label = self.menu_sub_LABEL_SET(demucs_frame, SHIFTS_TEXT)
|
|
self.shifts_Label.grid(pady=MENU_PADDING_1)
|
|
self.shifts_Option = ComboBoxEditableMenu(demucs_frame, values=DEMUCS_SHIFTS, width=MENU_COMBOBOX_WIDTH, textvariable=self.shifts_var, pattern=REG_SHIFTS, default=DEMUCS_SHIFTS[2])#
|
|
self.shifts_Option.grid(pady=MENU_PADDING_1)
|
|
self.help_hints(self.shifts_Label, text=SHIFTS_HELP)
|
|
|
|
self.overlap_Label = self.menu_sub_LABEL_SET(demucs_frame, OVERLAP_TEXT)
|
|
self.overlap_Label.grid(pady=MENU_PADDING_1)
|
|
self.overlap_Option = ComboBoxEditableMenu(demucs_frame, values=DEMUCS_OVERLAP, width=MENU_COMBOBOX_WIDTH, textvariable=self.overlap_var, pattern=REG_OVERLAP, default=DEMUCS_OVERLAP)#
|
|
self.overlap_Option.grid(pady=MENU_PADDING_1)
|
|
self.help_hints(self.overlap_Label, text=OVERLAP_HELP)
|
|
|
|
pitch_shift_Label = self.menu_sub_LABEL_SET(demucs_frame, SHIFT_CONVERSION_PITCH_TEXT)
|
|
pitch_shift_Label.grid(pady=MENU_PADDING_1)
|
|
pitch_shift_Option = ComboBoxEditableMenu(demucs_frame, values=SEMITONE_SEL, width=MENU_COMBOBOX_WIDTH, textvariable=self.semitone_shift_var, pattern=REG_SEMITONES, default=SEMI_DEF)#
|
|
pitch_shift_Option.grid(pady=MENU_PADDING_1)
|
|
self.help_hints(pitch_shift_Label, text=PITCH_SHIFT_HELP)
|
|
|
|
self.is_split_mode_Option = ttk.Checkbutton(demucs_frame, text=SPLIT_MODE_TEXT, width=DEMUCS_CHECKBOXS_WIDTH, variable=self.is_split_mode_var)
|
|
self.is_split_mode_Option.grid()
|
|
self.help_hints(self.is_split_mode_Option, text=IS_SPLIT_MODE_HELP)
|
|
|
|
self.is_demucs_combine_stems_Option = ttk.Checkbutton(demucs_frame, text=COMBINE_STEMS_TEXT, width=DEMUCS_CHECKBOXS_WIDTH, variable=self.is_demucs_combine_stems_var)
|
|
self.is_demucs_combine_stems_Option.grid()
|
|
self.help_hints(self.is_demucs_combine_stems_Option, text=IS_DEMUCS_COMBINE_STEMS_HELP)
|
|
|
|
is_invert_spec_Option = ttk.Checkbutton(demucs_frame, text=SPECTRAL_INVERSION_TEXT, width=DEMUCS_CHECKBOXS_WIDTH, variable=self.is_invert_spec_var)
|
|
is_invert_spec_Option.grid()
|
|
self.help_hints(is_invert_spec_Option, text=IS_INVERT_SPEC_HELP)
|
|
|
|
self.vocal_splitter_Button_opt(demuc_opt, demucs_frame, width=VR_BUT_WIDTH, pady=MENU_PADDING_1)
|
|
|
|
self.open_demucs_model_dir_Button = ttk.Button(demucs_frame, text=OPEN_MODELS_FOLDER_TEXT, command=lambda:OPEN_FILE_func(DEMUCS_MODELS_DIR), width=VR_BUT_WIDTH)
|
|
self.open_demucs_model_dir_Button.grid(pady=MENU_PADDING_1)
|
|
|
|
self.demucs_return_Button = ttk.Button(demucs_frame, text=BACK_TO_MAIN_MENU, command=lambda:(self.menu_advanced_demucs_options_close_window(), self.check_is_menu_settings_open()))
|
|
self.demucs_return_Button.grid(pady=MENU_PADDING_1)
|
|
|
|
self.demucs_close_Button = ttk.Button(demucs_frame, text=CLOSE_WINDOW, command=lambda:self.menu_advanced_demucs_options_close_window())
|
|
self.demucs_close_Button.grid(pady=MENU_PADDING_1)
|
|
|
|
frame_list = [demucs_pre_model_frame, demucs_frame]
|
|
self.menu_placement(demuc_opt, ADVANCED_DEMUCS_OPTIONS_TEXT, is_help_hints=True, close_function=self.menu_advanced_demucs_options_close_window, frame_list=frame_list)
|
|
|
|
def menu_advanced_mdx_options(self):#**
|
|
"""Open Advanced MDX Options"""
|
|
|
|
mdx_net_opt = tk.Toplevel()
|
|
|
|
self.is_open_menu_advanced_mdx_options.set(True)
|
|
self.menu_advanced_mdx_options_close_window = lambda:(self.is_open_menu_advanced_mdx_options.set(False), mdx_net_opt.destroy())
|
|
mdx_net_opt.protocol("WM_DELETE_WINDOW", self.menu_advanced_mdx_options_close_window)
|
|
|
|
tab1, tab3 = self.menu_tab_control(mdx_net_opt, self.mdx_secondary_model_vars, is_mdxnet=True)
|
|
|
|
mdx_net_frame = self.menu_FRAME_SET(tab1)
|
|
mdx_net_frame.grid(pady=0)
|
|
|
|
mdx_net23_frame = self.menu_FRAME_SET(tab3)
|
|
mdx_net23_frame.grid(pady=0)
|
|
|
|
mdx_opt_title = self.menu_title_LABEL_SET(mdx_net_frame, ADVANCED_MDXNET_OPTIONS_TEXT)
|
|
mdx_opt_title.grid(pady=MENU_PADDING_1)
|
|
|
|
compensate_Label = self.menu_sub_LABEL_SET(mdx_net_frame, VOLUME_COMPENSATION_TEXT)
|
|
compensate_Label.grid(pady=MENU_PADDING_4)
|
|
compensate_Option = ComboBoxEditableMenu(mdx_net_frame, values=VOL_COMPENSATION, width=MENU_COMBOBOX_WIDTH, textvariable=self.compensate_var, pattern=REG_COMPENSATION, default=VOL_COMPENSATION)#
|
|
compensate_Option.grid(pady=MENU_PADDING_4)
|
|
self.help_hints(compensate_Label, text=COMPENSATE_HELP)
|
|
|
|
mdx_segment_size_Label = self.menu_sub_LABEL_SET(mdx_net_frame, SEGMENT_SIZE_TEXT)
|
|
mdx_segment_size_Label.grid(pady=MENU_PADDING_4)
|
|
mdx_segment_size_Option = ComboBoxEditableMenu(mdx_net_frame, values=MDX_SEGMENTS, width=MENU_COMBOBOX_WIDTH, textvariable=self.mdx_segment_size_var, pattern=REG_MDX_SEG, default="256")#
|
|
mdx_segment_size_Option.grid(pady=MENU_PADDING_4)
|
|
self.help_hints(mdx_segment_size_Label, text=MDX_SEGMENT_SIZE_HELP)
|
|
|
|
overlap_mdx_Label = self.menu_sub_LABEL_SET(mdx_net_frame, OVERLAP_TEXT)
|
|
overlap_mdx_Label.grid(pady=MENU_PADDING_4)
|
|
overlap_mdx_Option = ComboBoxEditableMenu(mdx_net_frame, values=MDX_OVERLAP, width=MENU_COMBOBOX_WIDTH, textvariable=self.overlap_mdx_var, pattern=REG_OVERLAP, default=MDX_OVERLAP)#
|
|
overlap_mdx_Option.grid(pady=MENU_PADDING_4)
|
|
self.help_hints(overlap_mdx_Label, text=OVERLAP_HELP)
|
|
|
|
pitch_shift_Label = self.menu_sub_LABEL_SET(mdx_net_frame, SHIFT_CONVERSION_PITCH_TEXT)
|
|
pitch_shift_Label.grid(pady=MENU_PADDING_4)
|
|
pitch_shift_Option = ComboBoxEditableMenu(mdx_net_frame, values=SEMITONE_SEL, width=MENU_COMBOBOX_WIDTH, textvariable=self.semitone_shift_var, pattern=REG_SEMITONES, default=SEMI_DEF)#
|
|
pitch_shift_Option.grid(pady=MENU_PADDING_4)
|
|
self.help_hints(pitch_shift_Label, text=PITCH_SHIFT_HELP)
|
|
|
|
if not os.path.isfile(DENOISER_MODEL_PATH):
|
|
denoise_options_var_text = self.denoise_option_var.get()
|
|
denoise_options = [option for option in MDX_DENOISE_OPTION if option != DENOISE_M]
|
|
self.denoise_option_var.set(DENOISE_S if denoise_options_var_text == DENOISE_M else denoise_options_var_text)
|
|
else:
|
|
denoise_options = MDX_DENOISE_OPTION
|
|
|
|
denoise_option_Label = self.menu_sub_LABEL_SET(mdx_net_frame, DENOISE_OUTPUT_TEXT)
|
|
denoise_option_Label.grid(pady=MENU_PADDING_4)
|
|
denoise_option_Option = ComboBoxMenu(mdx_net_frame, textvariable=self.denoise_option_var, values=denoise_options, width=MENU_COMBOBOX_WIDTH)
|
|
denoise_option_Option.grid(pady=MENU_PADDING_4)
|
|
self.help_hints(denoise_option_Label, text=IS_DENOISE_HELP)
|
|
|
|
is_match_frequency_pitch_Option = ttk.Checkbutton(mdx_net_frame, text=MATCH_FREQ_CUTOFF_TEXT, width=MDX_CHECKBOXS_WIDTH, variable=self.is_match_frequency_pitch_var)
|
|
is_match_frequency_pitch_Option.grid(pady=0)
|
|
self.help_hints(is_match_frequency_pitch_Option, text=IS_FREQUENCY_MATCH_HELP)
|
|
|
|
is_invert_spec_Option = ttk.Checkbutton(mdx_net_frame, text=SPECTRAL_INVERSION_TEXT, width=MDX_CHECKBOXS_WIDTH, variable=self.is_invert_spec_var)
|
|
is_invert_spec_Option.grid(pady=0)
|
|
self.help_hints(is_invert_spec_Option, text=IS_INVERT_SPEC_HELP)
|
|
|
|
self.vocal_splitter_Button_opt(mdx_net_opt, mdx_net_frame, pady=MENU_PADDING_1, width=VR_BUT_WIDTH)
|
|
|
|
clear_mdx_cache_Button = ttk.Button(mdx_net_frame, text=CLEAR_AUTOSET_CACHE_TEXT, command=lambda:self.clear_cache(MDX_ARCH_TYPE), width=VR_BUT_WIDTH)
|
|
clear_mdx_cache_Button.grid(pady=MENU_PADDING_1)
|
|
self.help_hints(clear_mdx_cache_Button, text=CLEAR_CACHE_HELP)
|
|
|
|
open_mdx_model_dir_Button = ttk.Button(mdx_net_frame, text=OPEN_MODELS_FOLDER_TEXT, command=lambda:OPEN_FILE_func(MDX_MODELS_DIR), width=VR_BUT_WIDTH)
|
|
open_mdx_model_dir_Button.grid(pady=MENU_PADDING_1)
|
|
|
|
mdx_return_Button = ttk.Button(mdx_net_frame, text=BACK_TO_MAIN_MENU, command=lambda:(self.menu_advanced_mdx_options_close_window(), self.check_is_menu_settings_open()))
|
|
mdx_return_Button.grid(pady=MENU_PADDING_1)
|
|
|
|
mdx_close_Button = ttk.Button(mdx_net_frame, text=CLOSE_WINDOW, command=lambda:self.menu_advanced_mdx_options_close_window())
|
|
mdx_close_Button.grid(pady=MENU_PADDING_1)
|
|
|
|
mdx23_opt_title = self.menu_title_LABEL_SET(mdx_net23_frame, ADVANCED_MDXNET23_OPTIONS_TEXT)
|
|
mdx23_opt_title.grid(pady=MENU_PADDING_2)
|
|
|
|
mdx_batch_size_Label = self.menu_sub_LABEL_SET(mdx_net23_frame, BATCH_SIZE_TEXT)
|
|
mdx_batch_size_Label.grid(pady=MENU_PADDING_1)
|
|
mdx_batch_size_Option = ComboBoxEditableMenu(mdx_net23_frame, values=BATCH_SIZE, width=MENU_COMBOBOX_WIDTH, textvariable=self.mdx_batch_size_var, pattern=REG_BATCHES, default=BATCH_SIZE)#
|
|
mdx_batch_size_Option.grid(pady=MENU_PADDING_1)
|
|
self.help_hints(mdx_batch_size_Label, text=BATCH_SIZE_HELP)
|
|
|
|
overlap_mdx23_Label = self.menu_sub_LABEL_SET(mdx_net23_frame, OVERLAP_TEXT)
|
|
overlap_mdx23_Label.grid(pady=MENU_PADDING_1)
|
|
overlap_mdx23_Option = ComboBoxEditableMenu(mdx_net23_frame, values=MDX23_OVERLAP, width=MENU_COMBOBOX_WIDTH, textvariable=self.overlap_mdx23_var, pattern=REG_OVERLAP23, default="8")#
|
|
overlap_mdx23_Option.grid(pady=MENU_PADDING_1)
|
|
self.help_hints(overlap_mdx23_Label, text=OVERLAP_23_HELP)
|
|
|
|
is_mdx_c_seg_def_Option = ttk.Checkbutton(mdx_net23_frame, text=SEGMENT_DEFAULT_TEXT, width=MDX_CHECKBOXS_WIDTH, variable=self.is_mdx_c_seg_def_var)
|
|
is_mdx_c_seg_def_Option.grid(pady=0)
|
|
self.help_hints(is_mdx_c_seg_def_Option, text=IS_SEGMENT_DEFAULT_HELP)
|
|
|
|
is_mdx_combine_stems_Option = ttk.Checkbutton(mdx_net23_frame, text=COMBINE_STEMS_TEXT, width=MDX_CHECKBOXS_WIDTH, variable=self.is_mdx23_combine_stems_var)
|
|
is_mdx_combine_stems_Option.grid()
|
|
self.help_hints(is_mdx_combine_stems_Option, text=IS_DEMUCS_COMBINE_STEMS_HELP)
|
|
|
|
mdx23_close_Button = ttk.Button(mdx_net23_frame, text=CLOSE_WINDOW, command=lambda:self.menu_advanced_mdx_options_close_window())
|
|
mdx23_close_Button.grid(pady=MENU_PADDING_2)
|
|
|
|
frame_list = [mdx_net_frame, mdx_net23_frame]
|
|
self.menu_placement(mdx_net_opt, ADVANCED_MDXNET_OPTIONS_TEXT, is_help_hints=True, close_function=self.menu_advanced_mdx_options_close_window, frame_list=frame_list)
|
|
|
|
def menu_advanced_ensemble_options(self):#**
|
|
"""Open Ensemble Custom"""
|
|
|
|
custom_ens_opt = tk.Toplevel()
|
|
|
|
self.is_open_menu_advanced_ensemble_options.set(True)
|
|
self.menu_advanced_ensemble_options_close_window = lambda:(self.is_open_menu_advanced_ensemble_options.set(False), custom_ens_opt.destroy())
|
|
custom_ens_opt.protocol("WM_DELETE_WINDOW", self.menu_advanced_ensemble_options_close_window)
|
|
|
|
option_var = tk.StringVar(value=SELECT_SAVED_ENSEMBLE)
|
|
|
|
custom_ens_opt_frame = self.menu_FRAME_SET(custom_ens_opt)
|
|
custom_ens_opt_frame.grid(row=0)
|
|
|
|
settings_title_Label = self.menu_title_LABEL_SET(custom_ens_opt_frame, ADVANCED_OPTION_MENU_TEXT)
|
|
settings_title_Label.grid(pady=MENU_PADDING_2)
|
|
|
|
delete_entry_Label = self.menu_sub_LABEL_SET(custom_ens_opt_frame, REMOVE_SAVED_ENSEMBLE_TEXT)
|
|
delete_entry_Label.grid(pady=MENU_PADDING_1)
|
|
delete_entry_Option = ComboBoxMenu(custom_ens_opt_frame, textvariable=option_var, width=ENSEMBLE_CHECKBOXS_WIDTH+2)
|
|
delete_entry_Option.grid(padx=20,pady=MENU_PADDING_1)
|
|
|
|
is_save_all_outputs_ensemble_Option = ttk.Checkbutton(custom_ens_opt_frame, text=SAVE_ALL_OUTPUTS_TEXT, width=ENSEMBLE_CHECKBOXS_WIDTH, variable=self.is_save_all_outputs_ensemble_var)
|
|
is_save_all_outputs_ensemble_Option.grid(pady=0)
|
|
self.help_hints(is_save_all_outputs_ensemble_Option, text=IS_SAVE_ALL_OUTPUTS_ENSEMBLE_HELP)
|
|
|
|
is_append_ensemble_name_Option = ttk.Checkbutton(custom_ens_opt_frame, text=APPEND_ENSEMBLE_NAME_TEXT, width=ENSEMBLE_CHECKBOXS_WIDTH, variable=self.is_append_ensemble_name_var)
|
|
is_append_ensemble_name_Option.grid(pady=0)
|
|
self.help_hints(is_append_ensemble_name_Option, text=IS_APPEND_ENSEMBLE_NAME_HELP)
|
|
|
|
is_wav_ensemble_Option = ttk.Checkbutton(custom_ens_opt_frame, text=ENSEMBLE_WAVFORMS_TEXT, width=ENSEMBLE_CHECKBOXS_WIDTH, variable=self.is_wav_ensemble_var)
|
|
is_wav_ensemble_Option.grid(pady=0)
|
|
self.help_hints(is_wav_ensemble_Option, text=IS_WAV_ENSEMBLE_HELP)
|
|
|
|
ensemble_return_Button = ttk.Button(custom_ens_opt_frame, text=BACK_TO_MAIN_MENU, command=lambda:(self.menu_advanced_ensemble_options_close_window(), self.check_is_menu_settings_open()))
|
|
ensemble_return_Button.grid(pady=MENU_PADDING_1)
|
|
|
|
ensemble_close_Button = ttk.Button(custom_ens_opt_frame, text=CLOSE_WINDOW, command=lambda:self.menu_advanced_ensemble_options_close_window())
|
|
ensemble_close_Button.grid(pady=MENU_PADDING_1)
|
|
|
|
self.deletion_list_fill(delete_entry_Option, option_var, ENSEMBLE_CACHE_DIR, SELECT_SAVED_ENSEMBLE, menu_name='deleteensemble')
|
|
|
|
self.menu_placement(custom_ens_opt, ADVANCED_ENSEMBLE_OPTIONS_TEXT, is_help_hints=True, close_function=self.menu_advanced_ensemble_options_close_window)
|
|
|
|
def menu_advanced_align_options(self):#**
|
|
"""Open Ensemble Custom"""
|
|
|
|
advanced_align_opt = tk.Toplevel()
|
|
|
|
self.is_open_menu_advanced_align_options.set(True)
|
|
self.menu_advanced_align_options_close_window = lambda:(self.is_open_menu_advanced_align_options.set(False), advanced_align_opt.destroy())
|
|
advanced_align_opt.protocol("WM_DELETE_WINDOW", self.menu_advanced_align_options_close_window)
|
|
|
|
advanced_align_opt_frame = self.menu_FRAME_SET(advanced_align_opt)
|
|
advanced_align_opt_frame.grid(row=0)
|
|
|
|
settings_title_Label = self.menu_title_LABEL_SET(advanced_align_opt_frame, ADVANCED_ALIGN_TOOL_OPTIONS_TEXT)
|
|
settings_title_Label.grid(pady=MENU_PADDING_2)
|
|
|
|
phase_option_Label = self.menu_sub_LABEL_SET(advanced_align_opt_frame, SECONDARY_PHASE_TEXT)
|
|
phase_option_Label.grid(pady=4)
|
|
phase_option_Option = ComboBoxMenu(advanced_align_opt_frame, textvariable=self.phase_option_var, values=ALIGN_PHASE_OPTIONS, width=MENU_COMBOBOX_WIDTH)
|
|
phase_option_Option.grid(pady=4)
|
|
self.help_hints(phase_option_Label, text=IS_PHASE_HELP)
|
|
|
|
phase_shifts_Label = self.menu_sub_LABEL_SET(advanced_align_opt_frame, PHASE_SHIFTS_TEXT)
|
|
phase_shifts_Label.grid(pady=4)#
|
|
phase_shifts_Option = ComboBoxMenu(advanced_align_opt_frame, textvariable=self.phase_shifts_var, values=list(PHASE_SHIFTS_OPT.keys()), width=MENU_COMBOBOX_WIDTH)
|
|
phase_shifts_Option.grid(pady=4)
|
|
self.help_hints(phase_shifts_Label, text=PHASE_SHIFTS_ALIGN_HELP)
|
|
|
|
is_save_align_Option = ttk.Checkbutton(advanced_align_opt_frame, text=SAVE_ALIGNED_TRACK_TEXT, width=MDX_CHECKBOXS_WIDTH, variable=self.is_save_align_var)
|
|
is_save_align_Option.grid(pady=0)
|
|
self.help_hints(is_save_align_Option, text=IS_ALIGN_TRACK_HELP)
|
|
|
|
is_match_silence_Option = ttk.Checkbutton(advanced_align_opt_frame, text=SILENCE_MATCHING_TEXT, width=MDX_CHECKBOXS_WIDTH, variable=self.is_match_silence_var)
|
|
is_match_silence_Option.grid(pady=0)
|
|
self.help_hints(is_match_silence_Option, text=IS_MATCH_SILENCE_HELP)
|
|
|
|
is_spec_match_Option = ttk.Checkbutton(advanced_align_opt_frame, text=SPECTRAL_MATCHING_TEXT, width=MDX_CHECKBOXS_WIDTH, variable=self.is_spec_match_var)
|
|
is_spec_match_Option.grid(pady=0)
|
|
self.help_hints(is_spec_match_Option, text=IS_MATCH_SPEC_HELP)
|
|
|
|
ensemble_return_Button = ttk.Button(advanced_align_opt_frame, text=BACK_TO_MAIN_MENU, command=lambda:(self.menu_advanced_align_options_close_window(), self.check_is_menu_settings_open()))
|
|
ensemble_return_Button.grid(pady=MENU_PADDING_1)
|
|
|
|
ensemble_close_Button = ttk.Button(advanced_align_opt_frame, text=CLOSE_WINDOW, command=lambda:self.menu_advanced_align_options_close_window())
|
|
ensemble_close_Button.grid(pady=MENU_PADDING_1)
|
|
|
|
self.menu_placement(advanced_align_opt, ADVANCED_ALIGN_TOOL_OPTIONS_TEXT, is_help_hints=True, close_function=self.menu_advanced_align_options_close_window)
|
|
|
|
def menu_help(self):#**
|
|
"""Open Help Guide"""
|
|
|
|
help_guide_opt = tk.Toplevel()
|
|
|
|
self.is_open_menu_help.set(True)
|
|
self.menu_help_close_window = lambda:(self.is_open_menu_help.set(False), help_guide_opt.destroy())
|
|
help_guide_opt.protocol("WM_DELETE_WINDOW", self.menu_help_close_window)
|
|
|
|
tabControl = ttk.Notebook(help_guide_opt)
|
|
|
|
tab1 = ttk.Frame(tabControl)
|
|
tab2 = ttk.Frame(tabControl)
|
|
tab3 = ttk.Frame(tabControl)
|
|
tab4 = ttk.Frame(tabControl)
|
|
|
|
tabControl.add(tab1, text ='Credits')
|
|
tabControl.add(tab2, text ='Resources')
|
|
tabControl.add(tab3, text ='Application License & Version Information')
|
|
tabControl.add(tab4, text ='Additional Information')
|
|
|
|
tabControl.pack(expand = 1, fill ="both")
|
|
|
|
tab1.grid_rowconfigure(0, weight=1)
|
|
tab1.grid_columnconfigure(0, weight=1)
|
|
|
|
tab2.grid_rowconfigure(0, weight=1)
|
|
tab2.grid_columnconfigure(0, weight=1)
|
|
|
|
tab3.grid_rowconfigure(0, weight=1)
|
|
tab3.grid_columnconfigure(0, weight=1)
|
|
|
|
tab4.grid_rowconfigure(0, weight=1)
|
|
tab4.grid_columnconfigure(0, weight=1)
|
|
|
|
section_title_Label = lambda place, frame, text, font_size=FONT_SIZE_4: tk.Label(master=frame, text=text,font=(MAIN_FONT_NAME, f"{font_size}", "bold"), justify="center", fg="#F4F4F4").grid(row=place,column=0,padx=0,pady=MENU_PADDING_4)
|
|
description_Label = lambda place, frame, text, font=FONT_SIZE_2: tk.Label(master=frame, text=text, font=(MAIN_FONT_NAME, f"{font}"), justify="center", fg="#F6F6F7").grid(row=place,column=0,padx=0,pady=MENU_PADDING_4)
|
|
|
|
def credit_label(place, frame, text, link=None, message=None, is_link=False, is_top=False):
|
|
if is_top:
|
|
thank = tk.Label(master=frame, text=text, font=(MAIN_FONT_NAME, f"{FONT_SIZE_3}", "bold"), justify="center", fg="#13849f")
|
|
else:
|
|
thank = tk.Label(master=frame, text=text, font=(MAIN_FONT_NAME, f"{FONT_SIZE_3}", "underline" if is_link else "normal"), justify="center", fg="#13849f")
|
|
thank.configure(cursor="hand2") if is_link else None
|
|
thank.grid(row=place,column=0,padx=0,pady=1)
|
|
if link:
|
|
thank.bind("<Button-1>", lambda e:webbrowser.open_new_tab(link))
|
|
if message:
|
|
description_Label(place+1, frame, message)
|
|
|
|
def Link(place, frame, text, link, description, font=FONT_SIZE_2):
|
|
link_label = tk.Label(master=frame, text=text, font=(MAIN_FONT_NAME, f"{FONT_SIZE_4}", "underline"), foreground=FG_COLOR, justify="center", cursor="hand2")
|
|
link_label.grid(row=place,column=0,padx=0,pady=MENU_PADDING_1)
|
|
link_label.bind("<Button-1>", lambda e:webbrowser.open_new_tab(link))
|
|
description_Label(place+1, frame, description, font=font)
|
|
|
|
def right_click_menu(event):
|
|
right_click_menu = tk.Menu(self, font=(MAIN_FONT_NAME, FONT_SIZE_1), tearoff=0)
|
|
right_click_menu.add_command(label='Return to Settings Menu', command=lambda:(self.menu_help_close_window(), self.check_is_menu_settings_open()))
|
|
right_click_menu.add_command(label='Exit Window', command=lambda:self.menu_help_close_window())
|
|
|
|
try:
|
|
right_click_menu.tk_popup(event.x_root,event.y_root)
|
|
right_click_release_linux(right_click_menu, help_guide_opt)
|
|
finally:
|
|
right_click_menu.grab_release()
|
|
|
|
help_guide_opt.bind(right_click_button, lambda e:right_click_menu(e))
|
|
credits_Frame = tk.Frame(tab1, highlightthicknes=50)
|
|
credits_Frame.grid(row=0, column=0, padx=0, pady=0)
|
|
tk.Label(credits_Frame, image=self.credits_img).grid(row=1,column=0,padx=0,pady=MENU_PADDING_1)
|
|
|
|
section_title_Label(place=0,
|
|
frame=credits_Frame,
|
|
text="Core UVR Developers")
|
|
|
|
credit_label(place=2,
|
|
frame=credits_Frame,
|
|
text="Anjok07\nAufr33",
|
|
is_top=True)
|
|
|
|
section_title_Label(place=3,
|
|
frame=credits_Frame,
|
|
text="Special Thanks")
|
|
|
|
credit_label(place=6,
|
|
frame=credits_Frame,
|
|
text="Tsurumeso",
|
|
message="Developed the original VR Architecture AI code.",
|
|
link="https://github.com/tsurumeso/vocal-remover",
|
|
is_link=True)
|
|
|
|
credit_label(place=8,
|
|
frame=credits_Frame,
|
|
text="Kuielab & Woosung Choi",
|
|
message="Developed the original MDX-Net AI code.",
|
|
link="https://github.com/kuielab",
|
|
is_link=True)
|
|
|
|
credit_label(place=10,
|
|
frame=credits_Frame,
|
|
text="Adefossez & Demucs",
|
|
message="Core developer of Facebook's Demucs Music Source Separation.",
|
|
link="https://github.com/facebookresearch/demucs",
|
|
is_link=True)
|
|
|
|
credit_label(place=12,
|
|
frame=credits_Frame,
|
|
text="Bas Curtiz",
|
|
message="Designed the official UVR logo, icon, banner, splash screen.")
|
|
|
|
credit_label(place=14,
|
|
frame=credits_Frame,
|
|
text="DilanBoskan",
|
|
message="Your contributions at the start of this project were essential to the success of UVR. Thank you!")
|
|
|
|
credit_label(place=16,
|
|
frame=credits_Frame,
|
|
text="Audio Separation and CC Karaoke & Friends Discord Communities",
|
|
message="Thank you for the support!")
|
|
|
|
more_info_tab_Frame = tk.Frame(tab2, highlightthicknes=30)
|
|
more_info_tab_Frame.grid(row=0,column=0,padx=0,pady=0)
|
|
|
|
section_title_Label(place=3,
|
|
frame=more_info_tab_Frame,
|
|
text="Resources")
|
|
|
|
Link(place=4,
|
|
frame=more_info_tab_Frame,
|
|
text="Ultimate Vocal Remover (Official GitHub)",
|
|
link="https://github.com/Anjok07/ultimatevocalremovergui",
|
|
description="You can find updates, report issues, and give us a shout via our official GitHub.",
|
|
font=FONT_SIZE_1)
|
|
|
|
Link(place=8,
|
|
frame=more_info_tab_Frame,
|
|
text="X-Minus AI",
|
|
link="https://x-minus.pro/ai",
|
|
description="Many of the models provided are also on X-Minus.\n" + \
|
|
"X-Minus benefits users without the computing resources to run the GUI or models locally.",
|
|
font=FONT_SIZE_1)
|
|
|
|
Link(place=12,
|
|
frame=more_info_tab_Frame,
|
|
text="MVSep",
|
|
link="https://mvsep.com/quality_checker/leaderboard.php",
|
|
description="Some of our models are also on MVSep.\n" + \
|
|
"Click the link above for a list of some of the best settings \nand model combinations recorded by fellow UVR users.\nSpecial thanks to ZFTurbo for all his work on MVSep!",
|
|
font=FONT_SIZE_1)
|
|
|
|
Link(place=18,
|
|
frame=more_info_tab_Frame,
|
|
text="FFmpeg",
|
|
link="https://www.wikihow.com/Install-FFmpeg-on-Windows",
|
|
description="UVR relies on FFmpeg for processing non-wav audio files.\n" + \
|
|
"If you are missing FFmpeg, please see the installation guide via the link provided.",
|
|
font=FONT_SIZE_1)
|
|
|
|
Link(place=22,
|
|
frame=more_info_tab_Frame,
|
|
text="Rubber Band Library",
|
|
link="https://breakfastquay.com/rubberband/",
|
|
description="UVR uses the Rubber Band library for the sound stretch and pitch shift tool.\n" + \
|
|
"You can get more information on it via the link provided.",
|
|
font=FONT_SIZE_1)
|
|
|
|
Link(place=26,
|
|
frame=more_info_tab_Frame,
|
|
text="Matchering",
|
|
link="https://github.com/sergree/matchering",
|
|
description="UVR uses the Matchering library for the \"Matchering\" Audio Tool.\n" + \
|
|
"You can get more information on it via the link provided.",
|
|
font=FONT_SIZE_1)
|
|
|
|
Link(place=30,
|
|
frame=more_info_tab_Frame,
|
|
text="Official UVR BMAC",
|
|
link=DONATE_LINK_BMAC,
|
|
description="If you wish to support and donate to this project, click the link above!",
|
|
font=FONT_SIZE_1)
|
|
|
|
appplication_license_tab_Frame = tk.Frame(tab3)
|
|
appplication_license_tab_Frame.grid(row=0,column=0,padx=0,pady=0)
|
|
|
|
appplication_license_Label = tk.Label(appplication_license_tab_Frame, text='UVR License Information', font=(MAIN_FONT_NAME, f"{FONT_SIZE_6}", "bold"), justify="center", fg="#f4f4f4")
|
|
appplication_license_Label.grid(row=0,column=0,padx=0,pady=25)
|
|
|
|
appplication_license_Text = tk.Text(appplication_license_tab_Frame, font=(MAIN_FONT_NAME, f"{FONT_SIZE_4}"), fg="white", bg="black", width=72, wrap=tk.WORD, borderwidth=0)
|
|
appplication_license_Text.grid(row=1,column=0,padx=0,pady=0)
|
|
appplication_license_Text_scroll = ttk.Scrollbar(appplication_license_tab_Frame, orient=tk.VERTICAL)
|
|
appplication_license_Text.config(yscrollcommand=appplication_license_Text_scroll.set)
|
|
appplication_license_Text_scroll.configure(command=appplication_license_Text.yview)
|
|
appplication_license_Text.grid(row=4,sticky=tk.W)
|
|
appplication_license_Text_scroll.grid(row=4, column=1, sticky=tk.NS)
|
|
appplication_license_Text.insert("insert", LICENSE_TEXT(VERSION, current_patch))
|
|
appplication_license_Text.configure(state=tk.DISABLED)
|
|
|
|
application_change_log_tab_Frame = tk.Frame(tab4)
|
|
application_change_log_tab_Frame.grid(row=0,column=0,padx=0,pady=0)
|
|
|
|
application_change_log_Label = tk.Label(application_change_log_tab_Frame, text='Additional Information', font=(MAIN_FONT_NAME, f"{FONT_SIZE_6}", "bold"), justify="center", fg="#f4f4f4")
|
|
application_change_log_Label.grid(row=0,column=0,padx=0,pady=25)
|
|
|
|
application_change_log_Text = tk.Text(application_change_log_tab_Frame, font=(MAIN_FONT_NAME, f"{FONT_SIZE_4}"), fg="white", bg="black", width=72, wrap=tk.WORD, borderwidth=0)
|
|
application_change_log_Text.grid(row=1,column=0,padx=40 if is_macos else 30,pady=0)
|
|
application_change_log_Text_scroll = ttk.Scrollbar(application_change_log_tab_Frame, orient=tk.VERTICAL)
|
|
application_change_log_Text.config(yscrollcommand=application_change_log_Text_scroll.set)
|
|
application_change_log_Text_scroll.configure(command=application_change_log_Text.yview)
|
|
application_change_log_Text.grid(row=4,sticky=tk.W)
|
|
application_change_log_Text_scroll.grid(row=4, column=1, sticky=tk.NS)
|
|
application_change_log_Text.insert("insert", self.bulletin_data)
|
|
auto_hyperlink(application_change_log_Text)
|
|
application_change_log_Text.configure(state=tk.DISABLED)
|
|
|
|
self.menu_placement(help_guide_opt, "Information Guide")
|
|
|
|
def menu_error_log(self):#
|
|
"""Open Error Log"""
|
|
|
|
self.is_confirm_error_var.set(False)
|
|
|
|
copied_var = tk.StringVar(value='')
|
|
error_log_screen = tk.Toplevel()
|
|
|
|
self.is_open_menu_error_log.set(True)
|
|
self.menu_error_log_close_window = lambda:(self.is_open_menu_error_log.set(False), error_log_screen.destroy())
|
|
error_log_screen.protocol("WM_DELETE_WINDOW", self.menu_error_log_close_window)
|
|
|
|
error_log_frame = self.menu_FRAME_SET(error_log_screen)
|
|
error_log_frame.grid(row=0)
|
|
|
|
error_consol_title_Label = self.menu_title_LABEL_SET(error_log_frame, ERROR_CONSOLE_TEXT)
|
|
error_consol_title_Label.grid(row=1,column=0,padx=20,pady=MENU_PADDING_2)
|
|
|
|
error_details_Text = tk.Text(error_log_frame, font=(MAIN_FONT_NAME, f"{FONT_SIZE_1}"), fg="#D37B7B", bg="black", width=110, wrap=tk.WORD, borderwidth=0)
|
|
error_details_Text.grid(row=2,column=0,padx=0,pady=0)
|
|
error_details_Text.insert("insert", self.error_log_var.get())
|
|
error_details_Text.bind(right_click_button, lambda e:self.right_click_menu_popup(e, text_box=True))
|
|
self.current_text_box = error_details_Text
|
|
error_details_Text_scroll = ttk.Scrollbar(error_log_frame, orient=tk.VERTICAL)
|
|
error_details_Text.config(yscrollcommand=error_details_Text_scroll.set)
|
|
error_details_Text_scroll.configure(command=error_details_Text.yview)
|
|
error_details_Text.grid(row=2,sticky=tk.W)
|
|
error_details_Text_scroll.grid(row=2, column=1, sticky=tk.NS)
|
|
|
|
copy_text_Label = tk.Label(error_log_frame, textvariable=copied_var, font=(MAIN_FONT_NAME, f"{FONT_SIZE_0}"), justify="center", fg="#f4f4f4")
|
|
copy_text_Label.grid(padx=20,pady=0)
|
|
|
|
copy_text_Button = ttk.Button(error_log_frame, text=COPY_ALL_TEXT_TEXT, width=14, command=lambda:(pyperclip.copy(error_details_Text.get(1.0, tk.END+"-1c")), copied_var.set('Copied!')))
|
|
copy_text_Button.grid(padx=20,pady=MENU_PADDING_1)
|
|
|
|
report_issue_Button = ttk.Button(error_log_frame, text=REPORT_ISSUE_TEXT, width=14, command=lambda:webbrowser.open_new_tab(ISSUE_LINK))
|
|
report_issue_Button.grid(padx=20,pady=MENU_PADDING_1)
|
|
|
|
error_log_return_Button = ttk.Button(error_log_frame, text=BACK_TO_MAIN_MENU, command=lambda:(self.menu_error_log_close_window(), self.menu_settings()))
|
|
error_log_return_Button.grid(padx=20,pady=MENU_PADDING_1)
|
|
|
|
error_log_close_Button = ttk.Button(error_log_frame, text=CLOSE_WINDOW, command=lambda:self.menu_error_log_close_window())
|
|
error_log_close_Button.grid(padx=20,pady=MENU_PADDING_1)
|
|
|
|
self.menu_placement(error_log_screen, UVR_ERROR_LOG_TEXT)
|
|
|
|
def menu_secondary_model(self, tab, ai_network_vars: dict):
|
|
|
|
#Settings Tab 1
|
|
secondary_model_Frame = self.menu_FRAME_SET(tab)
|
|
secondary_model_Frame.grid(row=0)
|
|
|
|
settings_title_Label = self.menu_title_LABEL_SET(secondary_model_Frame, SECONDARY_MODEL_TEXT)
|
|
settings_title_Label.grid(row=0,column=0,padx=0,pady=MENU_PADDING_3)
|
|
|
|
voc_inst_list = self.model_list(VOCAL_STEM, INST_STEM, is_dry_check=True)
|
|
other_list = self.model_list(OTHER_STEM, NO_OTHER_STEM, is_dry_check=True)
|
|
bass_list = self.model_list(BASS_STEM, NO_BASS_STEM, is_dry_check=True)
|
|
drum_list = self.model_list(DRUM_STEM, NO_DRUM_STEM, is_dry_check=True)
|
|
|
|
voc_inst_secondary_model_var = ai_network_vars["voc_inst_secondary_model"]
|
|
other_secondary_model_var = ai_network_vars["other_secondary_model"]
|
|
bass_secondary_model_var = ai_network_vars["bass_secondary_model"]
|
|
drums_secondary_model_var = ai_network_vars["drums_secondary_model"]
|
|
voc_inst_secondary_model_scale_var = ai_network_vars['voc_inst_secondary_model_scale']
|
|
other_secondary_model_scale_var = ai_network_vars['other_secondary_model_scale']
|
|
bass_secondary_model_scale_var = ai_network_vars['bass_secondary_model_scale']
|
|
drums_secondary_model_scale_var = ai_network_vars['drums_secondary_model_scale']
|
|
is_secondary_model_activate_var = ai_network_vars["is_secondary_model_activate"]
|
|
|
|
change_state_lambda = lambda:change_state(tk.NORMAL if is_secondary_model_activate_var.get() else tk.DISABLED)
|
|
init_convert_to_percentage = lambda raw_value:f"{int(float(raw_value)*100)}%"
|
|
|
|
voc_inst_secondary_model_scale_LABEL_var = tk.StringVar(value=init_convert_to_percentage(voc_inst_secondary_model_scale_var.get()))
|
|
other_secondary_model_scale_LABEL_var = tk.StringVar(value=init_convert_to_percentage(other_secondary_model_scale_var.get()))
|
|
bass_secondary_model_scale_LABEL_var = tk.StringVar(value=init_convert_to_percentage(bass_secondary_model_scale_var.get()))
|
|
drums_secondary_model_scale_LABEL_var = tk.StringVar(value=init_convert_to_percentage(drums_secondary_model_scale_var.get()))
|
|
|
|
def change_state(change_state):
|
|
for child_widget in secondary_model_Frame.winfo_children():
|
|
if type(child_widget) is ComboBoxMenu:
|
|
change_state = READ_ONLY if change_state == tk.NORMAL else change_state
|
|
child_widget.configure(state=change_state)
|
|
elif type(child_widget) is ttk.Scale:
|
|
child_widget.configure(state=change_state)
|
|
|
|
def convert_to_percentage(raw_value, scale_var: tk.StringVar, label_var: tk.StringVar):
|
|
raw_value = '%0.2f' % float(raw_value)
|
|
scale_var.set(raw_value)
|
|
label_var.set(f"{int(float(raw_value)*100)}%")
|
|
|
|
def build_widgets(stem_pair: str, model_list: list, option_var: tk.StringVar, label_var: tk.StringVar, scale_var: tk.DoubleVar):
|
|
model_list.insert(0, NO_MODEL)
|
|
secondary_model_Label = self.menu_sub_LABEL_SET(secondary_model_Frame, f'{stem_pair}', font_size=FONT_SIZE_3)
|
|
secondary_model_Label.grid(pady=MENU_PADDING_1)
|
|
secondary_model_Option = ComboBoxMenu(secondary_model_Frame, textvariable=option_var, values=model_list, dropdown_name=stem_pair, offset=310, width=READ_ONLY_COMBO_WIDTH)
|
|
secondary_model_Option.grid(pady=MENU_PADDING_1)
|
|
secondary_scale_info_Label = tk.Label(secondary_model_Frame, textvariable=label_var, font=(MAIN_FONT_NAME, f"{FONT_SIZE_1}"), foreground=FG_COLOR)
|
|
secondary_scale_info_Label.grid(pady=0)
|
|
secondary_model_scale_Option = ttk.Scale(secondary_model_Frame, variable=scale_var, from_=0.01, to=0.99, command=lambda s:convert_to_percentage(s, scale_var, label_var), orient='horizontal')
|
|
secondary_model_scale_Option.grid(pady=2)
|
|
self.help_hints(secondary_model_Label, text=SECONDARY_MODEL_HELP)
|
|
self.help_hints(secondary_scale_info_Label, text=SECONDARY_MODEL_SCALE_HELP)
|
|
|
|
build_widgets(stem_pair=VOCAL_PAIR,
|
|
model_list=voc_inst_list,
|
|
option_var=voc_inst_secondary_model_var,
|
|
label_var=voc_inst_secondary_model_scale_LABEL_var,
|
|
scale_var=voc_inst_secondary_model_scale_var)
|
|
|
|
build_widgets(stem_pair=OTHER_PAIR,
|
|
model_list=other_list,
|
|
option_var=other_secondary_model_var,
|
|
label_var=other_secondary_model_scale_LABEL_var,
|
|
scale_var=other_secondary_model_scale_var)
|
|
|
|
build_widgets(stem_pair=BASS_PAIR,
|
|
model_list=bass_list,
|
|
option_var=bass_secondary_model_var,
|
|
label_var=bass_secondary_model_scale_LABEL_var,
|
|
scale_var=bass_secondary_model_scale_var)
|
|
|
|
build_widgets(stem_pair=DRUM_PAIR,
|
|
model_list=drum_list,
|
|
option_var=drums_secondary_model_var,
|
|
label_var=drums_secondary_model_scale_LABEL_var,
|
|
scale_var=drums_secondary_model_scale_var)
|
|
|
|
is_secondary_model_activate_Option = ttk.Checkbutton(secondary_model_Frame, text=ACTIVATE_SECONDARY_MODEL_TEXT, variable=is_secondary_model_activate_var, command=change_state_lambda)
|
|
is_secondary_model_activate_Option.grid(row=21,pady=MENU_PADDING_1)
|
|
self.help_hints(is_secondary_model_activate_Option, text=SECONDARY_MODEL_ACTIVATE_HELP)
|
|
|
|
change_state_lambda()
|
|
|
|
self.change_state_lambda = change_state_lambda
|
|
|
|
def menu_preproc_model(self, tab):
|
|
|
|
preproc_model_Frame = self.menu_FRAME_SET(tab)
|
|
preproc_model_Frame.grid(row=0)
|
|
|
|
demucs_pre_proc_model_title_Label = self.menu_title_LABEL_SET(preproc_model_Frame, PREPROCESS_MODEL_CHOOSE_TEXT)
|
|
demucs_pre_proc_model_title_Label.grid(pady=MENU_PADDING_3)
|
|
|
|
pre_proc_list = self.model_list(VOCAL_STEM, INST_STEM, is_dry_check=True, is_no_demucs=True)
|
|
pre_proc_list.insert(0, NO_MODEL)
|
|
|
|
enable_pre_proc_model = lambda:(is_demucs_pre_proc_model_inst_mix_Option.configure(state=tk.NORMAL), demucs_pre_proc_model_Option.configure(state=READ_ONLY))
|
|
disable_pre_proc_model = lambda:(is_demucs_pre_proc_model_inst_mix_Option.configure(state=tk.DISABLED), demucs_pre_proc_model_Option.configure(state=tk.DISABLED), self.is_demucs_pre_proc_model_inst_mix_var.set(False))
|
|
pre_proc_model_toggle = lambda:enable_pre_proc_model() if self.is_demucs_pre_proc_model_activate_var.get() else disable_pre_proc_model()
|
|
|
|
demucs_pre_proc_model_Label = self.menu_sub_LABEL_SET(preproc_model_Frame, SELECT_MODEL_TEXT, font_size=FONT_SIZE_3)
|
|
demucs_pre_proc_model_Label.grid()
|
|
demucs_pre_proc_model_Option = ComboBoxMenu(preproc_model_Frame, textvariable=self.demucs_pre_proc_model_var, values=pre_proc_list, dropdown_name='demucspre', offset=310, width=READ_ONLY_COMBO_WIDTH)
|
|
demucs_pre_proc_model_Option.grid(pady=MENU_PADDING_2)
|
|
|
|
is_demucs_pre_proc_model_inst_mix_Option = ttk.Checkbutton(preproc_model_Frame, text='Save Instrumental Mixture', width=DEMUCS_PRE_CHECKBOXS_WIDTH, variable=self.is_demucs_pre_proc_model_inst_mix_var)
|
|
is_demucs_pre_proc_model_inst_mix_Option.grid()
|
|
self.help_hints(is_demucs_pre_proc_model_inst_mix_Option, text=PRE_PROC_MODEL_INST_MIX_HELP)
|
|
|
|
is_demucs_pre_proc_model_activate_Option = ttk.Checkbutton(preproc_model_Frame, text=ACTIVATE_PRE_PROCESS_MODEL_TEXT, width=DEMUCS_PRE_CHECKBOXS_WIDTH, variable=self.is_demucs_pre_proc_model_activate_var, command=pre_proc_model_toggle)
|
|
is_demucs_pre_proc_model_activate_Option.grid()
|
|
self.help_hints(is_demucs_pre_proc_model_activate_Option, text=PRE_PROC_MODEL_ACTIVATE_HELP)
|
|
|
|
pre_proc_model_toggle()
|
|
|
|
def menu_manual_downloads(self):
|
|
|
|
manual_downloads_menu = tk.Toplevel()
|
|
model_selection_var = tk.StringVar(value=SELECT_MODEL_TEXT)
|
|
#info_text_var = tk.StringVar(value='')
|
|
|
|
if self.is_online:
|
|
model_data = self.online_data
|
|
|
|
# Save the data as a JSON file
|
|
with open(DOWNLOAD_MODEL_CACHE, 'w') as json_file:
|
|
json.dump(model_data, json_file)
|
|
|
|
else:
|
|
if os.path.isfile(DOWNLOAD_MODEL_CACHE):
|
|
with open(DOWNLOAD_MODEL_CACHE, 'r') as json_file:
|
|
model_data = json.load(json_file)
|
|
|
|
vr_download_list = model_data["vr_download_list"]
|
|
mdx_download_list = model_data["mdx_download_list"]
|
|
demucs_download_list = model_data["demucs_download_list"]
|
|
mdx_download_list.update(model_data["mdx23c_download_list"])
|
|
|
|
def create_link(link):
|
|
final_link = lambda:webbrowser.open_new_tab(link)
|
|
return final_link
|
|
|
|
def get_links():
|
|
for widgets in manual_downloads_link_Frame.winfo_children():
|
|
widgets.destroy()
|
|
|
|
main_selection = model_selection_var.get()
|
|
|
|
MAIN_ROW = 0
|
|
|
|
self.menu_sub_LABEL_SET(manual_downloads_link_Frame, 'Download Link(s)').grid(row=0,column=0,padx=0,pady=MENU_PADDING_4)
|
|
|
|
if VR_ARCH_TYPE in main_selection:
|
|
main_selection = vr_download_list[main_selection]
|
|
model_dir = VR_MODELS_DIR
|
|
elif MDX_ARCH_TYPE in main_selection or MDX_23_NAME in main_selection:
|
|
if isinstance(mdx_download_list[main_selection], dict):
|
|
main_selection = mdx_download_list[main_selection]
|
|
main_selection = list(main_selection.keys())[0]
|
|
else:
|
|
main_selection = mdx_download_list[main_selection]
|
|
|
|
model_dir = MDX_MODELS_DIR
|
|
|
|
elif DEMUCS_ARCH_TYPE in main_selection:
|
|
model_dir = DEMUCS_NEWER_REPO_DIR if 'v3' in main_selection or 'v4' in main_selection else DEMUCS_MODELS_DIR
|
|
main_selection = demucs_download_list[main_selection]
|
|
|
|
if type(main_selection) is dict:
|
|
for links in main_selection.values():
|
|
MAIN_ROW += 1
|
|
button_text = f" - Item {MAIN_ROW}" if len(main_selection.keys()) >= 2 else ''
|
|
link = create_link(links)
|
|
link_button = ttk.Button(manual_downloads_link_Frame, text=f"Open Link to Model{button_text}", command=link).grid(row=MAIN_ROW,column=0,padx=0,pady=MENU_PADDING_1)
|
|
else:
|
|
link = f"{NORMAL_REPO}{main_selection}"
|
|
link_button = ttk.Button(manual_downloads_link_Frame, text=OPEN_LINK_TO_MODEL_TEXT, command=lambda:webbrowser.open_new_tab(link))
|
|
link_button.grid(row=1,column=0,padx=0,pady=MENU_PADDING_2)
|
|
|
|
self.menu_sub_LABEL_SET(manual_downloads_link_Frame, SELECTED_MODEL_PLACE_PATH_TEXT).grid(row=MAIN_ROW+2,column=0,padx=0,pady=MENU_PADDING_4)
|
|
ttk.Button(manual_downloads_link_Frame, text=OPEN_MODEL_DIRECTORY_TEXT, command=lambda:OPEN_FILE_func(model_dir)).grid(row=MAIN_ROW+3,column=0,padx=0,pady=MENU_PADDING_1)
|
|
|
|
manual_downloads_menu_Frame = self.menu_FRAME_SET(manual_downloads_menu)
|
|
manual_downloads_menu_Frame.grid(row=0)
|
|
|
|
manual_downloads_link_Frame = self.menu_FRAME_SET(manual_downloads_menu, thickness=5)
|
|
manual_downloads_link_Frame.grid(row=1)
|
|
|
|
manual_downloads_menu_title_Label = self.menu_title_LABEL_SET(manual_downloads_menu_Frame, MANUAL_DOWNLOADS_TEXT, width=45)
|
|
manual_downloads_menu_title_Label.grid(row=0,column=0,padx=0,pady=MENU_PADDING_3)
|
|
|
|
manual_downloads_menu_select_Label = self.menu_sub_LABEL_SET(manual_downloads_menu_Frame, SELECT_MODEL_TEXT)
|
|
manual_downloads_menu_select_Label.grid(row=1,column=0,padx=0,pady=MENU_PADDING_1)
|
|
|
|
manual_downloads_menu_select_Option = ttk.OptionMenu(manual_downloads_menu_Frame, model_selection_var)
|
|
manual_downloads_menu_select_VR_Option = tk.Menu(manual_downloads_menu_select_Option['menu'])
|
|
manual_downloads_menu_select_MDX_Option = tk.Menu(manual_downloads_menu_select_Option['menu'])
|
|
manual_downloads_menu_select_DEMUCS_Option = tk.Menu(manual_downloads_menu_select_Option['menu'])
|
|
manual_downloads_menu_select_Option['menu'].add_cascade(label='VR Models', menu= manual_downloads_menu_select_VR_Option)
|
|
manual_downloads_menu_select_Option['menu'].add_cascade(label='MDX-Net Models', menu= manual_downloads_menu_select_MDX_Option)
|
|
manual_downloads_menu_select_Option['menu'].add_cascade(label='Demucs Models', menu= manual_downloads_menu_select_DEMUCS_Option)
|
|
|
|
for model_selection_vr in vr_download_list.keys():
|
|
if not os.path.isfile(os.path.join(VR_MODELS_DIR, vr_download_list[model_selection_vr])):
|
|
manual_downloads_menu_select_VR_Option.add_radiobutton(label=model_selection_vr, variable=model_selection_var, command=get_links)
|
|
|
|
for model_selection_mdx in mdx_download_list.keys():
|
|
|
|
model_name = mdx_download_list[model_selection_mdx]
|
|
|
|
if isinstance(model_name, dict):
|
|
items_list = list(model_name.items())
|
|
model_name, config = items_list[0]
|
|
config_link = f"{MDX23_CONFIG_CHECKS}{config}"
|
|
config_local = os.path.join(MDX_C_CONFIG_PATH, config)
|
|
if not os.path.isfile(config_local):
|
|
try:
|
|
with urllib.request.urlopen(config_link) as response:
|
|
with open(config_local, 'wb') as out_file:
|
|
out_file.write(response.read())
|
|
except Exception as e:
|
|
model_name = None
|
|
|
|
#print(model_name)
|
|
|
|
if model_name:
|
|
if not os.path.isfile(os.path.join(MDX_MODELS_DIR, model_name)):
|
|
manual_downloads_menu_select_MDX_Option.add_radiobutton(label=model_selection_mdx, variable=model_selection_var, command=get_links)
|
|
|
|
for model_selection_demucs in demucs_download_list.keys():
|
|
manual_downloads_menu_select_DEMUCS_Option.add_radiobutton(label=model_selection_demucs, variable=model_selection_var, command=get_links)
|
|
|
|
manual_downloads_menu_select_Option.grid(row=2,column=0,padx=0,pady=MENU_PADDING_1)
|
|
|
|
self.menu_placement(manual_downloads_menu, MANUAL_DOWNLOADS_TEXT, pop_up=True, close_function=lambda:manual_downloads_menu.destroy())
|
|
|
|
def invalid_tooltip(self, widget, pattern=None):
|
|
tooltip = ToolTip(widget)
|
|
invalid_message = lambda:tooltip.showtip(INVALID_INPUT_E, True)
|
|
|
|
def invalid_message_():
|
|
tooltip.showtip(INVALID_INPUT_E, True)
|
|
|
|
def validation(value):
|
|
if re.fullmatch(modified_pattern, value) is None:
|
|
return False
|
|
else:
|
|
return True
|
|
|
|
if not pattern:
|
|
pattern = r'^[a-zA-Z0-9 -]{0,25}$'
|
|
|
|
modified_pattern = f"({pattern}|)"
|
|
|
|
widget.configure(
|
|
validate='key',
|
|
validatecommand=(self.register(validation), '%P'),
|
|
invalidcommand=(self.register(invalid_message))
|
|
)
|
|
|
|
return invalid_message_
|
|
|
|
def pop_up_save_current_settings(self):
|
|
"""Save current application settings as..."""
|
|
|
|
settings_save = tk.Toplevel(root)
|
|
|
|
settings_save_var = tk.StringVar(value='')
|
|
|
|
settings_save_Frame = self.menu_FRAME_SET(settings_save)
|
|
settings_save_Frame.grid(row=1)
|
|
|
|
save_func = lambda:(self.pop_up_save_current_settings_sub_json_dump(settings_save_var.get()), settings_save.destroy())
|
|
validation = lambda value:False if re.fullmatch(REG_SAVE_INPUT, value) is None else True
|
|
|
|
settings_save_title = self.menu_title_LABEL_SET(settings_save_Frame, SAVE_CURRENT_SETTINGS_TEXT)
|
|
settings_save_title.grid()
|
|
|
|
settings_save_name_Label = self.menu_sub_LABEL_SET(settings_save_Frame, NAME_SETTINGS_TEXT)
|
|
settings_save_name_Label.grid(pady=MENU_PADDING_1)
|
|
settings_save_name_Entry = ttk.Entry(settings_save_Frame, textvariable=settings_save_var, justify='center', width=25)
|
|
settings_save_name_Entry.grid(pady=MENU_PADDING_1)
|
|
invalid_message = self.invalid_tooltip(settings_save_name_Entry)
|
|
settings_save_name_Entry.bind(right_click_button, self.right_click_menu_popup)
|
|
self.current_text_box = settings_save_name_Entry
|
|
settings_save_name_Entry.focus_set()
|
|
|
|
self.spacer_label(settings_save_Frame)
|
|
|
|
entry_rules_Label = tk.Label(settings_save_Frame, text=ENSEMBLE_INPUT_RULE, font=(MAIN_FONT_NAME, f"{FONT_SIZE_1}"), foreground='#868687', justify="left")
|
|
entry_rules_Label.grid()
|
|
|
|
settings_save_Button = ttk.Button(settings_save_Frame, text=SAVE_TEXT, command=lambda:save_func() if validation(settings_save_var.get()) else invalid_message())
|
|
settings_save_Button.grid(pady=MENU_PADDING_1)
|
|
|
|
stop_process_Button = ttk.Button(settings_save_Frame, text=CANCEL_TEXT, command=lambda:settings_save.destroy())
|
|
stop_process_Button.grid(pady=MENU_PADDING_1)
|
|
|
|
self.menu_placement(settings_save, SAVE_CURRENT_SETTINGS_TEXT, pop_up=True)
|
|
|
|
def pop_up_save_current_settings_sub_json_dump(self, settings_save_name: str):
|
|
"""Dumps current application settings to a json named after user input"""
|
|
|
|
if settings_save_name:
|
|
self.save_current_settings_var.set(settings_save_name)
|
|
settings_save_name = settings_save_name.replace(" ", "_")
|
|
current_settings = self.save_values(app_close=False)
|
|
|
|
saved_data_dump = json.dumps(current_settings, indent=4)
|
|
with open(os.path.join(SETTINGS_CACHE_DIR, f'{settings_save_name}.json'), "w") as outfile:
|
|
outfile.write(saved_data_dump)
|
|
|
|
def pop_up_update_confirmation(self):
|
|
"""Ask user is they want to update"""
|
|
|
|
is_new_update = self.online_data_refresh(confirmation_box=True)
|
|
is_download_in_app_var = tk.BooleanVar(value=False)
|
|
|
|
def update_type():
|
|
if is_download_in_app_var.get():
|
|
self.download_item(is_update_app=True)
|
|
else:
|
|
webbrowser.open_new_tab(self.download_update_link_var.get())
|
|
|
|
update_confirmation_win.destroy()
|
|
|
|
if is_new_update:
|
|
|
|
update_confirmation_win = tk.Toplevel()
|
|
|
|
update_confirmation_Frame = self.menu_FRAME_SET(update_confirmation_win)
|
|
update_confirmation_Frame.grid(row=0)
|
|
|
|
update_found_label = self.menu_title_LABEL_SET(update_confirmation_Frame, UPDATE_FOUND_TEXT, width=15)
|
|
update_found_label.grid(row=0,column=0,padx=0,pady=MENU_PADDING_2)
|
|
|
|
confirm_update_label = self.menu_sub_LABEL_SET(update_confirmation_Frame, UPDATE_CONFIRMATION_TEXT, font_size=FONT_SIZE_3)
|
|
confirm_update_label.grid(row=1,column=0,padx=0,pady=MENU_PADDING_1)
|
|
|
|
yes_button = ttk.Button(update_confirmation_Frame, text=YES_TEXT, command=update_type)
|
|
yes_button.grid(row=2,column=0,padx=0,pady=MENU_PADDING_1)
|
|
|
|
no_button = ttk.Button(update_confirmation_Frame, text=NO_TEXT, command=lambda:(update_confirmation_win.destroy()))
|
|
no_button.grid(row=3,column=0,padx=0,pady=MENU_PADDING_1)
|
|
|
|
if is_windows:
|
|
download_outside_application_button = ttk.Checkbutton(update_confirmation_Frame, variable=is_download_in_app_var, text='Download Update in Application')
|
|
download_outside_application_button.grid(row=4,column=0,padx=0,pady=MENU_PADDING_1)
|
|
|
|
self.menu_placement(update_confirmation_win, CONFIRM_UPDATE_TEXT, pop_up=True)
|
|
|
|
def pop_up_user_code_input(self):
|
|
"""Input VIP Code"""
|
|
|
|
self.user_code_validation_var.set('')
|
|
|
|
self.user_code = tk.Toplevel()
|
|
|
|
user_code_Frame = self.menu_FRAME_SET(self.user_code)
|
|
user_code_Frame.grid(row=0)
|
|
|
|
user_code_title_Label = self.menu_title_LABEL_SET(user_code_Frame, USER_DOWNLOAD_CODES_TEXT, width=20)
|
|
user_code_title_Label.grid(row=0,column=0,padx=0,pady=MENU_PADDING_1)
|
|
|
|
user_code_Label = self.menu_sub_LABEL_SET(user_code_Frame, DOWNLOAD_CODE_TEXT)
|
|
user_code_Label.grid(pady=MENU_PADDING_1)
|
|
|
|
self.user_code_Entry = ttk.Entry(user_code_Frame, textvariable=self.user_code_var, justify='center')
|
|
self.user_code_Entry.grid(pady=MENU_PADDING_1)
|
|
self.user_code_Entry.bind(right_click_button, self.right_click_menu_popup)
|
|
self.current_text_box = self.user_code_Entry
|
|
|
|
tooltip = ToolTip(self.user_code_Entry)
|
|
def invalid_message_(text, is_success_message):
|
|
tooltip.hidetip()
|
|
tooltip.showtip(text, True, is_success_message)
|
|
|
|
self.spacer_label(user_code_Frame)
|
|
|
|
user_code_confrim_Button = ttk.Button(user_code_Frame, text=CONFIRM_TEXT, command=lambda:self.download_validate_code(confirm=True, code_message=invalid_message_))
|
|
user_code_confrim_Button.grid(pady=MENU_PADDING_1)
|
|
|
|
user_code_cancel_Button = ttk.Button(user_code_Frame, text=CANCEL_TEXT, command=lambda:self.user_code.destroy())
|
|
user_code_cancel_Button.grid(pady=MENU_PADDING_1)
|
|
|
|
support_title_Label = self.menu_title_LABEL_SET(user_code_Frame, text=SUPPORT_UVR_TEXT, width=20)
|
|
support_title_Label.grid(pady=MENU_PADDING_1)
|
|
|
|
support_sub_Label = tk.Label(user_code_Frame, text=GET_DL_VIP_CODE_TEXT, font=(MAIN_FONT_NAME, f"{FONT_SIZE_1}"), foreground=FG_COLOR)
|
|
support_sub_Label.grid(pady=MENU_PADDING_1)
|
|
|
|
uvr_patreon_Button = ttk.Button(user_code_Frame, text=UVR_PATREON_LINK_TEXT, command=lambda:webbrowser.open_new_tab(DONATE_LINK_PATREON))
|
|
uvr_patreon_Button.grid(pady=MENU_PADDING_1)
|
|
|
|
bmac_patreon_Button=ttk.Button(user_code_Frame, text=BMAC_UVR_TEXT, command=lambda:webbrowser.open_new_tab(DONATE_LINK_BMAC))
|
|
bmac_patreon_Button.grid(pady=MENU_PADDING_1)
|
|
|
|
self.menu_placement(self.user_code, INPUT_CODE_TEXT, pop_up=True)
|
|
|
|
def pop_up_change_model_defaults(self, top_window):
|
|
"""
|
|
Change model defaults...
|
|
"""
|
|
|
|
def message_box_(text, is_success_message):
|
|
tooltip.hidetip()
|
|
tooltip.showtip(text, True, is_success_message)
|
|
|
|
def delete_entry():
|
|
model_data = self.assemble_model_data(model=change_model_defaults_var.get(), arch_type=ENSEMBLE_CHECK, is_change_def=True, is_get_hash_dir_only=True)[0]
|
|
hash_file = model_data.model_hash_dir
|
|
if hash_file:
|
|
if os.path.isfile(hash_file):
|
|
os.remove(hash_file)
|
|
message_box_("Defined Parameters Deleted", True)
|
|
else:
|
|
message_box_("No Defined Parameters Found", False)
|
|
|
|
self.update_checkbox_text()
|
|
|
|
def change_default():
|
|
model_data = self.assemble_model_data(model=change_model_defaults_var.get(), arch_type=ENSEMBLE_CHECK, is_change_def=True)[0]
|
|
if model_data.model_status:
|
|
message_box_("Model Parameters Changed", True)
|
|
self.update_checkbox_text()
|
|
|
|
change_model_defaults = tk.Toplevel(root)
|
|
change_model_defaults_var = tk.StringVar(value=NO_MODEL)
|
|
|
|
default_change_model_list = list(self.default_change_model_list)
|
|
default_change_model_list.insert(0, NO_MODEL)
|
|
|
|
change_model_defaults_Frame = self.menu_FRAME_SET(change_model_defaults)
|
|
change_model_defaults_Frame.grid(row=1)
|
|
|
|
change_model_defaults_title = self.menu_title_LABEL_SET(change_model_defaults_Frame, CHANGE_MODEL_DEFAULT_TEXT)
|
|
change_model_defaults_title.grid()
|
|
|
|
model_param_Label = self.menu_sub_LABEL_SET(change_model_defaults_Frame, SELECT_MODEL_TEXT)
|
|
model_param_Label.grid(pady=MENU_PADDING_1)
|
|
model_param_Option = ComboBoxMenu(change_model_defaults_Frame, dropdown_name='changemodeldefault', textvariable=change_model_defaults_var, values=default_change_model_list, offset=310, width=READ_ONLY_COMBO_WIDTH)
|
|
model_param_Option.grid(pady=MENU_PADDING_1)
|
|
tooltip = ToolTip(model_param_Option)
|
|
|
|
self.spacer_label(change_model_defaults_Frame)
|
|
|
|
change_params_Button = ttk.Button(change_model_defaults_Frame, text=CHANGE_PARAMETERS_TEXT, command=change_default, width=20)
|
|
change_params_Button.grid(pady=MENU_PADDING_1)
|
|
|
|
delete_params_Button = ttk.Button(change_model_defaults_Frame, text=DELETE_PARAMETERS_TEXT, command=delete_entry, width=20)
|
|
delete_params_Button.grid(pady=MENU_PADDING_1)
|
|
|
|
cancel_Button = ttk.Button(change_model_defaults_Frame, text=CANCEL_TEXT, command=lambda:change_model_defaults.destroy())
|
|
cancel_Button.grid(pady=MENU_PADDING_1)
|
|
|
|
self.menu_placement(change_model_defaults, CHANGE_MODEL_DEFAULT_TEXT, top_window=top_window)
|
|
|
|
def pop_up_set_vocal_splitter(self, top_window):
|
|
"""
|
|
Set vocal splitter
|
|
"""
|
|
|
|
try:
|
|
set_vocal_splitter = tk.Toplevel(root)
|
|
|
|
model_list = self.assemble_model_data(arch_type=KARAOKEE_CHECK, is_dry_check=True)
|
|
if not model_list:
|
|
self.set_vocal_splitter_var.set(NO_MODEL)
|
|
model_list.insert(0, NO_MODEL)
|
|
|
|
enable_voc_split_model = lambda:(model_select_Option.configure(state=READ_ONLY), save_inst_Button.configure(state=tk.NORMAL))
|
|
disable_voc_split_model = lambda:(model_select_Option.configure(state=tk.DISABLED), save_inst_Button.configure(state=tk.DISABLED), self.is_save_inst_set_vocal_splitter_var.set(False))
|
|
voc_split_model_toggle = lambda:enable_voc_split_model() if self.is_set_vocal_splitter_var.get() else disable_voc_split_model()
|
|
|
|
enable_deverb_opt = lambda:(deverb_vocals_Option.configure(state=READ_ONLY))
|
|
disable_deverb_opt= lambda:(deverb_vocals_Option.configure(state=tk.DISABLED))
|
|
deverb_opt_toggle = lambda:enable_deverb_opt() if self.is_deverb_vocals_var.get() else disable_deverb_opt()
|
|
|
|
set_vocal_splitter_Frame = self.menu_FRAME_SET(set_vocal_splitter)
|
|
set_vocal_splitter_Frame.grid(row=1)
|
|
|
|
set_vocal_splitter_title = self.menu_title_LABEL_SET(set_vocal_splitter_Frame, VOCAL_SPLIT_MODE_OPTIONS_TEXT)
|
|
set_vocal_splitter_title.grid(pady=MENU_PADDING_2)
|
|
|
|
model_select_Label = self.menu_sub_LABEL_SET(set_vocal_splitter_Frame, SELECT_MODEL_TEXT)
|
|
model_select_Label.grid(pady=MENU_PADDING_1)
|
|
model_select_Option = ComboBoxMenu(set_vocal_splitter_Frame, dropdown_name='setvocalsplit', textvariable=self.set_vocal_splitter_var, values=model_list, offset=310, width=READ_ONLY_COMBO_WIDTH)
|
|
model_select_Option.grid(pady=7)
|
|
self.help_hints(model_select_Option, text=VOC_SPLIT_MODEL_SELECT_HELP)#
|
|
|
|
save_inst_Button = ttk.Checkbutton(set_vocal_splitter_Frame, text=SAVE_SPLIT_VOCAL_INSTRUMENTALS_TEXT, variable=self.is_save_inst_set_vocal_splitter_var, width=SET_VOC_SPLIT_CHECK_WIDTH, command=voc_split_model_toggle)
|
|
save_inst_Button.grid()#
|
|
self.help_hints(save_inst_Button, text=IS_VOC_SPLIT_INST_SAVE_SELECT_HELP)#
|
|
|
|
change_params_Button = ttk.Checkbutton(set_vocal_splitter_Frame, text=ENABLE_VOCAL_SPLIT_MODE_TEXT, variable=self.is_set_vocal_splitter_var, width=SET_VOC_SPLIT_CHECK_WIDTH, command=voc_split_model_toggle)
|
|
change_params_Button.grid()#
|
|
self.help_hints(change_params_Button, text=IS_VOC_SPLIT_MODEL_SELECT_HELP)#
|
|
|
|
set_vocal_splitter_title = self.menu_title_LABEL_SET(set_vocal_splitter_Frame, VOCAL_DEVERB_OPTIONS_TEXT)
|
|
set_vocal_splitter_title.grid(pady=MENU_PADDING_2)
|
|
|
|
deverb_vocals_Label = self.menu_sub_LABEL_SET(set_vocal_splitter_Frame, 'Select Vocal Type to Deverb')
|
|
deverb_vocals_Label.grid(pady=MENU_PADDING_1)
|
|
deverb_vocals_Option = ComboBoxMenu(set_vocal_splitter_Frame, dropdown_name='setvocaldeverb', textvariable=self.deverb_vocal_opt_var, values=list(DEVERB_MAPPER.keys()), width=23)
|
|
deverb_vocals_Option.grid(pady=7)
|
|
self.help_hints(deverb_vocals_Option, text=IS_DEVERB_OPT_HELP)#
|
|
|
|
is_deverb_vocals_Option = ttk.Checkbutton(set_vocal_splitter_Frame, text=DEVERB_VOCALS_TEXT, width=15 if is_windows else 11, variable=self.is_deverb_vocals_var, command=deverb_opt_toggle)
|
|
is_deverb_vocals_Option.grid(pady=0)
|
|
self.help_hints(is_deverb_vocals_Option, text=IS_DEVERB_VOC_HELP)#
|
|
|
|
if not os.path.isfile(DEVERBER_MODEL_PATH):
|
|
self.is_deverb_vocals_var.set(False)
|
|
is_deverb_vocals_Option.configure(state=tk.DISABLED)
|
|
disable_deverb_opt()
|
|
|
|
cancel_Button = ttk.Button(set_vocal_splitter_Frame, text=CLOSE_WINDOW, command=lambda:set_vocal_splitter.destroy(), width=16)
|
|
cancel_Button.grid(pady=MENU_PADDING_3)
|
|
|
|
voc_split_model_toggle()
|
|
deverb_opt_toggle()
|
|
|
|
self.menu_placement(set_vocal_splitter, VOCAL_SPLIT_OPTIONS_TEXT, top_window=top_window, pop_up=True)
|
|
except Exception as e:
|
|
error_name = f'{type(e).__name__}'
|
|
traceback_text = ''.join(traceback.format_tb(e.__traceback__))
|
|
message = f'{error_name}: "{e}"\n{traceback_text}"'
|
|
self.error_log_var.set(message)
|
|
|
|
def pop_up_mdx_model(self, mdx_model_hash, model_path):
|
|
"""Opens MDX-Net model settings"""
|
|
|
|
is_compatible_model = True
|
|
is_ckpt = False
|
|
primary_stem = VOCAL_STEM
|
|
|
|
try:
|
|
if model_path.endswith(ONNX):
|
|
model = onnx.load(model_path)
|
|
model_shapes = [[d.dim_value for d in _input.type.tensor_type.shape.dim] for _input in model.graph.input][0]
|
|
dim_f = model_shapes[2]
|
|
dim_t = int(math.log(model_shapes[3], 2))
|
|
n_fft = '6144'
|
|
|
|
if model_path.endswith(CKPT):
|
|
is_ckpt = True
|
|
model_params = torch.load(model_path, map_location=lambda storage, loc: storage)
|
|
model_params = model_params['hyper_parameters']
|
|
dim_f = model_params['dim_f']
|
|
dim_t = int(math.log(model_params['dim_t'], 2))
|
|
n_fft = model_params['n_fft']
|
|
|
|
for stem in STEM_SET_MENU:
|
|
if model_params['target_name'] == stem.lower():
|
|
primary_stem = INST_STEM if model_params['target_name'] == OTHER_STEM.lower() else stem
|
|
|
|
except Exception as e:
|
|
error_name = f'{type(e).__name__}'
|
|
traceback_text = ''.join(traceback.format_tb(e.__traceback__))
|
|
message = f'{error_name}: "{e}"\n{traceback_text}"'
|
|
#self.error_log_var.set(message)
|
|
is_compatible_model = False
|
|
if is_ckpt:
|
|
self.pop_up_mdx_c_param(mdx_model_hash)
|
|
else:
|
|
dim_f = 0
|
|
dim_t = 0
|
|
self.error_dialoge(INVALID_ONNX_MODEL_ERROR)
|
|
self.error_log_var.set("{}".format(error_text('MDX-Net Model Settings', e)))
|
|
self.mdx_model_params = None
|
|
|
|
if is_compatible_model:
|
|
mdx_model_set = tk.Toplevel(root)
|
|
mdx_n_fft_scale_set_var = tk.StringVar(value=n_fft)
|
|
mdx_dim_f_set_var = tk.StringVar(value=dim_f)
|
|
mdx_dim_t_set_var = tk.StringVar(value=dim_t)
|
|
primary_stem_var = tk.StringVar(value=primary_stem)
|
|
mdx_compensate_var = tk.StringVar(value=1.035)
|
|
|
|
balance_value_var = tk.StringVar(value=0)
|
|
is_kara_model_var = tk.BooleanVar(value=False)
|
|
is_bv_model_var = tk.BooleanVar(value=False)
|
|
|
|
def toggle_kara():
|
|
if is_kara_model_var.get():
|
|
is_bv_model_var.set(False)
|
|
balance_value_Option.configure(state=tk.DISABLED)
|
|
|
|
def toggle_bv():
|
|
if is_bv_model_var.get():
|
|
is_kara_model_var.set(False)
|
|
balance_value_Option.configure(state=READ_ONLY)
|
|
else:
|
|
balance_value_Option.configure(state=tk.DISABLED)
|
|
|
|
def opt_menu_selection(selection):
|
|
if not selection in [VOCAL_STEM, INST_STEM]:
|
|
balance_value_Option.configure(state=tk.DISABLED)
|
|
is_kara_model_Option.configure(state=tk.DISABLED)
|
|
is_bv_model_Option.configure(state=tk.DISABLED)
|
|
is_kara_model_var.set(False)
|
|
is_bv_model_var.set(False)
|
|
balance_value_var.set(0)
|
|
else:
|
|
is_kara_model_Option.configure(state=tk.NORMAL)
|
|
is_bv_model_Option.configure(state=tk.NORMAL)
|
|
|
|
mdx_model_set_Frame = self.menu_FRAME_SET(mdx_model_set)
|
|
mdx_model_set_Frame.grid(row=2)
|
|
|
|
mdx_model_set_title = self.menu_title_LABEL_SET(mdx_model_set_Frame, SPECIFY_MDX_NET_MODEL_PARAMETERS_TEXT)
|
|
mdx_model_set_title.grid(pady=MENU_PADDING_3)
|
|
|
|
set_stem_name_Label = self.menu_sub_LABEL_SET(mdx_model_set_Frame, PRIMARY_STEM_TEXT)
|
|
set_stem_name_Label.grid(pady=MENU_PADDING_1)
|
|
set_stem_name_Option = ttk.OptionMenu(mdx_model_set_Frame, primary_stem_var, None, *STEM_SET_MENU, command=opt_menu_selection)
|
|
set_stem_name_Option.configure(width=15)
|
|
set_stem_name_Option.grid(pady=MENU_PADDING_1)
|
|
set_stem_name_Option['menu'].insert_separator(len(STEM_SET_MENU))
|
|
set_stem_name_Option['menu'].add_radiobutton(label=INPUT_STEM_NAME, command=tk._setit(primary_stem_var, INPUT_STEM_NAME, lambda e:self.pop_up_input_stem_name(primary_stem_var, mdx_model_set)))
|
|
self.help_hints(set_stem_name_Label, text=SET_STEM_NAME_HELP)
|
|
|
|
is_kara_model_Option = ttk.Checkbutton(mdx_model_set_Frame, text=KARAOKE_MODEL_TEXT, width=SET_MENUS_CHECK_WIDTH, variable=is_kara_model_var, command=toggle_kara)
|
|
is_kara_model_Option.grid(pady=0)
|
|
|
|
is_bv_model_Option = ttk.Checkbutton(mdx_model_set_Frame, text=BV_MODEL_TEXT, width=SET_MENUS_CHECK_WIDTH, variable=is_bv_model_var, command=toggle_bv)
|
|
is_bv_model_Option.grid(pady=0)
|
|
|
|
balance_value_Label = self.menu_sub_LABEL_SET(mdx_model_set_Frame, BALANCE_VALUE_TEXT)
|
|
balance_value_Label.grid(pady=MENU_PADDING_1)
|
|
balance_value_Option = ComboBoxMenu(mdx_model_set_Frame, textvariable=balance_value_var, values=BALANCE_VALUES, width=COMBO_WIDTH)
|
|
balance_value_Option.configure(state=tk.DISABLED)
|
|
balance_value_Option.grid(pady=MENU_PADDING_1)
|
|
#self.help_hints(balance_value_Label, text=balance_value_HELP)
|
|
|
|
mdx_dim_t_set_Label = self.menu_sub_LABEL_SET(mdx_model_set_Frame, 'Dim_t')
|
|
mdx_dim_t_set_Label.grid(pady=MENU_PADDING_1)
|
|
mdx_dim_f_set_Label = self.menu_sub_LABEL_SET(mdx_model_set_Frame, MDX_MENU_WAR_TEXT)
|
|
mdx_dim_f_set_Label.grid(pady=MENU_PADDING_1)
|
|
mdx_dim_t_set_Option = ComboBoxEditableMenu(mdx_model_set_Frame, values=('7', '8'), textvariable=mdx_dim_t_set_var, pattern=REG_SHIFTS, default=mdx_dim_t_set_var.get(), width=COMBO_WIDTH, is_stay_disabled=is_ckpt)
|
|
mdx_dim_t_set_Option.grid(pady=MENU_PADDING_1)
|
|
self.help_hints(mdx_dim_t_set_Label, text=MDX_DIM_T_SET_HELP)
|
|
|
|
mdx_dim_f_set_Label = self.menu_sub_LABEL_SET(mdx_model_set_Frame, 'Dim_f')
|
|
mdx_dim_f_set_Label.grid(pady=MENU_PADDING_1)
|
|
mdx_dim_f_set_Label = self.menu_sub_LABEL_SET(mdx_model_set_Frame, MDX_MENU_WAR_TEXT)
|
|
mdx_dim_f_set_Label.grid(pady=MENU_PADDING_1)
|
|
mdx_dim_f_set_Option = ComboBoxEditableMenu(mdx_model_set_Frame, values=(MDX_POP_DIMF), textvariable=mdx_dim_f_set_var, pattern=REG_SHIFTS, default=mdx_dim_f_set_var.get(), width=COMBO_WIDTH, is_stay_disabled=is_ckpt)
|
|
mdx_dim_f_set_Option.grid(pady=MENU_PADDING_1)
|
|
self.help_hints(mdx_dim_f_set_Label, text=MDX_DIM_F_SET_HELP)
|
|
|
|
mdx_n_fft_scale_set_Label = self.menu_sub_LABEL_SET(mdx_model_set_Frame, 'N_FFT Scale')
|
|
mdx_n_fft_scale_set_Label.grid(pady=MENU_PADDING_1)
|
|
mdx_n_fft_scale_set_Option = ComboBoxEditableMenu(mdx_model_set_Frame, values=(MDX_POP_NFFT), textvariable=mdx_n_fft_scale_set_var, pattern=REG_SHIFTS, default=mdx_n_fft_scale_set_var.get(), width=COMBO_WIDTH, is_stay_disabled=is_ckpt)
|
|
mdx_n_fft_scale_set_Option.grid(pady=MENU_PADDING_1)
|
|
self.help_hints(mdx_n_fft_scale_set_Label, text=MDX_N_FFT_SCALE_SET_HELP)
|
|
|
|
mdx_compensate_Label = self.menu_sub_LABEL_SET(mdx_model_set_Frame, VOLUME_COMPENSATION_TEXT)
|
|
mdx_compensate_Label.grid(pady=MENU_PADDING_1)
|
|
mdx_compensate_Entry = ComboBoxEditableMenu(mdx_model_set_Frame, values=('1.035', '1.08'), textvariable=mdx_compensate_var, pattern=REG_VOL_COMP, default=mdx_compensate_var.get(), width=COMBO_WIDTH)
|
|
mdx_compensate_Entry.grid(pady=MENU_PADDING_1)
|
|
self.help_hints(mdx_compensate_Label, text=POPUP_COMPENSATE_HELP)
|
|
|
|
mdx_param_set_Button = ttk.Button(mdx_model_set_Frame, text=CONFIRM_TEXT, command=lambda:pull_data())
|
|
mdx_param_set_Button.grid(pady=MENU_PADDING_2)
|
|
|
|
stop_process_Button = ttk.Button(mdx_model_set_Frame, text=CANCEL_TEXT, command=lambda:cancel())
|
|
stop_process_Button.grid(pady=0)
|
|
|
|
if is_ckpt:
|
|
mdx_dim_t_set_Option.configure(state=tk.DISABLED)
|
|
mdx_dim_f_set_Option.configure(state=tk.DISABLED)
|
|
mdx_n_fft_scale_set_Option.configure(state=tk.DISABLED)
|
|
|
|
def pull_data():
|
|
mdx_model_params = {
|
|
'compensate': float(mdx_compensate_var.get()),
|
|
'mdx_dim_f_set': int(mdx_dim_f_set_var.get()),
|
|
'mdx_dim_t_set': int(mdx_dim_t_set_var.get()),
|
|
'mdx_n_fft_scale_set': int(mdx_n_fft_scale_set_var.get()),
|
|
'primary_stem': primary_stem_var.get(),
|
|
IS_KARAOKEE: bool(is_kara_model_var.get()),
|
|
IS_BV_MODEL: bool(is_bv_model_var.get()),
|
|
IS_BV_MODEL_REBAL: float(balance_value_var.get())
|
|
}
|
|
|
|
self.pop_up_mdx_model_sub_json_dump(mdx_model_params, mdx_model_hash)
|
|
mdx_model_set.destroy()
|
|
|
|
def cancel():
|
|
mdx_model_set.destroy()
|
|
|
|
mdx_model_set.protocol("WM_DELETE_WINDOW", cancel)
|
|
|
|
frame_list = [mdx_model_set_Frame]
|
|
opt_menu_selection(primary_stem_var.get())
|
|
self.menu_placement(mdx_model_set, SPECIFY_PARAMETERS_TEXT, pop_up=False if is_macos else True, frame_list=frame_list)
|
|
|
|
def pop_up_mdx_model_sub_json_dump(self, mdx_model_params, mdx_model_hash):
|
|
"""Dumps current selected MDX-Net model settings to a json named after model hash"""
|
|
|
|
self.mdx_model_params = mdx_model_params
|
|
|
|
mdx_model_params_dump = json.dumps(mdx_model_params, indent=4)
|
|
with open(os.path.join(MDX_HASH_DIR, f'{mdx_model_hash}.json'), "w") as outfile:
|
|
outfile.write(mdx_model_params_dump)
|
|
|
|
def pop_up_mdx_c_param(self, mdx_model_hash):
|
|
"""Opens MDX-C param settings"""
|
|
|
|
mdx_c_param_menu = tk.Toplevel()
|
|
|
|
get_mdx_c_params = lambda dir, ext:tuple(os.path.splitext(x)[0] for x in os.listdir(dir) if x.endswith(ext))
|
|
new_mdx_c_params = get_mdx_c_params(MDX_C_CONFIG_PATH, YAML)
|
|
mdx_c_model_param_var = tk.StringVar(value=NONE_SELECTED)
|
|
|
|
def pull_data():
|
|
mdx_c_model_params = {
|
|
'config_yaml': f"{mdx_c_model_param_var.get()}{YAML}"}
|
|
|
|
if not mdx_c_model_param_var.get() == NONE_SELECTED:
|
|
self.pop_up_mdx_model_sub_json_dump(mdx_c_model_params, mdx_model_hash)
|
|
mdx_c_param_menu.destroy()
|
|
else:
|
|
self.mdx_model_params = None
|
|
|
|
def cancel():
|
|
self.mdx_model_params = None
|
|
mdx_c_param_menu.destroy()
|
|
|
|
mdx_c_param_Frame = self.menu_FRAME_SET(mdx_c_param_menu)
|
|
mdx_c_param_Frame.grid(row=0)
|
|
|
|
mdx_c_param_title_title = self.menu_title_LABEL_SET(mdx_c_param_Frame, MDXNET_C_MODEL_PARAMETERS_TEXT, width=28)
|
|
mdx_c_param_title_title.grid(row=0,column=0,padx=0,pady=0)
|
|
|
|
mdx_c_model_param_Label = self.menu_sub_LABEL_SET(mdx_c_param_Frame, SELECT_MODEL_PARAM_TEXT)
|
|
mdx_c_model_param_Label.grid(pady=MENU_PADDING_1)
|
|
mdx_c_model_param_Option = ComboBoxMenu(mdx_c_param_Frame, textvariable=mdx_c_model_param_var, values=new_mdx_c_params, width=30)
|
|
mdx_c_model_param_Option.grid(padx=20,pady=MENU_PADDING_1)
|
|
self.help_hints(mdx_c_model_param_Label, text=VR_MODEL_PARAM_HELP)
|
|
|
|
mdx_c_param_confrim_Button = ttk.Button(mdx_c_param_Frame, text=CONFIRM_TEXT, command=lambda:pull_data())
|
|
mdx_c_param_confrim_Button.grid(pady=MENU_PADDING_1)
|
|
|
|
mdx_c_param_cancel_Button = ttk.Button(mdx_c_param_Frame, text=CANCEL_TEXT, command=cancel)
|
|
mdx_c_param_cancel_Button.grid(pady=MENU_PADDING_1)
|
|
|
|
mdx_c_param_menu.protocol("WM_DELETE_WINDOW", cancel)
|
|
|
|
self.menu_placement(mdx_c_param_menu, CHOOSE_MODEL_PARAM_TEXT, pop_up=True)
|
|
|
|
def pop_up_vr_param(self, vr_model_hash):
|
|
"""Opens VR param settings"""
|
|
|
|
vr_param_menu = tk.Toplevel()
|
|
|
|
get_vr_params = lambda dir, ext:tuple(os.path.splitext(x)[0] for x in os.listdir(dir) if x.endswith(ext))
|
|
new_vr_params = get_vr_params(VR_PARAM_DIR, JSON)
|
|
vr_model_param_var = tk.StringVar(value=NONE_SELECTED)
|
|
vr_model_stem_var = tk.StringVar(value='Vocals')
|
|
vr_model_nout_var = tk.StringVar(value=32)
|
|
vr_model_nout_lstm_var = tk.StringVar(value=128)
|
|
is_new_vr_model_var = tk.BooleanVar(value=False)
|
|
balance_value_var = tk.StringVar(value=0)
|
|
is_kara_model_var = tk.BooleanVar(value=False)
|
|
is_bv_model_var = tk.BooleanVar(value=False)
|
|
|
|
enable_new_vr_op = lambda:(vr_model_nout_Option.configure(state=READ_ONLY), vr_model_nout_lstm_Option.configure(state=READ_ONLY))
|
|
disable_new_vr_op = lambda:(vr_model_nout_Option.configure(state=tk.DISABLED), vr_model_nout_lstm_Option.configure(state=tk.DISABLED))
|
|
vr_new_toggle = lambda:enable_new_vr_op() if is_new_vr_model_var.get() else disable_new_vr_op()
|
|
|
|
def pull_data():
|
|
if is_new_vr_model_var.get():
|
|
vr_model_params = {
|
|
'vr_model_param': vr_model_param_var.get(),
|
|
'primary_stem': vr_model_stem_var.get(),
|
|
'nout': int(vr_model_nout_var.get()),
|
|
'nout_lstm': int(vr_model_nout_lstm_var.get()),
|
|
IS_KARAOKEE: bool(is_kara_model_var.get()),
|
|
IS_BV_MODEL: bool(is_bv_model_var.get()),
|
|
IS_BV_MODEL_REBAL: float(balance_value_var.get())
|
|
}
|
|
else:
|
|
vr_model_params = {
|
|
'vr_model_param': vr_model_param_var.get(),
|
|
'primary_stem': vr_model_stem_var.get(),
|
|
IS_KARAOKEE: bool(is_kara_model_var.get()),
|
|
IS_BV_MODEL: bool(is_bv_model_var.get()),
|
|
IS_BV_MODEL_REBAL: float(balance_value_var.get())}
|
|
|
|
if not vr_model_param_var.get() == NONE_SELECTED:
|
|
self.pop_up_vr_param_sub_json_dump(vr_model_params, vr_model_hash)
|
|
vr_param_menu.destroy()
|
|
else:
|
|
self.vr_model_params = None
|
|
self.error_dialoge(INVALID_PARAM_MODEL_ERROR)
|
|
|
|
def cancel():
|
|
self.vr_model_params = None
|
|
vr_param_menu.destroy()
|
|
|
|
def toggle_kara():
|
|
if is_kara_model_var.get():
|
|
is_bv_model_var.set(False)
|
|
balance_value_Option.configure(state=tk.DISABLED)
|
|
|
|
def toggle_bv():
|
|
if is_bv_model_var.get():
|
|
is_kara_model_var.set(False)
|
|
balance_value_Option.configure(state=READ_ONLY)
|
|
else:
|
|
balance_value_Option.configure(state=tk.DISABLED)
|
|
|
|
def opt_menu_selection(selection):
|
|
if not selection in [VOCAL_STEM, INST_STEM]:
|
|
balance_value_Option.configure(state=tk.DISABLED)
|
|
is_kara_model_Option.configure(state=tk.DISABLED)
|
|
is_bv_model_Option.configure(state=tk.DISABLED)
|
|
is_kara_model_var.set(False)
|
|
is_bv_model_var.set(False)
|
|
balance_value_var.set(0)
|
|
else:
|
|
is_kara_model_Option.configure(state=tk.NORMAL)
|
|
is_bv_model_Option.configure(state=tk.NORMAL)
|
|
|
|
vr_param_Frame = self.menu_FRAME_SET(vr_param_menu)
|
|
vr_param_Frame.grid(row=0, padx=20)
|
|
|
|
vr_param_title_title = self.menu_title_LABEL_SET(vr_param_Frame, SPECIFY_VR_MODEL_PARAMETERS_TEXT)
|
|
vr_param_title_title.grid()
|
|
|
|
vr_model_stem_Label = self.menu_sub_LABEL_SET(vr_param_Frame, PRIMARY_STEM_TEXT)
|
|
vr_model_stem_Label.grid(pady=MENU_PADDING_1)
|
|
vr_model_stem_Option = ttk.OptionMenu(vr_param_Frame, vr_model_stem_var, None, *STEM_SET_MENU, command=opt_menu_selection)
|
|
vr_model_stem_Option.configure(width=15)
|
|
vr_model_stem_Option.grid(pady=MENU_PADDING_1)
|
|
vr_model_stem_Option['menu'].insert_separator(len(STEM_SET_MENU))
|
|
vr_model_stem_Option['menu'].add_radiobutton(label=INPUT_STEM_NAME, command=tk._setit(vr_model_stem_var, INPUT_STEM_NAME, lambda e:self.pop_up_input_stem_name(vr_model_stem_var, vr_param_menu)))
|
|
self.help_hints(vr_model_stem_Label, text=SET_STEM_NAME_HELP)
|
|
|
|
is_kara_model_Option = ttk.Checkbutton(vr_param_Frame, text=KARAOKE_MODEL_TEXT, width=SET_MENUS_CHECK_WIDTH, variable=is_kara_model_var, command=toggle_kara)
|
|
is_kara_model_Option.grid(pady=0)
|
|
|
|
is_bv_model_Option = ttk.Checkbutton(vr_param_Frame, text=BV_MODEL_TEXT, width=SET_MENUS_CHECK_WIDTH, variable=is_bv_model_var, command=toggle_bv)
|
|
is_bv_model_Option.grid(pady=0)
|
|
|
|
balance_value_Label = self.menu_sub_LABEL_SET(vr_param_Frame, BALANCE_VALUE_TEXT)
|
|
balance_value_Label.grid(pady=MENU_PADDING_1)
|
|
balance_value_Option = ComboBoxMenu(vr_param_Frame, textvariable=balance_value_var, values=BALANCE_VALUES, width=COMBO_WIDTH)
|
|
balance_value_Option.configure(state=tk.DISABLED)
|
|
balance_value_Option.grid(pady=MENU_PADDING_1)
|
|
|
|
is_new_vr_model_Option = ttk.Checkbutton(vr_param_Frame, text=VR_51_MODEL_TEXT, width=SET_MENUS_CHECK_WIDTH, variable=is_new_vr_model_var, command=vr_new_toggle)
|
|
is_new_vr_model_Option.grid(pady=MENU_PADDING_1)
|
|
|
|
vr_model_nout_Label = self.menu_sub_LABEL_SET(vr_param_Frame, 'Out Channels')
|
|
vr_model_nout_Label.grid(pady=MENU_PADDING_1)
|
|
vr_model_nout_Option = ComboBoxEditableMenu(vr_param_Frame, values=NOUT_SEL, textvariable=vr_model_nout_var, pattern=REG_SHIFTS, default='32', width=COMBO_WIDTH)
|
|
vr_model_nout_Option.grid(pady=MENU_PADDING_1)
|
|
#self.help_hints(vr_model_nout_Label, text=VR_MODEL_NOUT_HELP)
|
|
|
|
vr_model_nout_lstm_Label = self.menu_sub_LABEL_SET(vr_param_Frame, 'Out Channels (LSTM layer)')
|
|
vr_model_nout_lstm_Label.grid(pady=MENU_PADDING_1)
|
|
vr_model_nout_lstm_Option = ComboBoxEditableMenu(vr_param_Frame, values=NOUT_LSTM_SEL, textvariable=vr_model_nout_lstm_var, pattern=REG_SHIFTS, default='128', width=COMBO_WIDTH)#
|
|
vr_model_nout_lstm_Option.grid(pady=MENU_PADDING_1)
|
|
#self.help_hints(vr_model_param_Label, text=VR_MODEL_NOUT_LSTM_HELP)
|
|
|
|
vr_model_param_Label = self.menu_sub_LABEL_SET(vr_param_Frame, SELECT_MODEL_PARAM_TEXT)
|
|
vr_model_param_Label.grid(pady=MENU_PADDING_1)
|
|
vr_model_param_Option = ComboBoxMenu(vr_param_Frame, textvariable=vr_model_param_var, values=new_vr_params, width=30)
|
|
vr_model_param_Option.grid(pady=MENU_PADDING_1)
|
|
self.help_hints(vr_model_param_Label, text=VR_MODEL_PARAM_HELP)
|
|
|
|
vr_param_confrim_Button = ttk.Button(vr_param_Frame, text=CONFIRM_TEXT, command=lambda:pull_data())
|
|
vr_param_confrim_Button.grid(pady=MENU_PADDING_1)
|
|
|
|
vr_param_cancel_Button = ttk.Button(vr_param_Frame, text=CANCEL_TEXT, command=cancel)
|
|
vr_param_cancel_Button.grid(pady=MENU_PADDING_1)
|
|
|
|
vr_new_toggle()
|
|
opt_menu_selection(vr_model_stem_var.get())
|
|
|
|
vr_param_menu.protocol("WM_DELETE_WINDOW", cancel)
|
|
|
|
frame_list = [vr_param_Frame]
|
|
self.menu_placement(vr_param_menu, CHOOSE_MODEL_PARAM_TEXT, pop_up=False if is_macos else True, frame_list=frame_list)
|
|
|
|
def pop_up_vr_param_sub_json_dump(self, vr_model_params, vr_model_hash):
|
|
"""Dumps current selected VR model settings to a json named after model hash"""
|
|
|
|
self.vr_model_params = vr_model_params
|
|
|
|
vr_model_params_dump = json.dumps(vr_model_params, indent=4)
|
|
|
|
with open(os.path.join(VR_HASH_DIR, f'{vr_model_hash}.json'), "w") as outfile:
|
|
outfile.write(vr_model_params_dump)
|
|
|
|
def pop_up_input_stem_name(self, stem_var:tk.StringVar, parent_window:tk.Toplevel):
|
|
"""
|
|
Input Stem Name
|
|
"""
|
|
|
|
stem_input_save = tk.Toplevel(root)
|
|
|
|
def close_window(is_cancel=True):
|
|
|
|
if is_cancel or not stem_input_save_var.get():
|
|
stem_var.set(VOCAL_STEM)
|
|
else:
|
|
stem_input_save_text = stem_input_save_var.get().capitalize()
|
|
|
|
if stem_input_save_text == VOCAL_STEM:
|
|
stem_text = INST_STEM if is_inverse_stem_var.get() else stem_input_save_text
|
|
elif stem_input_save_text == INST_STEM:
|
|
stem_text = VOCAL_STEM if is_inverse_stem_var.get() else stem_input_save_text
|
|
else:
|
|
stem_text = f"{NO_STEM}{stem_input_save_text}" if is_inverse_stem_var.get() else stem_input_save_text
|
|
|
|
stem_var.set(stem_text)
|
|
|
|
stem_input_save.destroy()
|
|
|
|
parent_window.attributes('-topmost', 'true') if OPERATING_SYSTEM == "Linux" else None
|
|
parent_window.grab_set()
|
|
root.wait_window(parent_window)
|
|
|
|
stem_input_save_var = tk.StringVar(value='')
|
|
is_inverse_stem_var = tk.BooleanVar(value=False)
|
|
|
|
validation = lambda value:False if re.fullmatch(REG_INPUT_STEM_NAME, value) is None else True
|
|
stem_input_save_Frame = self.menu_FRAME_SET(stem_input_save)
|
|
stem_input_save_Frame.grid(row=1)
|
|
|
|
stem_input_save_title = self.menu_title_LABEL_SET(stem_input_save_Frame, INPUT_STEM_NAME_TEXT)
|
|
stem_input_save_title.grid(pady=0)
|
|
|
|
stem_input_name_Label = self.menu_sub_LABEL_SET(stem_input_save_Frame, STEM_NAME_TEXT)
|
|
stem_input_name_Label.grid(pady=MENU_PADDING_1)
|
|
stem_input_name_Entry = ttk.Combobox(stem_input_save_Frame, textvariable=stem_input_save_var, values=STEM_SET_MENU_2, justify='center', width=25)
|
|
invalid_message = self.invalid_tooltip(stem_input_name_Entry, REG_INPUT_STEM_NAME)
|
|
stem_input_name_Entry.grid(pady=MENU_PADDING_1)
|
|
stem_input_name_Entry.focus_set()
|
|
|
|
self.spacer_label(stem_input_save_Frame)
|
|
|
|
is_inverse_stem_Button = ttk.Checkbutton(stem_input_save_Frame, text=IS_INVERSE_STEM_TEXT, variable=is_inverse_stem_var)
|
|
is_inverse_stem_Button.grid(pady=0)
|
|
|
|
entry_rules_Label = tk.Label(stem_input_save_Frame, text=STEM_INPUT_RULE, font=(MAIN_FONT_NAME, f"{FONT_SIZE_1}"), foreground='#868687', justify="left")
|
|
entry_rules_Label.grid(pady=MENU_PADDING_1)
|
|
|
|
mdx_param_set_Button = ttk.Button(stem_input_save_Frame, text=DONE_MENU_TEXT, command=lambda:close_window(is_cancel=False) if validation(stem_input_save_var.get()) else invalid_message())
|
|
mdx_param_set_Button.grid(pady=MENU_PADDING_1)
|
|
|
|
stop_process_Button = ttk.Button(stem_input_save_Frame, text=CANCEL_TEXT, command=close_window)
|
|
stop_process_Button.grid(pady=MENU_PADDING_1)
|
|
|
|
stem_input_save.protocol("WM_DELETE_WINDOW", close_window)
|
|
|
|
frame_list = [stem_input_save_Frame]
|
|
self.menu_placement(stem_input_save, INPUT_UNIQUE_STEM_NAME_TEXT, pop_up=True, frame_list=frame_list)
|
|
|
|
def pop_up_save_ensemble(self):
|
|
"""
|
|
Save Ensemble as...
|
|
"""
|
|
|
|
ensemble_save = tk.Toplevel(root)
|
|
|
|
ensemble_save_var = tk.StringVar(value='')
|
|
|
|
ensemble_save_Frame = self.menu_FRAME_SET(ensemble_save)
|
|
ensemble_save_Frame.grid(row=1)
|
|
|
|
validation = lambda value:False if re.fullmatch(REG_SAVE_INPUT, value) is None else True
|
|
save_func = lambda:(self.pop_up_save_ensemble_sub_json_dump(self.ensemble_listbox_get_all_selected_models(), ensemble_save_var.get()), ensemble_save.destroy())
|
|
|
|
if len(self.ensemble_listbox_get_all_selected_models()) <= 1:
|
|
ensemble_save_title = self.menu_title_LABEL_SET(ensemble_save_Frame, ENSEMBLE_WARNING_NOT_ENOUGH_SHORT_TEXT, width=20)
|
|
ensemble_save_title.grid()
|
|
|
|
ensemble_save_title = self.menu_sub_LABEL_SET(ensemble_save_Frame, ENSEMBLE_WARNING_NOT_ENOUGH_TEXT)
|
|
ensemble_save_title.grid(pady=MENU_PADDING_1)
|
|
|
|
stop_process_Button = ttk.Button(ensemble_save_Frame, text=OK_TEXT, command=lambda:ensemble_save.destroy())
|
|
stop_process_Button.grid()
|
|
else:
|
|
ensemble_save_title = self.menu_title_LABEL_SET(ensemble_save_Frame, SAVE_CURRENT_ENSEMBLE_TEXT)
|
|
ensemble_save_title.grid()
|
|
|
|
ensemble_name_Label = self.menu_sub_LABEL_SET(ensemble_save_Frame, ENSEMBLE_NAME_TEXT)
|
|
ensemble_name_Label.grid(pady=MENU_PADDING_1)
|
|
ensemble_name_Entry = ttk.Entry(ensemble_save_Frame, textvariable=ensemble_save_var, justify='center', width=25)
|
|
ensemble_name_Entry.grid(pady=MENU_PADDING_1)
|
|
invalid_message = self.invalid_tooltip(ensemble_name_Entry)
|
|
ensemble_name_Entry.focus_set()
|
|
self.spacer_label(ensemble_save_Frame)
|
|
|
|
entry_rules_Label = tk.Label(ensemble_save_Frame, text=ENSEMBLE_INPUT_RULE, font=(MAIN_FONT_NAME, f"{FONT_SIZE_1}"), foreground='#868687', justify="left")
|
|
entry_rules_Label.grid()
|
|
|
|
mdx_param_set_Button = ttk.Button(ensemble_save_Frame, text=SAVE_TEXT, command=lambda:save_func() if validation(ensemble_save_var.get()) else invalid_message())
|
|
mdx_param_set_Button.grid(pady=MENU_PADDING_1)
|
|
|
|
stop_process_Button = ttk.Button(ensemble_save_Frame, text=CANCEL_TEXT, command=lambda:ensemble_save.destroy())
|
|
stop_process_Button.grid(pady=MENU_PADDING_1)
|
|
|
|
self.menu_placement(ensemble_save, SAVE_CURRENT_ENSEMBLE_TEXT, pop_up=True)
|
|
|
|
def pop_up_save_ensemble_sub_json_dump(self, selected_ensemble_model, ensemble_save_name: str):
|
|
"""Dumps current ensemble settings to a json named after user input"""
|
|
|
|
if ensemble_save_name:
|
|
self.chosen_ensemble_var.set(ensemble_save_name)
|
|
ensemble_save_name = ensemble_save_name.replace(" ", "_")
|
|
saved_data = {
|
|
'ensemble_main_stem': self.ensemble_main_stem_var.get(),
|
|
'ensemble_type': self.ensemble_type_var.get(),
|
|
'selected_models': selected_ensemble_model,
|
|
}
|
|
|
|
saved_data_dump = json.dumps(saved_data, indent=4)
|
|
with open(os.path.join(ENSEMBLE_CACHE_DIR, f'{ensemble_save_name}.json'), "w") as outfile:
|
|
outfile.write(saved_data_dump)
|
|
|
|
def deletion_list_fill(self, option_menu: ComboBoxMenu, selection_var: tk.StringVar, selection_dir, var_set, menu_name=None):
|
|
"""Fills the saved settings menu located in tab 2 of the main settings window"""
|
|
|
|
def command_callback(event=None):
|
|
self.deletion_entry(selection_var.get(), selection_dir, refresh_menu)
|
|
selection_var.set(var_set)
|
|
|
|
def refresh_menu(remove=None):
|
|
selection_list = self.last_found_ensembles if menu_name == 'deleteensemble' else self.last_found_settings
|
|
main_var = self.chosen_ensemble_var if menu_name == 'deleteensemble' else self.save_current_settings_var
|
|
|
|
if remove and remove in selection_list:
|
|
selection_list = list(selection_list)
|
|
selection_list.remove(remove)
|
|
main_var.set(CHOOSE_ENSEMBLE_OPTION)
|
|
|
|
self.update_menus(option_widget=option_menu,
|
|
style_name=menu_name,
|
|
command=command_callback,
|
|
new_items=selection_list)
|
|
|
|
refresh_menu()
|
|
|
|
def deletion_entry(self, selection: str, path, callback):
|
|
"""Deletes selected user saved application settings"""
|
|
|
|
if selection not in [SELECT_SAVED_SET, SELECT_SAVED_ENSEMBLE]:
|
|
saved_path = os.path.join(path, f'{selection.replace(" ", "_")}.json')
|
|
confirm = self.message_box(DELETE_ENS_ENTRY)
|
|
if confirm:
|
|
if os.path.isfile(saved_path):
|
|
os.remove(saved_path)
|
|
callback(selection)
|
|
|
|
#--Download Center Methods--
|
|
|
|
def online_data_refresh(self, user_refresh=True, confirmation_box=False, refresh_list_Button=False, is_start_up=False, is_download_complete=False):
|
|
"""Checks for application updates"""
|
|
|
|
def online_check():
|
|
if not is_start_up:
|
|
self.app_update_status_Text_var.set(LOADING_VERSION_INFO_TEXT)
|
|
self.app_update_button_Text_var.set(CHECK_FOR_UPDATES_TEXT)
|
|
|
|
is_new_update = False
|
|
try:
|
|
self.online_data = json.load(urllib.request.urlopen(DOWNLOAD_CHECKS))
|
|
self.is_online = True
|
|
|
|
try:
|
|
with urllib.request.urlopen(BULLETIN_CHECK) as response:
|
|
self.bulletin_data = response.read().decode('utf-8')
|
|
|
|
if not is_windows:
|
|
self.bulletin_data = read_bulliten_text_mac(CR_TEXT, self.bulletin_data)
|
|
else:
|
|
self.bulletin_data = self.bulletin_data.replace("~", "•")
|
|
|
|
except Exception as e:
|
|
self.bulletin_data = INFO_UNAVAILABLE_TEXT
|
|
print(e)
|
|
|
|
if user_refresh:
|
|
self.download_list_state()
|
|
for widget in self.download_center_Buttons:
|
|
widget.configure(state=tk.NORMAL)
|
|
|
|
if refresh_list_Button:
|
|
self.download_progress_info_var.set('Download List Refreshed!')
|
|
|
|
if OPERATING_SYSTEM=="Darwin":
|
|
self.lastest_version = self.online_data["current_version_mac"]
|
|
elif OPERATING_SYSTEM=="Linux":
|
|
self.lastest_version = self.online_data["current_version_linux"]
|
|
else:
|
|
self.lastest_version = self.online_data["current_version"]
|
|
|
|
if self.lastest_version == current_patch and not is_start_up:
|
|
self.app_update_status_Text_var.set('UVR Version Current')
|
|
else:
|
|
is_new_update = True
|
|
is_beta_version = True if self.lastest_version == PREVIOUS_PATCH_WIN and BETA_VERSION in current_patch else False
|
|
|
|
if not is_start_up:
|
|
if is_beta_version:
|
|
self.app_update_status_Text_var.set(f"Roll Back: {self.lastest_version}")
|
|
self.app_update_button_Text_var.set(ROLL_BACK_TEXT)
|
|
else:
|
|
self.app_update_status_Text_var.set(f"Update Found: {self.lastest_version}")
|
|
self.app_update_button_Text_var.set('Click Here to Update')
|
|
|
|
if OPERATING_SYSTEM == "Windows":
|
|
self.download_update_link_var.set('{}{}{}'.format(UPDATE_REPO, self.lastest_version, application_extension))
|
|
self.download_update_path_var.set(os.path.join(BASE_PATH, f'{self.lastest_version}{application_extension}'))
|
|
elif OPERATING_SYSTEM == "Darwin":
|
|
self.download_update_link_var.set(UPDATE_MAC_ARM_REPO if SYSTEM_PROC == ARM or ARM in SYSTEM_ARCH else UPDATE_MAC_X86_64_REPO)
|
|
elif OPERATING_SYSTEM == "Linux":
|
|
self.download_update_link_var.set(UPDATE_LINUX_REPO)
|
|
|
|
if not user_refresh:
|
|
if not is_beta_version and not self.lastest_version == current_patch:
|
|
self.command_Text.write(NEW_UPDATE_FOUND_TEXT(self.lastest_version))
|
|
|
|
|
|
is_update_params = self.is_auto_update_model_params if is_start_up else self.is_auto_update_model_params_var.get()
|
|
|
|
if is_update_params and is_start_up or is_download_complete:
|
|
self.download_model_settings()
|
|
|
|
# if is_download_complete:
|
|
# self.download_model_settings()
|
|
|
|
except Exception as e:
|
|
self.offline_state_set(is_start_up)
|
|
is_new_update = False
|
|
|
|
if user_refresh:
|
|
self.download_list_state(disable_only=True)
|
|
for widget in self.download_center_Buttons:
|
|
widget.configure(state=tk.DISABLED)
|
|
|
|
try:
|
|
self.error_log_var.set(error_text('Online Data Refresh', e))
|
|
except Exception as e:
|
|
print(e)
|
|
|
|
return is_new_update
|
|
|
|
if confirmation_box:
|
|
return online_check()
|
|
else:
|
|
self.current_thread = KThread(target=online_check)
|
|
self.current_thread.setDaemon(True) if not is_windows else None
|
|
self.current_thread.start()
|
|
|
|
def offline_state_set(self, is_start_up=False):
|
|
"""Changes relevant settings and "Download Center" buttons if no internet connection is available"""
|
|
|
|
if not is_start_up and self.is_menu_settings_open:
|
|
self.app_update_status_Text_var.set(f'Version Status: {NO_CONNECTION}')
|
|
self.download_progress_info_var.set(NO_CONNECTION)
|
|
self.app_update_button_Text_var.set('Refresh')
|
|
self.refresh_list_Button.configure(state=tk.NORMAL)
|
|
self.stop_download_Button_DISABLE()
|
|
self.enable_tabs()
|
|
|
|
self.is_online = False
|
|
|
|
def download_validate_code(self, confirm=False, code_message=None):
|
|
"""Verifies the VIP download code"""
|
|
|
|
self.decoded_vip_link = vip_downloads(self.user_code_var.get())
|
|
|
|
if confirm:
|
|
if not self.decoded_vip_link == NO_CODE:
|
|
info_text = 'VIP Models Added!'
|
|
is_success_message = True
|
|
else:
|
|
info_text = 'Incorrect Code'
|
|
is_success_message = False
|
|
|
|
self.download_progress_info_var.set(info_text)
|
|
self.user_code_validation_var.set(info_text)
|
|
|
|
if code_message:
|
|
code_message(info_text, is_success_message)
|
|
|
|
self.download_list_fill()
|
|
|
|
def download_list_fill(self, model_type=ALL_TYPES):
|
|
"""Fills the download lists with the data retrieved from the update check."""
|
|
|
|
self.download_demucs_models_list.clear()
|
|
|
|
model_download_mdx_list, model_download_mdx_name = [], "mdxdownload"
|
|
model_download_vr_list, model_download_vr_name = [], "vrdownload"
|
|
model_download_demucs_list, model_download_demucs_name = [], "demucsmdxdownload"
|
|
|
|
self.vr_download_list = self.online_data["vr_download_list"]
|
|
self.mdx_download_list = self.online_data["mdx_download_list"]
|
|
self.demucs_download_list = self.online_data["demucs_download_list"]
|
|
self.mdx_download_list.update(self.online_data["mdx23c_download_list"])
|
|
|
|
if not self.decoded_vip_link is NO_CODE:
|
|
self.vr_download_list.update(self.online_data["vr_download_vip_list"])
|
|
self.mdx_download_list.update(self.online_data["mdx_download_vip_list"])
|
|
self.mdx_download_list.update(self.online_data["mdx23c_download_vip_list"])
|
|
|
|
def configure_combobox(combobox:ComboBoxMenu, values:list, variable:tk.StringVar, arch_type, name):
|
|
values = [NO_NEW_MODELS] if not values else values
|
|
combobox['values'] = values
|
|
combobox.update_dropdown_size(values, name, offset=310,
|
|
command=lambda s: self.download_model_select(variable.get(), arch_type, variable))
|
|
|
|
if model_type in [VR_ARCH_TYPE, ALL_TYPES]:
|
|
for (selectable, model) in self.vr_download_list.items():
|
|
if not os.path.isfile(os.path.join(VR_MODELS_DIR, model)):
|
|
model_download_vr_list.append(selectable)
|
|
|
|
configure_combobox(self.model_download_vr_Option, model_download_vr_list, self.model_download_vr_var, VR_ARCH_TYPE, model_download_vr_name)
|
|
|
|
if model_type in [MDX_ARCH_TYPE, ALL_TYPES]:
|
|
for (selectable, model) in self.mdx_download_list.items():
|
|
if isinstance(model, dict):
|
|
items_list = list(model.items())
|
|
model_name, config = items_list[0]
|
|
config_link = f"{MDX23_CONFIG_CHECKS}{config}"
|
|
config_local = os.path.join(MDX_C_CONFIG_PATH, config)
|
|
if not os.path.isfile(config_local):
|
|
with urllib.request.urlopen(config_link) as response:
|
|
with open(config_local, 'wb') as out_file:
|
|
out_file.write(response.read())
|
|
else:
|
|
model_name = str(model)
|
|
|
|
if not os.path.isfile(os.path.join(MDX_MODELS_DIR, model_name)):
|
|
model_download_mdx_list.append(selectable)
|
|
|
|
configure_combobox(self.model_download_mdx_Option, model_download_mdx_list, self.model_download_mdx_var, MDX_ARCH_TYPE, model_download_mdx_name)
|
|
|
|
if model_type in [DEMUCS_ARCH_TYPE, ALL_TYPES]:
|
|
for (selectable, model) in self.demucs_download_list.items():
|
|
for name in model.items():
|
|
if [True for x in DEMUCS_NEWER_ARCH_TYPES if x in selectable]:
|
|
if not os.path.isfile(os.path.join(DEMUCS_NEWER_REPO_DIR, name[0])):
|
|
self.download_demucs_models_list.append(selectable)
|
|
else:
|
|
if not os.path.isfile(os.path.join(DEMUCS_MODELS_DIR, name[0])):
|
|
self.download_demucs_models_list.append(selectable)
|
|
|
|
self.download_demucs_models_list = list(dict.fromkeys(self.download_demucs_models_list))
|
|
|
|
for option_name in self.download_demucs_models_list:
|
|
model_download_demucs_list.append(option_name)
|
|
|
|
configure_combobox(self.model_download_demucs_Option, model_download_demucs_list, self.model_download_demucs_var, DEMUCS_ARCH_TYPE, model_download_demucs_name)
|
|
|
|
def download_model_settings(self):
|
|
'''Update the newest model settings'''
|
|
|
|
try:
|
|
self.vr_hash_MAPPER = json.load(urllib.request.urlopen(VR_MODEL_DATA_LINK))
|
|
self.mdx_hash_MAPPER = json.load(urllib.request.urlopen(MDX_MODEL_DATA_LINK))
|
|
self.mdx_name_select_MAPPER = json.load(urllib.request.urlopen(MDX_MODEL_NAME_DATA_LINK))
|
|
self.demucs_name_select_MAPPER = json.load(urllib.request.urlopen(DEMUCS_MODEL_NAME_DATA_LINK))
|
|
|
|
vr_hash_MAPPER_dump = json.dumps(self.vr_hash_MAPPER, indent=4)
|
|
with open(VR_HASH_JSON, "w") as outfile:
|
|
outfile.write(vr_hash_MAPPER_dump)
|
|
|
|
mdx_hash_MAPPER_dump = json.dumps(self.mdx_hash_MAPPER, indent=4)
|
|
with open(MDX_HASH_JSON, "w") as outfile:
|
|
outfile.write(mdx_hash_MAPPER_dump)
|
|
|
|
mdx_name_select_MAPPER_dump = json.dumps(self.mdx_name_select_MAPPER, indent=4)
|
|
with open(MDX_MODEL_NAME_SELECT, "w") as outfile:
|
|
outfile.write(mdx_name_select_MAPPER_dump)
|
|
|
|
demucs_name_select_MAPPER_dump = json.dumps(self.demucs_name_select_MAPPER, indent=4)
|
|
with open(DEMUCS_MODEL_NAME_SELECT, "w") as outfile:
|
|
outfile.write(demucs_name_select_MAPPER_dump)
|
|
|
|
except Exception as e:
|
|
self.vr_hash_MAPPER = load_model_hash_data(VR_HASH_JSON)
|
|
self.mdx_hash_MAPPER = load_model_hash_data(MDX_HASH_JSON)
|
|
self.mdx_name_select_MAPPER = load_model_hash_data(MDX_MODEL_NAME_SELECT)
|
|
self.demucs_name_select_MAPPER = load_model_hash_data(DEMUCS_MODEL_NAME_SELECT)
|
|
self.error_log_var.set(e)
|
|
print(e)
|
|
|
|
def download_list_state(self, reset=True, disable_only=False):
|
|
"""Makes sure only the models from the chosen AI network are selectable."""
|
|
|
|
for widget in self.download_lists:
|
|
widget.configure(state=tk.DISABLED)
|
|
|
|
if reset:
|
|
for download_list_var in self.download_list_vars:
|
|
if self.is_online:
|
|
download_list_var.set(NO_MODEL)
|
|
self.download_Button.configure(state=tk.NORMAL)
|
|
else:
|
|
download_list_var.set(NO_CONNECTION)
|
|
self.download_Button.configure(state=tk.DISABLED)
|
|
disable_only = True
|
|
|
|
if not disable_only:
|
|
self.download_Button.configure(state=tk.NORMAL)
|
|
if self.select_download_var.get() == VR_ARCH_TYPE:
|
|
self.model_download_vr_Option.configure(state=READ_ONLY)
|
|
self.selected_download_var = self.model_download_vr_var
|
|
self.download_list_fill(model_type=VR_ARCH_TYPE)
|
|
if self.select_download_var.get() == MDX_ARCH_TYPE:
|
|
self.model_download_mdx_Option.configure(state=READ_ONLY)
|
|
self.selected_download_var = self.model_download_mdx_var
|
|
self.download_list_fill(model_type=MDX_ARCH_TYPE)
|
|
if self.select_download_var.get() == DEMUCS_ARCH_TYPE:
|
|
self.model_download_demucs_Option.configure(state=READ_ONLY)
|
|
self.selected_download_var = self.model_download_demucs_var
|
|
self.download_list_fill(model_type=DEMUCS_ARCH_TYPE)
|
|
|
|
self.stop_download_Button_DISABLE()
|
|
|
|
def download_model_select(self, selection, type, var:tk.StringVar):
|
|
"""Prepares the data needed to download selected model."""
|
|
|
|
self.download_demucs_newer_models.clear()
|
|
|
|
if selection == NO_NEW_MODELS:
|
|
selection = NO_MODEL
|
|
var.set(NO_MODEL)
|
|
|
|
model_repo = self.decoded_vip_link if VIP_SELECTION in selection else NORMAL_REPO
|
|
is_demucs_newer = [True for x in DEMUCS_NEWER_ARCH_TYPES if x in selection]
|
|
|
|
if type == VR_ARCH_TYPE:
|
|
for selected_model in self.vr_download_list.items():
|
|
if selection in selected_model:
|
|
self.download_link_path_var.set("{}{}".format(model_repo, selected_model[1]))
|
|
self.download_save_path_var.set(os.path.join(VR_MODELS_DIR, selected_model[1]))
|
|
break
|
|
|
|
if type == MDX_ARCH_TYPE:
|
|
for selected_model in self.mdx_download_list.items():
|
|
if selection in selected_model:
|
|
if isinstance(selected_model[1], dict):
|
|
model_name = list(selected_model[1].keys())[0]
|
|
else:
|
|
model_name = str(selected_model[1])
|
|
self.download_link_path_var.set("{}{}".format(model_repo, model_name))
|
|
self.download_save_path_var.set(os.path.join(MDX_MODELS_DIR, model_name))
|
|
break
|
|
|
|
if type == DEMUCS_ARCH_TYPE:
|
|
for selected_model, model_data in self.demucs_download_list.items():
|
|
if selection == selected_model:
|
|
for key, value in model_data.items():
|
|
if is_demucs_newer:
|
|
self.download_demucs_newer_models.append([os.path.join(DEMUCS_NEWER_REPO_DIR, key), value])
|
|
else:
|
|
self.download_save_path_var.set(os.path.join(DEMUCS_MODELS_DIR, key))
|
|
self.download_link_path_var.set(value)
|
|
|
|
def download_item(self, is_update_app=False):
|
|
"""Downloads the model selected."""
|
|
|
|
if not is_update_app:
|
|
if self.selected_download_var.get() == NO_MODEL:
|
|
self.download_progress_info_var.set(NO_MODEL)
|
|
return
|
|
|
|
for widget in self.download_center_Buttons:
|
|
widget.configure(state=tk.DISABLED)
|
|
self.refresh_list_Button.configure(state=tk.DISABLED)
|
|
self.manual_download_Button.configure(state=tk.DISABLED)
|
|
|
|
is_demucs_newer = [True for x in DEMUCS_NEWER_ARCH_TYPES if x in self.selected_download_var.get()]
|
|
|
|
self.download_list_state(reset=False, disable_only=True)
|
|
self.stop_download_Button_ENABLE()
|
|
self.disable_tabs()
|
|
|
|
def download_progress_bar(current, total, model=80):
|
|
progress = ('%s' % (100 * current // total))
|
|
self.download_progress_bar_var.set(int(progress))
|
|
self.download_progress_percent_var.set(progress + ' %')
|
|
|
|
def push_download():
|
|
self.is_download_thread_active = True
|
|
try:
|
|
if is_update_app:
|
|
self.download_progress_info_var.set(DOWNLOADING_UPDATE)
|
|
if os.path.isfile(self.download_update_path_var.get()):
|
|
self.download_progress_info_var.set(FILE_EXISTS)
|
|
else:
|
|
wget.download(self.download_update_link_var.get(), self.download_update_path_var.get(), bar=download_progress_bar)
|
|
|
|
self.download_post_action(DOWNLOAD_UPDATE_COMPLETE)
|
|
else:
|
|
if self.select_download_var.get() == DEMUCS_ARCH_TYPE and is_demucs_newer:
|
|
for model_num, model_data in enumerate(self.download_demucs_newer_models, start=1):
|
|
self.download_progress_info_var.set('{} {}/{}...'.format(DOWNLOADING_ITEM, model_num, len(self.download_demucs_newer_models)))
|
|
if os.path.isfile(model_data[0]):
|
|
continue
|
|
else:
|
|
wget.download(model_data[1], model_data[0], bar=download_progress_bar)
|
|
else:
|
|
self.download_progress_info_var.set(SINGLE_DOWNLOAD)
|
|
if os.path.isfile(self.download_save_path_var.get()):
|
|
self.download_progress_info_var.set(FILE_EXISTS)
|
|
else:
|
|
wget.download(self.download_link_path_var.get(), self.download_save_path_var.get(), bar=download_progress_bar)
|
|
|
|
self.download_post_action(DOWNLOAD_COMPLETE)
|
|
|
|
except Exception as e:
|
|
self.error_log_var.set(error_text(DOWNLOADING_ITEM, e))
|
|
self.download_progress_info_var.set(DOWNLOAD_FAILED)
|
|
|
|
if type(e).__name__ == 'URLError':
|
|
self.offline_state_set()
|
|
else:
|
|
self.download_progress_percent_var.set(f"{type(e).__name__}")
|
|
self.download_post_action(DOWNLOAD_FAILED)
|
|
|
|
self.active_download_thread = KThread(target=push_download)
|
|
self.active_download_thread.start()
|
|
|
|
def download_post_action(self, action):
|
|
"""Resets the widget variables in the "Download Center" based on the state of the download."""
|
|
|
|
for widget in self.download_center_Buttons:
|
|
widget.configure(state=tk.NORMAL)
|
|
self.refresh_list_Button.configure(state=tk.NORMAL)
|
|
self.manual_download_Button.configure(state=tk.NORMAL)
|
|
|
|
self.enable_tabs()
|
|
self.stop_download_Button_DISABLE()
|
|
|
|
if action == DOWNLOAD_FAILED:
|
|
try:
|
|
self.active_download_thread.terminate()
|
|
finally:
|
|
self.download_progress_info_var.set(DOWNLOAD_FAILED)
|
|
self.download_list_state(reset=False)
|
|
if action == DOWNLOAD_STOPPED:
|
|
try:
|
|
self.active_download_thread.terminate()
|
|
finally:
|
|
self.download_progress_info_var.set(DOWNLOAD_STOPPED)
|
|
self.download_list_state(reset=False)
|
|
if action == DOWNLOAD_COMPLETE:
|
|
self.online_data_refresh(is_download_complete=True)
|
|
self.download_progress_info_var.set(DOWNLOAD_COMPLETE)
|
|
self.download_list_state()
|
|
if action == DOWNLOAD_UPDATE_COMPLETE:
|
|
self.download_progress_info_var.set(DOWNLOAD_UPDATE_COMPLETE)
|
|
if os.path.isfile(self.download_update_path_var.get()):
|
|
subprocess.Popen(self.download_update_path_var.get())
|
|
self.download_list_state()
|
|
|
|
|
|
self.is_download_thread_active = False
|
|
|
|
self.delete_temps()
|
|
|
|
#--Refresh/Loop Methods--
|
|
|
|
def update_loop(self):
|
|
"""Update the model dropdown menus"""
|
|
|
|
if self.clear_cache_torch:
|
|
clear_gpu_cache()
|
|
self.clear_cache_torch = False
|
|
|
|
if self.is_process_stopped:
|
|
if self.thread_check(self.active_processing_thread):
|
|
self.conversion_Button_Text_var.set(STOP_PROCESSING)
|
|
self.conversion_Button.configure(state=tk.DISABLED)
|
|
self.stop_Button.configure(state=tk.DISABLED)
|
|
else:
|
|
self.stop_Button.configure(state=tk.NORMAL)
|
|
self.conversion_Button_Text_var.set(START_PROCESSING)
|
|
self.conversion_Button.configure(state=tk.NORMAL)
|
|
self.progress_bar_main_var.set(0)
|
|
clear_gpu_cache()
|
|
self.is_process_stopped = False
|
|
|
|
if self.is_confirm_error_var.get():
|
|
self.check_is_menu_open(ERROR_OPTION)
|
|
self.is_confirm_error_var.set(False)
|
|
|
|
if self.is_check_splash and is_windows:
|
|
|
|
while not self.msg_queue.empty():
|
|
message = self.msg_queue.get_nowait()
|
|
print(message)
|
|
|
|
close_process(self.msg_queue)
|
|
self.is_check_splash = False
|
|
|
|
#self.auto_save()
|
|
|
|
self.update_available_models()
|
|
self.after(600, self.update_loop)
|
|
|
|
def update_menus(self, option_widget:ComboBoxMenu, style_name, command, new_items, last_items=None, base_options=None):
|
|
|
|
if new_items != last_items:
|
|
formatted_items = [item.replace("_", " ") for item in new_items]
|
|
if not formatted_items and base_options:
|
|
base_options = [option for option in base_options if option != OPT_SEPARATOR_SAVE]
|
|
|
|
final_options = formatted_items + base_options if base_options else formatted_items
|
|
option_widget['values'] = final_options
|
|
option_widget.update_dropdown_size(formatted_items, style_name, command=command)
|
|
return new_items
|
|
return last_items
|
|
|
|
def update_available_models(self):
|
|
"""
|
|
Loops through all models in each model directory and adds them to the appropriate model menu.
|
|
Also updates ensemble listbox and user saved settings list.
|
|
"""
|
|
|
|
def fix_name(name, mapper:dict): return next((new_name for old_name, new_name in mapper.items() if name in old_name), name)
|
|
|
|
new_vr_models = self.get_files_from_dir(VR_MODELS_DIR, PTH)
|
|
new_mdx_models = self.get_files_from_dir(MDX_MODELS_DIR, (ONNX, CKPT), is_mdxnet=True)
|
|
new_demucs_models = self.get_files_from_dir(DEMUCS_MODELS_DIR, (CKPT, '.gz', '.th')) + self.get_files_from_dir(DEMUCS_NEWER_REPO_DIR, YAML)
|
|
new_ensembles_found = self.get_files_from_dir(ENSEMBLE_CACHE_DIR, JSON)
|
|
new_settings_found = self.get_files_from_dir(SETTINGS_CACHE_DIR, JSON)
|
|
new_models_found = new_vr_models + new_mdx_models + new_demucs_models
|
|
is_online = self.is_online_model_menu
|
|
|
|
def loop_directories(option_menu:ComboBoxMenu, option_var, model_list, model_type, name_mapper=None):
|
|
current_selection = option_menu.get()
|
|
option_list = [fix_name(file_name, name_mapper) for file_name in model_list] if name_mapper else model_list
|
|
sorted_options = natsort.natsorted(option_list)
|
|
option_list_option_menu = sorted_options + [OPT_SEPARATOR, DOWNLOAD_MORE] if self.is_online else sorted_options
|
|
|
|
if not option_list and self.is_online:
|
|
option_list_option_menu = [option for option in option_list_option_menu if option != OPT_SEPARATOR]
|
|
|
|
option_menu['values'] = option_list_option_menu
|
|
option_menu.set(current_selection)
|
|
option_menu.update_dropdown_size(option_list, model_type)
|
|
|
|
if self.is_root_defined_var.get() and model_type == MDX_ARCH_TYPE and self.chosen_process_method_var.get() == MDX_ARCH_TYPE:
|
|
self.selection_action_models_sub(current_selection, model_type, option_var)
|
|
|
|
return tuple(f"{model_type}{ENSEMBLE_PARTITION}{model_name}" for model_name in sorted_options)
|
|
|
|
if new_models_found != self.last_found_models or is_online != self.is_online:
|
|
self.model_data_table = []
|
|
|
|
vr_model_list = loop_directories(self.vr_model_Option, self.vr_model_var, new_vr_models, VR_ARCH_TYPE, name_mapper=None)
|
|
mdx_model_list = loop_directories(self.mdx_net_model_Option, self.mdx_net_model_var, new_mdx_models, MDX_ARCH_TYPE, name_mapper=self.mdx_name_select_MAPPER)
|
|
demucs_model_list = loop_directories(self.demucs_model_Option, self.demucs_model_var, new_demucs_models, DEMUCS_ARCH_TYPE, name_mapper=self.demucs_name_select_MAPPER)
|
|
|
|
self.ensemble_model_list = vr_model_list + mdx_model_list + demucs_model_list
|
|
self.default_change_model_list = vr_model_list + mdx_model_list
|
|
self.last_found_models = new_models_found
|
|
self.is_online_model_menu = self.is_online
|
|
|
|
if not self.chosen_ensemble_var.get() == CHOOSE_ENSEMBLE_OPTION:
|
|
self.selection_action_chosen_ensemble(self.chosen_ensemble_var.get())
|
|
else:
|
|
if not self.ensemble_main_stem_var.get() == CHOOSE_STEM_PAIR:
|
|
self.selection_action_ensemble_stems(self.ensemble_main_stem_var.get(), auto_update=self.ensemble_listbox_get_all_selected_models())
|
|
else:
|
|
self.ensemble_listbox_clear_and_insert_new(self.ensemble_model_list)
|
|
|
|
self.last_found_ensembles = self.update_menus(option_widget=self.chosen_ensemble_Option,
|
|
style_name='savedensembles',
|
|
command=None,
|
|
new_items=new_ensembles_found,
|
|
last_items=self.last_found_ensembles,
|
|
base_options=ENSEMBLE_OPTIONS
|
|
)
|
|
|
|
self.last_found_settings = self.update_menus(option_widget=self.save_current_settings_Option,
|
|
style_name='savedsettings',
|
|
command=None,
|
|
new_items=new_settings_found,
|
|
last_items=self.last_found_settings,
|
|
base_options=SAVE_SET_OPTIONS
|
|
)
|
|
|
|
def update_main_widget_states_mdx(self):
|
|
if not self.mdx_net_model_var.get() == DOWNLOAD_MORE:
|
|
self.update_main_widget_states()
|
|
|
|
def move_widget_offscreen(self, widget, step=10):
|
|
current_x = widget.winfo_x()
|
|
current_y = widget.winfo_y()
|
|
if current_x > -1000: # assuming -1000 is your off-screen target
|
|
widget.place(x=current_x - step, y=current_y)
|
|
widget.after(10, lambda: self.move_widget_offscreen(widget, step))
|
|
|
|
def update_main_widget_states(self):
|
|
"""Updates main widget states based on chosen process method"""
|
|
|
|
def place_widgets(*widgets):
|
|
for widget in widgets:
|
|
widget()
|
|
|
|
def general_shared_buttons():
|
|
place_widgets(self.is_gpu_conversion_Option_place,
|
|
self.model_sample_mode_Option_place)
|
|
|
|
def stem_save_options():
|
|
place_widgets(self.is_primary_stem_only_Option_place,
|
|
self.is_secondary_stem_only_Option_place)
|
|
|
|
def stem_save_demucs_options():
|
|
place_widgets(self.is_primary_stem_only_Demucs_Option_place,
|
|
self.is_secondary_stem_only_Demucs_Option_place)
|
|
|
|
def no_ensemble_shared():
|
|
place_widgets(self.save_current_settings_Label_place,
|
|
self.save_current_settings_Option_place)
|
|
|
|
process_method = self.chosen_process_method_var.get()
|
|
audio_tool = self.chosen_audio_tool_var.get()
|
|
|
|
for widget in self.GUI_LIST:
|
|
widget.place(x=-1000, y=-1000)
|
|
|
|
if process_method == MDX_ARCH_TYPE:
|
|
place_widgets(self.mdx_net_model_Label_place,
|
|
self.mdx_net_model_Option_place,
|
|
general_shared_buttons,
|
|
stem_save_options,
|
|
no_ensemble_shared)
|
|
elif process_method == VR_ARCH_PM:
|
|
place_widgets(self.vr_model_Label_place, self.vr_model_Option_place,
|
|
self.aggression_setting_Label_place,
|
|
self.aggression_setting_Option_place,
|
|
self.window_size_Label_place,
|
|
self.window_size_Option_place,
|
|
general_shared_buttons,
|
|
stem_save_options,
|
|
no_ensemble_shared)
|
|
elif process_method == DEMUCS_ARCH_TYPE:
|
|
place_widgets(self.demucs_model_Label_place,
|
|
self.demucs_model_Option_place,
|
|
self.demucs_stems_Label_place,
|
|
self.demucs_stems_Option_place,
|
|
self.segment_Label_place,
|
|
self.segment_Option_place,
|
|
general_shared_buttons,
|
|
stem_save_demucs_options,
|
|
no_ensemble_shared)
|
|
elif process_method == AUDIO_TOOLS:
|
|
place_widgets(self.chosen_audio_tool_Label_place,
|
|
self.chosen_audio_tool_Option_place)
|
|
|
|
if audio_tool == ALIGN_INPUTS:
|
|
self.file_one_sub_var.set(FILE_ONE_MAIN_LABEL)
|
|
self.file_two_sub_var.set(FILE_TWO_MAIN_LABEL)
|
|
elif audio_tool == MATCH_INPUTS:
|
|
self.file_one_sub_var.set(FILE_ONE_MATCH_MAIN_LABEL)
|
|
self.file_two_sub_var.set(FILE_TWO_MATCH_MAIN_LABEL)
|
|
|
|
audio_tool_options = {
|
|
MANUAL_ENSEMBLE: [self.choose_algorithm_Label_place,
|
|
self.choose_algorithm_Option_place,
|
|
self.is_wav_ensemble_Option_place],
|
|
TIME_STRETCH: [lambda: self.model_sample_mode_Option_place(rely=5),
|
|
self.time_stretch_rate_Label_place,
|
|
self.time_stretch_rate_Option_place],
|
|
CHANGE_PITCH: [self.is_time_correction_Option_place,
|
|
lambda: self.model_sample_mode_Option_place(rely=6),
|
|
self.pitch_rate_Label_place,
|
|
self.pitch_rate_Option_place],
|
|
ALIGN_INPUTS: [self.fileOne_Label_place,
|
|
self.fileOne_Entry_place,
|
|
self.fileTwo_Label_place,
|
|
self.fileTwo_Entry_place,
|
|
self.fileOne_Open_place,
|
|
self.fileTwo_Open_place,
|
|
self.intro_analysis_Label_place,
|
|
self.intro_analysis_Option_place,
|
|
self.db_analysis_Label_place,
|
|
self.db_analysis_Option_place,
|
|
self.time_window_Label_place,
|
|
self.time_window_Option_place],
|
|
MATCH_INPUTS: [self.fileOne_Label_place,
|
|
self.fileOne_Entry_place,
|
|
self.fileTwo_Label_place,
|
|
self.fileTwo_Entry_place,
|
|
self.fileOne_Open_place,
|
|
self.fileTwo_Open_place,
|
|
self.wav_type_set_Label_place,
|
|
self.wav_type_set_Option_place],
|
|
}
|
|
place_widgets(*audio_tool_options.get(audio_tool, []))
|
|
elif process_method == ENSEMBLE_MODE:
|
|
place_widgets(self.chosen_ensemble_Label_place,
|
|
self.chosen_ensemble_Option_place,
|
|
self.ensemble_main_stem_Label_place,
|
|
self.ensemble_main_stem_Option_place,
|
|
self.ensemble_type_Label_place,
|
|
self.ensemble_type_Option_place,
|
|
self.ensemble_listbox_Label_place,
|
|
self.ensemble_listbox_Option_place,
|
|
self.ensemble_listbox_Option_pack,
|
|
general_shared_buttons,
|
|
stem_save_options)
|
|
|
|
if not self.is_gpu_available:
|
|
self.is_gpu_conversion_Disable()
|
|
|
|
self.update_inputPaths()
|
|
|
|
def update_button_states(self):
|
|
"""Updates the available stems for selected Demucs model"""
|
|
|
|
if self.chosen_process_method_var.get() == DEMUCS_ARCH_TYPE:
|
|
if self.demucs_stems_var.get() == ALL_STEMS:
|
|
self.update_stem_checkbox_labels(PRIMARY_STEM, demucs=True)
|
|
elif self.demucs_stems_var.get() == VOCAL_STEM:
|
|
self.update_stem_checkbox_labels(VOCAL_STEM, demucs=True, is_disable_demucs_boxes=False)
|
|
self.is_stem_only_Demucs_Options_Enable()
|
|
else:
|
|
self.is_stem_only_Demucs_Options_Enable()
|
|
|
|
#self.demucs_stems_Option['menu'].delete(0,'end')
|
|
|
|
if not self.demucs_model_var.get() == CHOOSE_MODEL:
|
|
if DEMUCS_UVR_MODEL in self.demucs_model_var.get():
|
|
stems = DEMUCS_2_STEM_OPTIONS
|
|
elif DEMUCS_6_STEM_MODEL in self.demucs_model_var.get():
|
|
stems = DEMUCS_6_STEM_OPTIONS
|
|
else:
|
|
stems = DEMUCS_4_STEM_OPTIONS
|
|
|
|
self.demucs_stems_Option['values'] = stems
|
|
self.demucs_stems_Option.command(lambda e:self.update_stem_checkbox_labels(self.demucs_stems_var.get(), demucs=True))
|
|
|
|
def update_button_states_mdx(self, model_stems):
|
|
"""Updates the available stems for selected Demucs model"""
|
|
|
|
model_stems = [stem for stem in model_stems]
|
|
|
|
if len(model_stems) >= 3:
|
|
model_stems.insert(0, ALL_STEMS)
|
|
self.mdxnet_stems_var.set(ALL_STEMS)
|
|
else:
|
|
self.mdxnet_stems_var.set(model_stems[0])
|
|
|
|
if self.mdxnet_stems_var.get() == ALL_STEMS:
|
|
self.update_stem_checkbox_labels(PRIMARY_STEM, disable_boxes=True)
|
|
elif self.mdxnet_stems_var.get() == VOCAL_STEM:
|
|
self.update_stem_checkbox_labels(VOCAL_STEM)
|
|
self.is_stem_only_Options_Enable()
|
|
else:
|
|
self.is_stem_only_Options_Enable()
|
|
|
|
if not self.mdx_net_model_var.get() == CHOOSE_MODEL:
|
|
self.mdxnet_stems_Option['values'] = model_stems
|
|
self.mdxnet_stems_Option.command(lambda e:self.update_stem_checkbox_labels(self.mdxnet_stems_var.get()))
|
|
|
|
def update_stem_checkbox_labels(self, selection, demucs=False, disable_boxes=False, is_disable_demucs_boxes=True):
|
|
"""Updates the "save only" checkboxes based on the model selected"""
|
|
|
|
stem_text = self.is_primary_stem_only_Text_var, self.is_secondary_stem_only_Text_var
|
|
|
|
if selection == ALL_STEMS:
|
|
selection = PRIMARY_STEM
|
|
else:
|
|
self.is_stem_only_Options_Enable()
|
|
|
|
if disable_boxes or selection == PRIMARY_STEM:
|
|
self.is_primary_stem_only_Option.configure(state=tk.DISABLED)
|
|
self.is_secondary_stem_only_Option.configure(state=tk.DISABLED)
|
|
self.is_primary_stem_only_var.set(False)
|
|
self.is_secondary_stem_only_var.set(False)
|
|
else:
|
|
self.is_primary_stem_only_Option.configure(state=tk.NORMAL)
|
|
self.is_secondary_stem_only_Option.configure(state=tk.NORMAL)
|
|
|
|
if demucs:
|
|
stem_text = self.is_primary_stem_only_Demucs_Text_var, self.is_secondary_stem_only_Demucs_Text_var
|
|
|
|
if is_disable_demucs_boxes:
|
|
self.is_primary_stem_only_Demucs_Option.configure(state=tk.DISABLED)
|
|
self.is_secondary_stem_only_Demucs_Option.configure(state=tk.DISABLED)
|
|
self.is_primary_stem_only_Demucs_var.set(False)
|
|
self.is_secondary_stem_only_Demucs_var.set(False)
|
|
|
|
if not selection == PRIMARY_STEM:
|
|
self.is_primary_stem_only_Demucs_Option.configure(state=tk.NORMAL)
|
|
self.is_secondary_stem_only_Demucs_Option.configure(state=tk.NORMAL)
|
|
|
|
stem_text[0].set(f"{selection} Only")
|
|
stem_text[1].set(f"{secondary_stem(selection)} Only")
|
|
|
|
def update_ensemble_algorithm_menu(self, is_4_stem=False):
|
|
options = ENSEMBLE_TYPE_4_STEM if is_4_stem else ENSEMBLE_TYPE
|
|
|
|
if not "/" in self.ensemble_type_var.get() or is_4_stem:
|
|
self.ensemble_type_var.set(options[0])
|
|
|
|
self.ensemble_type_Option["values"] = options
|
|
|
|
def selection_action(self, event, option_var, is_mdx_net=False):
|
|
selected_value = event.widget.get()
|
|
selected_value = CHOOSE_MODEL if selected_value == OPT_SEPARATOR else selected_value
|
|
option_var.set(selected_value)
|
|
if is_mdx_net:
|
|
self.update_main_widget_states_mdx()
|
|
self.selection_action_models(selected_value)
|
|
|
|
def selection_action_models(self, selection):
|
|
"""Accepts model names and verifies their state."""
|
|
|
|
# Handle different selections.
|
|
if selection in CHOOSE_MODEL:
|
|
self.update_stem_checkbox_labels(PRIMARY_STEM, disable_boxes=True)
|
|
else:
|
|
self.is_stem_only_Options_Enable()
|
|
|
|
# Process method matching current selection.
|
|
self._handle_model_by_chosen_method(selection)
|
|
|
|
# Handle Ensemble mode case.
|
|
if self.chosen_process_method_var.get() == ENSEMBLE_MODE:
|
|
return self._handle_ensemble_mode_selection(selection)
|
|
|
|
if not self.is_menu_settings_open and selection == DOWNLOAD_MORE:
|
|
self.update_checkbox_text()
|
|
self.menu_settings(select_tab_3=True)
|
|
|
|
def _handle_model_by_chosen_method(self, selection):
|
|
"""Handles model selection based on the currently chosen method."""
|
|
current_method = self.chosen_process_method_var.get()
|
|
model_var = self.method_mapper.get(current_method)
|
|
if model_var:
|
|
self.selection_action_models_sub(selection, current_method, model_var)
|
|
|
|
def _handle_ensemble_mode_selection(self, selection):
|
|
"""Handles the case where the current method is 'ENSEMBLE_MODE'."""
|
|
model_data = self.assemble_model_data(selection, ENSEMBLE_CHECK)[0]
|
|
if not model_data.model_status:
|
|
return self.model_stems_list.index(selection)
|
|
return False
|
|
|
|
def selection_action_models_sub(self, selection, ai_type, var: tk.StringVar):
|
|
"""Takes input directly from the selection_action_models parent function"""
|
|
|
|
if selection == DOWNLOAD_MORE:
|
|
is_model_status = False
|
|
else:
|
|
model_data = self.assemble_model_data(selection, ai_type)[0]
|
|
is_model_status = model_data.model_status
|
|
|
|
if not is_model_status:
|
|
var.set(CHOOSE_MODEL)
|
|
if ai_type == MDX_ARCH_TYPE:
|
|
self.mdx_segment_size_Label_place()
|
|
self.mdx_segment_size_Option_place()
|
|
self.overlap_mdx_Label_place()
|
|
self.overlap_mdx_Option_place()
|
|
self.update_stem_checkbox_labels(PRIMARY_STEM, disable_boxes=True)
|
|
else:
|
|
if ai_type == DEMUCS_ARCH_TYPE:
|
|
if not self.demucs_stems_var.get().lower() in model_data.demucs_source_list:
|
|
self.demucs_stems_var.set(ALL_STEMS if model_data.demucs_stem_count == 4 else VOCAL_STEM)
|
|
|
|
self.update_button_states()
|
|
else:
|
|
if model_data.is_mdx_c and len(model_data.mdx_model_stems) >= 2:
|
|
if len(model_data.mdx_model_stems) >= 3:
|
|
self.mdxnet_stems_Label_place()
|
|
self.mdxnet_stems_Option_place()
|
|
else:
|
|
self.mdx_segment_size_Label_place()
|
|
self.mdx_segment_size_Option_place()
|
|
self.overlap_mdx_Label_place()
|
|
self.overlap_mdx23_Option_place()
|
|
self.update_button_states_mdx(model_data.mdx_model_stems)
|
|
else:
|
|
if ai_type == MDX_ARCH_TYPE:
|
|
self.mdx_segment_size_Label_place()
|
|
self.mdx_segment_size_Option_place()
|
|
self.overlap_mdx_Label_place()
|
|
self.overlap_mdx_Option_place()
|
|
|
|
stem = model_data.primary_stem
|
|
self.update_stem_checkbox_labels(stem)
|
|
|
|
def selection_action_process_method(self, selection, from_widget=False, is_from_conv_menu=False):
|
|
"""Checks model and variable status when toggling between process methods"""
|
|
|
|
if is_from_conv_menu:
|
|
self.update_main_widget_states()
|
|
|
|
if from_widget:
|
|
self.save_current_settings_var.set(CHOOSE_ENSEMBLE_OPTION)
|
|
|
|
if selection == ENSEMBLE_MODE:
|
|
ensemble_choice = self.ensemble_main_stem_var.get()
|
|
if ensemble_choice in [CHOOSE_STEM_PAIR, FOUR_STEM_ENSEMBLE, MULTI_STEM_ENSEMBLE]:
|
|
self.update_stem_checkbox_labels(PRIMARY_STEM, disable_boxes=True)
|
|
else:
|
|
self.update_stem_checkbox_labels(self.return_ensemble_stems(is_primary=True))
|
|
self.is_stem_only_Options_Enable()
|
|
return
|
|
|
|
for method_type, model_var in self.method_mapper.items():
|
|
if method_type in selection:
|
|
self.selection_action_models(model_var.get())
|
|
break
|
|
|
|
def selection_action_chosen_ensemble(self, selection):
|
|
"""Activates specific actions depending on selected ensemble option"""
|
|
|
|
if selection not in ENSEMBLE_OPTIONS:
|
|
self.selection_action_chosen_ensemble_load_saved(selection)
|
|
elif selection == SAVE_ENSEMBLE:
|
|
self.chosen_ensemble_var.set(CHOOSE_ENSEMBLE_OPTION)
|
|
self.pop_up_save_ensemble()
|
|
elif selection == OPT_SEPARATOR_SAVE:
|
|
self.chosen_ensemble_var.set(CHOOSE_ENSEMBLE_OPTION)
|
|
elif selection == CLEAR_ENSEMBLE:
|
|
self.ensemble_listbox_Option.selection_clear(0, 'end')
|
|
self.chosen_ensemble_var.set(CHOOSE_ENSEMBLE_OPTION)
|
|
|
|
def selection_action_chosen_ensemble_load_saved(self, saved_ensemble):
|
|
"""Loads the data from selected saved ensemble"""
|
|
|
|
saved_data = None
|
|
saved_ensemble = saved_ensemble.replace(" ", "_")
|
|
saved_ensemble_path = os.path.join(ENSEMBLE_CACHE_DIR, f'{saved_ensemble}.json')
|
|
|
|
if os.path.isfile(saved_ensemble_path):
|
|
saved_data = json.load(open(saved_ensemble_path))
|
|
|
|
if saved_data:
|
|
self.selection_action_ensemble_stems(saved_data['ensemble_main_stem'], from_menu=False)
|
|
self.ensemble_main_stem_var.set(saved_data['ensemble_main_stem'])
|
|
self.ensemble_type_var.set(saved_data['ensemble_type'])
|
|
self.saved_model_list = saved_data['selected_models']
|
|
|
|
for saved_model in self.saved_model_list:
|
|
status = self.assemble_model_data(saved_model, ENSEMBLE_CHECK)[0].model_status
|
|
if not status:
|
|
self.saved_model_list.remove(saved_model)
|
|
|
|
indexes = self.ensemble_listbox_get_indexes_for_files(self.model_stems_list, self.saved_model_list)
|
|
|
|
for i in indexes:
|
|
self.ensemble_listbox_Option.selection_set(i)
|
|
|
|
self.update_checkbox_text()
|
|
|
|
def selection_action_ensemble_stems(self, selection: str, from_menu=True, auto_update=None):
|
|
"""Filters out all models from ensemble listbox that are incompatible with selected ensemble stem"""
|
|
|
|
is_multi_stem = False
|
|
|
|
if not selection == CHOOSE_STEM_PAIR:
|
|
if selection in [FOUR_STEM_ENSEMBLE, MULTI_STEM_ENSEMBLE]:
|
|
self.update_stem_checkbox_labels(PRIMARY_STEM, disable_boxes=True)
|
|
self.update_ensemble_algorithm_menu(is_4_stem=True)
|
|
self.ensemble_primary_stem = PRIMARY_STEM
|
|
self.ensemble_secondary_stem = SECONDARY_STEM
|
|
is_4_stem_check = True
|
|
if selection == MULTI_STEM_ENSEMBLE:
|
|
is_multi_stem = True
|
|
else:
|
|
self.update_ensemble_algorithm_menu()
|
|
self.is_stem_only_Options_Enable()
|
|
stems = selection.partition("/")
|
|
self.update_stem_checkbox_labels(stems[0])
|
|
self.ensemble_primary_stem = stems[0]
|
|
self.ensemble_secondary_stem = stems[2]
|
|
is_4_stem_check = False
|
|
|
|
self.model_stems_list = self.model_list(self.ensemble_primary_stem, self.ensemble_secondary_stem, is_4_stem_check=is_4_stem_check, is_multi_stem=is_multi_stem)
|
|
self.ensemble_listbox_Option.configure(state=tk.NORMAL)
|
|
self.ensemble_listbox_clear_and_insert_new(self.model_stems_list)
|
|
|
|
if auto_update:
|
|
indexes = self.ensemble_listbox_get_indexes_for_files(self.model_stems_list, auto_update)
|
|
self.ensemble_listbox_select_from_indexs(indexes)
|
|
else:
|
|
self.ensemble_listbox_Option.configure(state=tk.DISABLED)
|
|
self.update_stem_checkbox_labels(PRIMARY_STEM, disable_boxes=True)
|
|
self.model_stems_list = ()
|
|
|
|
if from_menu:
|
|
self.chosen_ensemble_var.set(CHOOSE_ENSEMBLE_OPTION)
|
|
|
|
def selection_action_saved_settings(self, selection, process_method=None):
|
|
"""Activates specific action based on the selected settings from the saved settings selections"""
|
|
|
|
if self.thread_check(self.active_processing_thread):
|
|
self.error_dialoge(SET_TO_ANY_PROCESS_ERROR)
|
|
return
|
|
|
|
chosen_process_method = self.chosen_process_method_var.get()
|
|
if process_method:
|
|
chosen_process_method = process_method
|
|
|
|
if selection in SAVE_SET_OPTIONS:
|
|
self.handle_special_options(selection, chosen_process_method)
|
|
else:
|
|
self.handle_saved_settings(selection, chosen_process_method)
|
|
|
|
self.update_checkbox_text()
|
|
|
|
def handle_special_options(self, selection, process_method):
|
|
"""Handles actions for special options."""
|
|
|
|
if selection == SAVE_SETTINGS:
|
|
self.save_current_settings_var.set(SELECT_SAVED_SET)
|
|
self.pop_up_save_current_settings()
|
|
|
|
elif selection == RESET_TO_DEFAULT:
|
|
self.save_current_settings_var.set(SELECT_SAVED_SET)
|
|
self.load_saved_settings(DEFAULT_DATA, process_method)
|
|
|
|
elif selection == OPT_SEPARATOR_SAVE:
|
|
self.save_current_settings_var.set(SELECT_SAVED_SET)
|
|
|
|
def handle_saved_settings(self, selection, process_method):
|
|
"""Handles actions for saved settings."""
|
|
|
|
selection = selection.replace(" ", "_")
|
|
saved_ensemble_path = os.path.join(SETTINGS_CACHE_DIR, f'{selection}.json')
|
|
|
|
if os.path.isfile(saved_ensemble_path):
|
|
with open(saved_ensemble_path, 'r') as file:
|
|
saved_data = json.load(file)
|
|
|
|
if saved_data:
|
|
self.load_saved_settings(saved_data, process_method)
|
|
|
|
#--Processing Methods--
|
|
|
|
def process_input_selections(self):
|
|
"""Grabbing all audio files from selected directories."""
|
|
|
|
input_list = []
|
|
|
|
ext = FFMPEG_EXT if not self.is_accept_any_input_var.get() else ANY_EXT
|
|
|
|
for i in self.inputPaths:
|
|
if os.path.isfile(i):
|
|
if i.endswith(ext):
|
|
input_list.append(i)
|
|
for root, dirs, files in os.walk(i):
|
|
for file in files:
|
|
if file.endswith(ext):
|
|
file = os.path.join(root, file)
|
|
if os.path.isfile(file):
|
|
input_list.append(file)
|
|
|
|
self.inputPaths = tuple(input_list)
|
|
|
|
def process_check_wav_type(self):
|
|
if self.wav_type_set_var.get() == '32-bit Float':
|
|
self.wav_type_set = 'FLOAT'
|
|
elif self.wav_type_set_var.get() == '64-bit Float':#
|
|
self.wav_type_set = 'FLOAT' if not self.save_format_var.get() == WAV else 'DOUBLE'
|
|
else:
|
|
self.wav_type_set = self.wav_type_set_var.get()
|
|
|
|
def process_preliminary_checks(self):
|
|
"""Verifies a valid model is chosen"""
|
|
|
|
self.process_check_wav_type()
|
|
|
|
if self.chosen_process_method_var.get() == ENSEMBLE_MODE:
|
|
continue_process = lambda:False if len(self.ensemble_listbox_get_all_selected_models()) <= 1 else True
|
|
if self.chosen_process_method_var.get() == VR_ARCH_PM:
|
|
continue_process = lambda:False if self.vr_model_var.get() == CHOOSE_MODEL else True
|
|
if self.chosen_process_method_var.get() == MDX_ARCH_TYPE:
|
|
continue_process = lambda:False if self.mdx_net_model_var.get() == CHOOSE_MODEL else True
|
|
if self.chosen_process_method_var.get() == DEMUCS_ARCH_TYPE:
|
|
continue_process = lambda:False if self.demucs_model_var.get() == CHOOSE_MODEL else True
|
|
|
|
return continue_process()
|
|
|
|
def process_storage_check(self):
|
|
"""Verifies storage requirments"""
|
|
|
|
total, used, free = shutil.disk_usage("/")
|
|
|
|
space_details = f'Detected Total Space: {int(total/1.074e+9)} GB\'s\n' +\
|
|
f'Detected Used Space: {int(used/1.074e+9)} GB\'s\n' +\
|
|
f'Detected Free Space: {int(free/1.074e+9)} GB\'s\n'
|
|
|
|
appropriate_storage = True
|
|
|
|
if int(free/1.074e+9) <= int(2):
|
|
self.error_dialoge([STORAGE_ERROR[0], f'{STORAGE_ERROR[1]}{space_details}'])
|
|
appropriate_storage = False
|
|
|
|
if int(free/1.074e+9) in [3, 4, 5, 6, 7, 8]:
|
|
appropriate_storage = self.message_box([STORAGE_WARNING[0], f'{STORAGE_WARNING[1]}{space_details}{CONFIRM_WARNING}'])
|
|
|
|
return appropriate_storage
|
|
|
|
def process_initialize(self):
|
|
"""Verifies the input/output directories are valid and prepares to thread the main process."""
|
|
|
|
if not (
|
|
self.chosen_process_method_var.get() == AUDIO_TOOLS
|
|
and self.chosen_audio_tool_var.get() in [ALIGN_INPUTS, MATCH_INPUTS]
|
|
and self.fileOneEntry_var.get()
|
|
and self.fileTwoEntry_var.get()
|
|
) and not (
|
|
self.inputPaths and os.path.isfile(self.inputPaths[0])
|
|
):
|
|
self.error_dialoge(INVALID_INPUT)
|
|
return
|
|
|
|
|
|
if not os.path.isdir(self.export_path_var.get()):
|
|
self.error_dialoge(INVALID_EXPORT)
|
|
return
|
|
|
|
if not self.process_storage_check():
|
|
return
|
|
|
|
if self.chosen_process_method_var.get() != AUDIO_TOOLS:
|
|
if not self.process_preliminary_checks():
|
|
error_msg = INVALID_ENSEMBLE if self.chosen_process_method_var.get() == ENSEMBLE_MODE else INVALID_MODEL
|
|
self.error_dialoge(error_msg)
|
|
return
|
|
target_function = self.process_start
|
|
else:
|
|
target_function = self.process_tool_start
|
|
|
|
self.active_processing_thread = KThread(target=target_function)
|
|
self.active_processing_thread.start()
|
|
|
|
def process_button_init(self):
|
|
self.auto_save()
|
|
self.conversion_Button_Text_var.set(WAIT_PROCESSING)
|
|
self.conversion_Button.configure(state=tk.DISABLED)
|
|
self.command_Text.clear()
|
|
|
|
def process_get_baseText(self, total_files, file_num, is_dual=False):
|
|
"""Create the base text for the command widget"""
|
|
|
|
init_text = 'Files' if is_dual else 'File'
|
|
|
|
text = '{init_text} {file_num}/{total_files} '.format(init_text=init_text,
|
|
file_num=file_num,
|
|
total_files=total_files)
|
|
|
|
return text
|
|
|
|
def process_update_progress(self, total_files, step: float = 1):
|
|
"""Calculate the progress for the progress widget in the GUI"""
|
|
|
|
total_count = self.true_model_count * total_files
|
|
base = (100 / total_count)
|
|
progress = base * self.iteration - base
|
|
progress += base * step
|
|
|
|
self.progress_bar_main_var.set(progress)
|
|
|
|
self.conversion_Button_Text_var.set(f'Process Progress: {int(progress)}%')
|
|
|
|
def confirm_stop_process(self):
|
|
"""Asks for confirmation before halting active process"""
|
|
|
|
self.auto_save()
|
|
|
|
if self.thread_check(self.active_processing_thread):
|
|
confirm = messagebox.askyesno(parent=root, title=STOP_PROCESS_CONFIRM[0], message=STOP_PROCESS_CONFIRM[1])
|
|
|
|
if confirm:
|
|
try:
|
|
self.active_processing_thread.terminate()
|
|
finally:
|
|
self.is_process_stopped = True
|
|
self.command_Text.write(PROCESS_STOPPED_BY_USER)
|
|
else:
|
|
self.clear_cache_torch = True
|
|
|
|
def process_end(self, error=None):
|
|
"""End of process actions"""
|
|
|
|
self.auto_save()
|
|
self.cached_sources_clear()
|
|
self.clear_cache_torch = True
|
|
self.conversion_Button_Text_var.set(START_PROCESSING)
|
|
self.conversion_Button.configure(state=tk.NORMAL)
|
|
self.progress_bar_main_var.set(0)
|
|
|
|
if error:
|
|
error_message_box_text = f'{error_dialouge(error)}{ERROR_OCCURED[1]}'
|
|
confirm = messagebox.askyesno(parent=root,
|
|
title=ERROR_OCCURED[0],
|
|
message=error_message_box_text)
|
|
|
|
if confirm:
|
|
self.is_confirm_error_var.set(True)
|
|
self.clear_cache_torch = True
|
|
|
|
self.clear_cache_torch = True
|
|
|
|
if MODEL_MISSING_CHECK in error_message_box_text:
|
|
self.update_checkbox_text()
|
|
|
|
def process_tool_start(self):
|
|
"""Start the conversion for all the given mp3 and wav files"""
|
|
|
|
def time_elapsed():
|
|
return f'Time Elapsed: {time.strftime("%H:%M:%S", time.gmtime(int(time.perf_counter() - stime)))}'
|
|
|
|
def get_audio_file_base(audio_file):
|
|
if audio_tool.audio_tool == MANUAL_ENSEMBLE:
|
|
return f'{os.path.splitext(os.path.basename(inputPaths[0]))[0]}'
|
|
elif audio_tool.audio_tool in [ALIGN_INPUTS, MATCH_INPUTS]:
|
|
return f'{os.path.splitext(os.path.basename(audio_file[0]))[0]}'
|
|
else:
|
|
return f'{os.path.splitext(os.path.basename(audio_file))[0]}'
|
|
|
|
def handle_ensemble(inputPaths, audio_file_base):
|
|
self.progress_bar_main_var.set(50)
|
|
if self.choose_algorithm_var.get() == COMBINE_INPUTS:
|
|
audio_tool.combine_audio(inputPaths, audio_file_base)
|
|
else:
|
|
audio_tool.ensemble_manual(inputPaths, audio_file_base)
|
|
self.progress_bar_main_var.set(100)
|
|
self.command_Text.write(DONE)
|
|
|
|
def handle_alignment_match(audio_file, audio_file_base, command_Text, set_progress_bar):
|
|
audio_file_2_base = f'{os.path.splitext(os.path.basename(audio_file[1]))[0]}'
|
|
if audio_tool.audio_tool == MATCH_INPUTS:
|
|
audio_tool.match_inputs(audio_file, audio_file_base, command_Text)
|
|
else:
|
|
command_Text(f"{PROCESS_STARTING_TEXT}\n")
|
|
audio_tool.align_inputs(audio_file, audio_file_base, audio_file_2_base, command_Text, set_progress_bar)
|
|
self.progress_bar_main_var.set(base * file_num)
|
|
self.command_Text.write(f"{DONE}\n")
|
|
|
|
def handle_pitch_time_shift(audio_file, audio_file_base):
|
|
audio_tool.pitch_or_time_shift(audio_file, audio_file_base)
|
|
self.progress_bar_main_var.set(base * file_num)
|
|
self.command_Text.write(DONE)
|
|
|
|
multiple_files = False
|
|
stime = time.perf_counter()
|
|
self.process_button_init()
|
|
inputPaths = self.inputPaths
|
|
is_verified_audio = True
|
|
is_dual = False
|
|
is_model_sample_mode = self.model_sample_mode_var.get()
|
|
self.iteration = 0
|
|
self.true_model_count = 1
|
|
self.process_check_wav_type()
|
|
process_complete_text = PROCESS_COMPLETE
|
|
|
|
if self.chosen_audio_tool_var.get() in [ALIGN_INPUTS, MATCH_INPUTS]:
|
|
if self.DualBatch_inputPaths:
|
|
inputPaths = tuple(self.DualBatch_inputPaths)
|
|
else:
|
|
if not self.fileOneEntry_Full_var.get() or not self.fileTwoEntry_Full_var.get():
|
|
self.command_Text.write(NOT_ENOUGH_ERROR_TEXT)
|
|
self.process_end()
|
|
return
|
|
else:
|
|
inputPaths = [(self.fileOneEntry_Full_var.get(), self.fileTwoEntry_Full_var.get())]
|
|
|
|
try:
|
|
total_files = len(inputPaths)
|
|
if self.chosen_audio_tool_var.get() == TIME_STRETCH:
|
|
audio_tool = AudioTools(TIME_STRETCH)
|
|
self.progress_bar_main_var.set(2)
|
|
elif self.chosen_audio_tool_var.get() == CHANGE_PITCH:
|
|
audio_tool = AudioTools(CHANGE_PITCH)
|
|
self.progress_bar_main_var.set(2)
|
|
elif self.chosen_audio_tool_var.get() == MANUAL_ENSEMBLE:
|
|
if self.chosen_audio_tool_var.get() == MANUAL_ENSEMBLE:
|
|
audio_tool = Ensembler(is_manual_ensemble=True)
|
|
multiple_files = True
|
|
if total_files <= 1:
|
|
self.command_Text.write(NOT_ENOUGH_ERROR_TEXT)
|
|
self.process_end()
|
|
return
|
|
elif self.chosen_audio_tool_var.get() in [ALIGN_INPUTS, MATCH_INPUTS]:
|
|
audio_tool = AudioTools(self.chosen_audio_tool_var.get())
|
|
self.progress_bar_main_var.set(2)
|
|
is_dual = True
|
|
|
|
for file_num, audio_file in enumerate(inputPaths, start=1):
|
|
self.iteration += 1
|
|
base = (100 / total_files)
|
|
audio_file_base = get_audio_file_base(audio_file)
|
|
self.base_text = self.process_get_baseText(total_files=total_files, file_num=total_files if multiple_files else file_num, is_dual=is_dual)
|
|
command_Text = lambda text: self.command_Text.write(self.base_text + text)
|
|
|
|
set_progress_bar = lambda step, inference_iterations=0:self.process_update_progress(total_files=total_files, step=(step + (inference_iterations)))
|
|
|
|
if not self.verify_audio(audio_file):
|
|
error_text_console = f'{self.base_text}"{os.path.basename(audio_file)}\" {MISSING_MESS_TEXT}\n'
|
|
if total_files >= 2:
|
|
self.command_Text.write(f'\n{error_text_console}')
|
|
is_verified_audio = False
|
|
continue
|
|
|
|
audio_tool_action = audio_tool.audio_tool
|
|
if audio_tool_action not in [MANUAL_ENSEMBLE, ALIGN_INPUTS, MATCH_INPUTS]:
|
|
audio_file = self.create_sample(audio_file) if is_model_sample_mode else audio_file
|
|
self.command_Text.write(f'{NEW_LINE if file_num != 1 else NO_LINE}{self.base_text}"{os.path.basename(audio_file)}\".{NEW_LINES}')
|
|
elif audio_tool_action in [ALIGN_INPUTS, MATCH_INPUTS]:
|
|
text_write = ("File 1", "File 2") if audio_tool_action == ALIGN_INPUTS else ("Target", "Reference")
|
|
if audio_file[0] != audio_file[1]:
|
|
self.command_Text.write(f'{self.base_text}{text_write[0]}: "{os.path.basename(audio_file[0])}"{NEW_LINE}')
|
|
self.command_Text.write(f'{self.base_text}{text_write[1]}: "{os.path.basename(audio_file[1])}"{NEW_LINES}')
|
|
else:
|
|
self.command_Text.write(f'{self.base_text}{text_write[0]} & {text_write[1]} {SIMILAR_TEXT}{NEW_LINES}')
|
|
continue
|
|
elif audio_tool_action == MANUAL_ENSEMBLE:
|
|
for n, i in enumerate(inputPaths):
|
|
self.command_Text.write(f'File {n+1} "{os.path.basename(i)}"{NEW_LINE}')
|
|
self.command_Text.write(NEW_LINE)
|
|
|
|
is_verified_audio = True
|
|
|
|
if not audio_tool_action in [ALIGN_INPUTS, MATCH_INPUTS]:
|
|
command_Text(PROCESS_STARTING_TEXT)
|
|
|
|
if audio_tool_action == MANUAL_ENSEMBLE:
|
|
handle_ensemble(inputPaths, audio_file_base)
|
|
break
|
|
if audio_tool_action in [ALIGN_INPUTS, MATCH_INPUTS]:
|
|
process_complete_text = PROCESS_COMPLETE_2
|
|
handle_alignment_match(audio_file, audio_file_base, command_Text, set_progress_bar)
|
|
if audio_tool_action in [TIME_STRETCH, CHANGE_PITCH]:
|
|
handle_pitch_time_shift(audio_file, audio_file_base)
|
|
|
|
if total_files == 1 and not is_verified_audio:
|
|
self.command_Text.write(f'{error_text_console}\n{PROCESS_FAILED}')
|
|
self.command_Text.write(time_elapsed())
|
|
playsound(FAIL_CHIME) if self.is_task_complete_var.get() else None
|
|
else:
|
|
self.command_Text.write('{}{}'.format(process_complete_text, time_elapsed()))
|
|
playsound(COMPLETE_CHIME) if self.is_task_complete_var.get() else None
|
|
|
|
self.process_end()
|
|
|
|
except Exception as e:
|
|
self.error_log_var.set(error_text(self.chosen_audio_tool_var.get(), e))
|
|
self.command_Text.write(f'\n\n{PROCESS_FAILED}')
|
|
self.command_Text.write(time_elapsed())
|
|
playsound(FAIL_CHIME) if self.is_task_complete_var.get() else None
|
|
self.process_end(error=e)
|
|
|
|
def process_determine_secondary_model(self, process_method, main_model_primary_stem, is_primary_stem_only=False, is_secondary_stem_only=False):
|
|
"""Obtains the correct secondary model data for conversion."""
|
|
|
|
secondary_model_scale = None
|
|
secondary_model = tk.StringVar(value=NO_MODEL)
|
|
|
|
if process_method == VR_ARCH_TYPE:
|
|
secondary_model_vars = self.vr_secondary_model_vars
|
|
if process_method == MDX_ARCH_TYPE:
|
|
secondary_model_vars = self.mdx_secondary_model_vars
|
|
if process_method == DEMUCS_ARCH_TYPE:
|
|
secondary_model_vars = self.demucs_secondary_model_vars
|
|
|
|
if main_model_primary_stem in [VOCAL_STEM, INST_STEM]:
|
|
secondary_model = secondary_model_vars["voc_inst_secondary_model"]
|
|
secondary_model_scale = secondary_model_vars["voc_inst_secondary_model_scale"].get()
|
|
if main_model_primary_stem in [OTHER_STEM, NO_OTHER_STEM]:
|
|
secondary_model = secondary_model_vars["other_secondary_model"]
|
|
secondary_model_scale = secondary_model_vars["other_secondary_model_scale"].get()
|
|
if main_model_primary_stem in [DRUM_STEM, NO_DRUM_STEM]:
|
|
secondary_model = secondary_model_vars["drums_secondary_model"]
|
|
secondary_model_scale = secondary_model_vars["drums_secondary_model_scale"].get()
|
|
if main_model_primary_stem in [BASS_STEM, NO_BASS_STEM]:
|
|
secondary_model = secondary_model_vars["bass_secondary_model"]
|
|
secondary_model_scale = secondary_model_vars["bass_secondary_model_scale"].get()
|
|
|
|
if secondary_model_scale:
|
|
secondary_model_scale = float(secondary_model_scale)
|
|
|
|
if not secondary_model.get() == NO_MODEL:
|
|
secondary_model = ModelData(secondary_model.get(),
|
|
is_secondary_model=True,
|
|
primary_model_primary_stem=main_model_primary_stem,
|
|
is_primary_model_primary_stem_only=is_primary_stem_only,
|
|
is_primary_model_secondary_stem_only=is_secondary_stem_only)
|
|
if not secondary_model.model_status:
|
|
secondary_model = None
|
|
else:
|
|
secondary_model = None
|
|
|
|
return secondary_model, secondary_model_scale
|
|
|
|
def process_determine_demucs_pre_proc_model(self, primary_stem=None):
|
|
"""Obtains the correct pre-process secondary model data for conversion."""
|
|
|
|
# Check if a pre-process model is set and it's not the 'NO_MODEL' value
|
|
if self.demucs_pre_proc_model_var.get() != NO_MODEL and self.is_demucs_pre_proc_model_activate_var.get():
|
|
pre_proc_model = ModelData(self.demucs_pre_proc_model_var.get(),
|
|
primary_model_primary_stem=primary_stem,
|
|
is_pre_proc_model=True)
|
|
|
|
# Return the model if it's valid
|
|
if pre_proc_model.model_status:
|
|
return pre_proc_model
|
|
|
|
return None
|
|
|
|
def process_determine_vocal_split_model(self):
|
|
"""Obtains the correct vocal splitter secondary model data for conversion."""
|
|
|
|
# Check if a vocal splitter model is set and if it's not the 'NO_MODEL' value
|
|
if self.set_vocal_splitter_var.get() != NO_MODEL and self.is_set_vocal_splitter_var.get():
|
|
vocal_splitter_model = ModelData(self.set_vocal_splitter_var.get(), is_vocal_split_model=True)
|
|
|
|
# Return the model if it's valid
|
|
if vocal_splitter_model.model_status:
|
|
return vocal_splitter_model
|
|
|
|
return None
|
|
|
|
def check_only_selection_stem(self, checktype):
|
|
|
|
chosen_method = self.chosen_process_method_var.get()
|
|
is_demucs = chosen_method == DEMUCS_ARCH_TYPE#
|
|
|
|
stem_primary_label = self.is_primary_stem_only_Demucs_Text_var.get() if is_demucs else self.is_primary_stem_only_Text_var.get()
|
|
stem_primary_bool = self.is_primary_stem_only_Demucs_var.get() if is_demucs else self.is_primary_stem_only_var.get()
|
|
stem_secondary_label = self.is_secondary_stem_only_Demucs_Text_var.get() if is_demucs else self.is_secondary_stem_only_Text_var.get()
|
|
stem_secondary_bool = self.is_secondary_stem_only_Demucs_var.get() if is_demucs else self.is_secondary_stem_only_var.get()
|
|
|
|
if checktype == VOCAL_STEM_ONLY:
|
|
return not (
|
|
(not VOCAL_STEM_ONLY == stem_primary_label and stem_primary_bool) or
|
|
(not VOCAL_STEM_ONLY in stem_secondary_label and stem_secondary_bool)
|
|
)
|
|
elif checktype == INST_STEM_ONLY:
|
|
return (
|
|
(INST_STEM_ONLY == stem_primary_label and stem_primary_bool and self.is_save_inst_set_vocal_splitter_var.get() and self.set_vocal_splitter_var.get() != NO_MODEL) or
|
|
(INST_STEM_ONLY == stem_secondary_label and stem_secondary_bool and self.is_save_inst_set_vocal_splitter_var.get() and self.set_vocal_splitter_var.get() != NO_MODEL)
|
|
)
|
|
elif checktype == IS_SAVE_VOC_ONLY:
|
|
return (
|
|
(VOCAL_STEM_ONLY == stem_primary_label and stem_primary_bool) or
|
|
(VOCAL_STEM_ONLY == stem_secondary_label and stem_secondary_bool)
|
|
)
|
|
elif checktype == IS_SAVE_INST_ONLY:
|
|
return (
|
|
(INST_STEM_ONLY == stem_primary_label and stem_primary_bool) or
|
|
(INST_STEM_ONLY == stem_secondary_label and stem_secondary_bool)
|
|
)
|
|
|
|
def determine_voc_split(self, models):
|
|
is_vocal_active = self.check_only_selection_stem(VOCAL_STEM_ONLY) or self.check_only_selection_stem(INST_STEM_ONLY)
|
|
|
|
if self.set_vocal_splitter_var.get() != NO_MODEL and self.is_set_vocal_splitter_var.get() and is_vocal_active:
|
|
model_stems_list = self.model_list(VOCAL_STEM, INST_STEM, is_dry_check=True, is_check_vocal_split=True)
|
|
if any(model.model_basename in model_stems_list for model in models):
|
|
return 1
|
|
|
|
return 0
|
|
|
|
def process_start(self):
|
|
"""Start the conversion for all the given mp3 and wav files"""
|
|
|
|
stime = time.perf_counter()
|
|
time_elapsed = lambda:f'Time Elapsed: {time.strftime("%H:%M:%S", time.gmtime(int(time.perf_counter() - stime)))}'
|
|
export_path = self.export_path_var.get()
|
|
is_ensemble = False
|
|
self.true_model_count = 0
|
|
self.iteration = 0
|
|
is_verified_audio = True
|
|
self.process_button_init()
|
|
inputPaths = self.inputPaths
|
|
inputPath_total_len = len(inputPaths)
|
|
is_model_sample_mode = self.model_sample_mode_var.get()
|
|
|
|
try:
|
|
if self.chosen_process_method_var.get() == ENSEMBLE_MODE:
|
|
model, ensemble = self.assemble_model_data(), Ensembler()
|
|
export_path, is_ensemble = ensemble.ensemble_folder_name, True
|
|
if self.chosen_process_method_var.get() == VR_ARCH_PM:
|
|
model = self.assemble_model_data(self.vr_model_var.get(), VR_ARCH_TYPE)
|
|
if self.chosen_process_method_var.get() == MDX_ARCH_TYPE:
|
|
model = self.assemble_model_data(self.mdx_net_model_var.get(), MDX_ARCH_TYPE)
|
|
if self.chosen_process_method_var.get() == DEMUCS_ARCH_TYPE:
|
|
model = self.assemble_model_data(self.demucs_model_var.get(), DEMUCS_ARCH_TYPE)
|
|
|
|
self.cached_source_model_list_check(model)
|
|
|
|
true_model_4_stem_count = sum(m.demucs_4_stem_added_count if m.process_method == DEMUCS_ARCH_TYPE else 0 for m in model)
|
|
true_model_pre_proc_model_count = sum(2 if m.pre_proc_model_activated else 0 for m in model)
|
|
self.true_model_count = sum(2 if m.is_secondary_model_activated else 1 for m in model) + true_model_4_stem_count + true_model_pre_proc_model_count + self.determine_voc_split(model)
|
|
|
|
#print("self.true_model_count", self.true_model_count)
|
|
|
|
for file_num, audio_file in enumerate(inputPaths, start=1):
|
|
self.cached_sources_clear()
|
|
base_text = self.process_get_baseText(total_files=inputPath_total_len, file_num=file_num)
|
|
|
|
if self.verify_audio(audio_file):
|
|
audio_file = self.create_sample(audio_file) if is_model_sample_mode else audio_file
|
|
self.command_Text.write(f'{NEW_LINE if not file_num ==1 else NO_LINE}{base_text}"{os.path.basename(audio_file)}\".{NEW_LINES}')
|
|
is_verified_audio = True
|
|
else:
|
|
error_text_console = f'{base_text}"{os.path.basename(audio_file)}\" {MISSING_MESS_TEXT}\n'
|
|
self.command_Text.write(f'\n{error_text_console}') if inputPath_total_len >= 2 else None
|
|
self.iteration += self.true_model_count
|
|
is_verified_audio = False
|
|
continue
|
|
|
|
for current_model_num, current_model in enumerate(model, start=1):
|
|
self.iteration += 1
|
|
|
|
if is_ensemble:
|
|
self.command_Text.write(f'Ensemble Mode - {current_model.model_basename} - Model {current_model_num}/{len(model)}{NEW_LINES}')
|
|
|
|
model_name_text = f'({current_model.model_basename})' if not is_ensemble else ''
|
|
self.command_Text.write(base_text + f'{LOADING_MODEL_TEXT} {model_name_text}...')
|
|
|
|
set_progress_bar = lambda step, inference_iterations=0:self.process_update_progress(total_files=inputPath_total_len, step=(step + (inference_iterations)))
|
|
write_to_console = lambda progress_text, base_text=base_text:self.command_Text.write(base_text + progress_text)
|
|
|
|
audio_file_base = f"{file_num}_{os.path.splitext(os.path.basename(audio_file))[0]}"
|
|
audio_file_base = audio_file_base if not self.is_testing_audio_var.get() or is_ensemble else f"{round(time.time())}_{audio_file_base}"
|
|
audio_file_base = audio_file_base if not is_ensemble else f"{audio_file_base}_{current_model.model_basename}"
|
|
if not is_ensemble:
|
|
audio_file_base = audio_file_base if not self.is_add_model_name_var.get() else f"{audio_file_base}_{current_model.model_basename}"
|
|
|
|
if self.is_create_model_folder_var.get() and not is_ensemble:
|
|
export_path = os.path.join(Path(self.export_path_var.get()), current_model.model_basename, os.path.splitext(os.path.basename(audio_file))[0])
|
|
if not os.path.isdir(export_path):os.makedirs(export_path)
|
|
|
|
process_data = {
|
|
'model_data': current_model,
|
|
'export_path': export_path,
|
|
'audio_file_base': audio_file_base,
|
|
'audio_file': audio_file,
|
|
'set_progress_bar': set_progress_bar,
|
|
'write_to_console': write_to_console,
|
|
'process_iteration': self.process_iteration,
|
|
'cached_source_callback': self.cached_source_callback,
|
|
'cached_model_source_holder': self.cached_model_source_holder,
|
|
'list_all_models': self.all_models,
|
|
'is_ensemble_master': is_ensemble,
|
|
'is_4_stem_ensemble': True if self.ensemble_main_stem_var.get() in [FOUR_STEM_ENSEMBLE, MULTI_STEM_ENSEMBLE] and is_ensemble else False}
|
|
|
|
if current_model.process_method == VR_ARCH_TYPE:
|
|
seperator = SeperateVR(current_model, process_data)
|
|
if current_model.process_method == MDX_ARCH_TYPE:
|
|
seperator = SeperateMDXC(current_model, process_data) if current_model.is_mdx_c else SeperateMDX(current_model, process_data)
|
|
if current_model.process_method == DEMUCS_ARCH_TYPE:
|
|
seperator = SeperateDemucs(current_model, process_data)
|
|
|
|
seperator.seperate()
|
|
|
|
if is_ensemble:
|
|
self.command_Text.write('\n')
|
|
|
|
if is_ensemble:
|
|
|
|
audio_file_base = audio_file_base.replace(f"_{current_model.model_basename}","")
|
|
self.command_Text.write(base_text + ENSEMBLING_OUTPUTS)
|
|
|
|
if self.ensemble_main_stem_var.get() in [FOUR_STEM_ENSEMBLE, MULTI_STEM_ENSEMBLE]:
|
|
stem_list = extract_stems(audio_file_base, export_path)
|
|
for output_stem in stem_list:
|
|
ensemble.ensemble_outputs(audio_file_base, export_path, output_stem, is_4_stem=True)
|
|
else:
|
|
if not self.is_secondary_stem_only_var.get():
|
|
ensemble.ensemble_outputs(audio_file_base, export_path, PRIMARY_STEM)
|
|
if not self.is_primary_stem_only_var.get():
|
|
ensemble.ensemble_outputs(audio_file_base, export_path, SECONDARY_STEM)
|
|
ensemble.ensemble_outputs(audio_file_base, export_path, SECONDARY_STEM, is_inst_mix=True)
|
|
|
|
self.command_Text.write(DONE)
|
|
|
|
if is_model_sample_mode:
|
|
if os.path.isfile(audio_file):
|
|
os.remove(audio_file)
|
|
|
|
clear_gpu_cache()
|
|
|
|
shutil.rmtree(export_path) if is_ensemble and len(os.listdir(export_path)) == 0 else None
|
|
|
|
if inputPath_total_len == 1 and not is_verified_audio:
|
|
self.command_Text.write(f'{error_text_console}\n{PROCESS_FAILED}')
|
|
self.command_Text.write(time_elapsed())
|
|
playsound(FAIL_CHIME) if self.is_task_complete_var.get() else None
|
|
else:
|
|
set_progress_bar(1.0)
|
|
self.command_Text.write(PROCESS_COMPLETE)
|
|
self.command_Text.write(time_elapsed())
|
|
playsound(COMPLETE_CHIME) if self.is_task_complete_var.get() else None
|
|
|
|
self.process_end()
|
|
|
|
except Exception as e:
|
|
self.error_log_var.set("{}{}".format(error_text(self.chosen_process_method_var.get(), e), self.get_settings_list()))
|
|
self.command_Text.write(f'\n\n{PROCESS_FAILED}')
|
|
self.command_Text.write(time_elapsed())
|
|
playsound(FAIL_CHIME) if self.is_task_complete_var.get() else None
|
|
self.process_end(error=e)
|
|
|
|
#--Varible Methods--
|
|
|
|
def load_to_default_confirm(self):
|
|
"""Reset settings confirmation after asking for confirmation"""
|
|
if self.thread_check(self.active_processing_thread):
|
|
self.error_dialogue(SET_TO_DEFAULT_PROCESS_ERROR)
|
|
return
|
|
|
|
confirm = messagebox.askyesno(
|
|
parent=root,
|
|
title=RESET_ALL_TO_DEFAULT_WARNING[0],
|
|
message=RESET_ALL_TO_DEFAULT_WARNING[1]
|
|
)
|
|
if not confirm:
|
|
return
|
|
|
|
self.load_saved_settings(DEFAULT_DATA, is_default_reset=True)
|
|
self.update_checkbox_text()
|
|
|
|
if self.pre_proc_model_toggle is not None and self.is_open_menu_advanced_demucs_options.get():
|
|
self.pre_proc_model_toggle()
|
|
|
|
if (self.change_state_lambda is not None and (
|
|
self.is_open_menu_advanced_vr_options.get() or
|
|
self.is_open_menu_advanced_mdx_options.get() or
|
|
self.is_open_menu_advanced_demucs_options.get()
|
|
)):
|
|
self.change_state_lambda()
|
|
|
|
def load_saved_vars(self, data):
|
|
"""Initializes primary Tkinter vars"""
|
|
|
|
for key, value in DEFAULT_DATA.items():
|
|
if not key in data.keys():
|
|
data = {**data, **{key:value}}
|
|
data['batch_size'] = DEF_OPT
|
|
|
|
## ADD_BUTTON
|
|
self.chosen_process_method_var = tk.StringVar(value=data['chosen_process_method'])
|
|
|
|
#VR Architecture Vars
|
|
self.vr_model_var = tk.StringVar(value=data['vr_model'])
|
|
self.aggression_setting_var = tk.StringVar(value=data['aggression_setting'])
|
|
self.window_size_var = tk.StringVar(value=data['window_size'])
|
|
self.mdx_segment_size_var = tk.StringVar(value=data['mdx_segment_size'])
|
|
self.batch_size_var = tk.StringVar(value=data['batch_size'])
|
|
self.crop_size_var = tk.StringVar(value=data['crop_size'])
|
|
self.is_tta_var = tk.BooleanVar(value=data['is_tta'])
|
|
self.is_output_image_var = tk.BooleanVar(value=data['is_output_image'])
|
|
self.is_post_process_var = tk.BooleanVar(value=data['is_post_process'])
|
|
self.is_high_end_process_var = tk.BooleanVar(value=data['is_high_end_process'])
|
|
self.post_process_threshold_var = tk.StringVar(value=data['post_process_threshold'])
|
|
self.vr_voc_inst_secondary_model_var = tk.StringVar(value=data['vr_voc_inst_secondary_model'])
|
|
self.vr_other_secondary_model_var = tk.StringVar(value=data['vr_other_secondary_model'])
|
|
self.vr_bass_secondary_model_var = tk.StringVar(value=data['vr_bass_secondary_model'])
|
|
self.vr_drums_secondary_model_var = tk.StringVar(value=data['vr_drums_secondary_model'])
|
|
self.vr_is_secondary_model_activate_var = tk.BooleanVar(value=data['vr_is_secondary_model_activate'])
|
|
self.vr_voc_inst_secondary_model_scale_var = tk.StringVar(value=data['vr_voc_inst_secondary_model_scale'])
|
|
self.vr_other_secondary_model_scale_var = tk.StringVar(value=data['vr_other_secondary_model_scale'])
|
|
self.vr_bass_secondary_model_scale_var = tk.StringVar(value=data['vr_bass_secondary_model_scale'])
|
|
self.vr_drums_secondary_model_scale_var = tk.StringVar(value=data['vr_drums_secondary_model_scale'])
|
|
|
|
#Demucs Vars
|
|
self.demucs_model_var = tk.StringVar(value=data['demucs_model'])
|
|
self.segment_var = tk.StringVar(value=data['segment'])
|
|
self.overlap_var = tk.StringVar(value=data['overlap'])
|
|
self.overlap_mdx_var = tk.StringVar(value=data['overlap_mdx'])
|
|
self.overlap_mdx23_var = tk.StringVar(value=data['overlap_mdx23'])
|
|
self.shifts_var = tk.StringVar(value=data['shifts'])
|
|
self.chunks_demucs_var = tk.StringVar(value=data['chunks_demucs'])
|
|
self.margin_demucs_var = tk.StringVar(value=data['margin_demucs'])
|
|
self.is_chunk_demucs_var = tk.BooleanVar(value=data['is_chunk_demucs'])
|
|
self.is_chunk_mdxnet_var = tk.BooleanVar(value=False)
|
|
self.is_primary_stem_only_Demucs_var = tk.BooleanVar(value=data['is_primary_stem_only_Demucs'])
|
|
self.is_secondary_stem_only_Demucs_var = tk.BooleanVar(value=data['is_secondary_stem_only_Demucs'])
|
|
self.is_split_mode_var = tk.BooleanVar(value=data['is_split_mode'])
|
|
self.is_demucs_combine_stems_var = tk.BooleanVar(value=data['is_demucs_combine_stems'])#is_mdx23_combine_stems
|
|
self.is_mdx23_combine_stems_var = tk.BooleanVar(value=data['is_mdx23_combine_stems'])
|
|
self.demucs_voc_inst_secondary_model_var = tk.StringVar(value=data['demucs_voc_inst_secondary_model'])
|
|
self.demucs_other_secondary_model_var = tk.StringVar(value=data['demucs_other_secondary_model'])
|
|
self.demucs_bass_secondary_model_var = tk.StringVar(value=data['demucs_bass_secondary_model'])
|
|
self.demucs_drums_secondary_model_var = tk.StringVar(value=data['demucs_drums_secondary_model'])
|
|
self.demucs_is_secondary_model_activate_var = tk.BooleanVar(value=data['demucs_is_secondary_model_activate'])
|
|
self.demucs_voc_inst_secondary_model_scale_var = tk.StringVar(value=data['demucs_voc_inst_secondary_model_scale'])
|
|
self.demucs_other_secondary_model_scale_var = tk.StringVar(value=data['demucs_other_secondary_model_scale'])
|
|
self.demucs_bass_secondary_model_scale_var = tk.StringVar(value=data['demucs_bass_secondary_model_scale'])
|
|
self.demucs_drums_secondary_model_scale_var = tk.StringVar(value=data['demucs_drums_secondary_model_scale'])
|
|
self.demucs_pre_proc_model_var = tk.StringVar(value=data['demucs_pre_proc_model'])
|
|
self.is_demucs_pre_proc_model_activate_var = tk.BooleanVar(value=data['is_demucs_pre_proc_model_activate'])
|
|
self.is_demucs_pre_proc_model_inst_mix_var = tk.BooleanVar(value=data['is_demucs_pre_proc_model_inst_mix'])
|
|
|
|
#MDX-Net Vars
|
|
self.mdx_net_model_var = tk.StringVar(value=data['mdx_net_model'])
|
|
self.chunks_var = tk.StringVar(value=data['chunks'])
|
|
self.margin_var = tk.StringVar(value=data['margin'])
|
|
self.compensate_var = tk.StringVar(value=data['compensate'])
|
|
self.denoise_option_var = tk.StringVar(value=data['denoise_option'])#
|
|
self.phase_option_var = tk.StringVar(value=data['phase_option'])#
|
|
self.phase_shifts_var = tk.StringVar(value=data['phase_shifts'])#
|
|
self.is_save_align_var = tk.BooleanVar(value=data['is_save_align'])#,
|
|
self.is_match_silence_var = tk.BooleanVar(value=data['is_match_silence'])#
|
|
self.is_spec_match_var = tk.BooleanVar(value=data['is_spec_match'])#
|
|
self.is_match_frequency_pitch_var = tk.BooleanVar(value=data['is_match_frequency_pitch'])#
|
|
self.is_mdx_c_seg_def_var = tk.BooleanVar(value=data['is_mdx_c_seg_def'])#
|
|
self.is_invert_spec_var = tk.BooleanVar(value=data['is_invert_spec'])#
|
|
self.is_deverb_vocals_var = tk.BooleanVar(value=data['is_deverb_vocals'])#
|
|
self.deverb_vocal_opt_var = tk.StringVar(value=data['deverb_vocal_opt'])#
|
|
self.voc_split_save_opt_var = tk.StringVar(value=data['voc_split_save_opt'])#
|
|
self.is_mixer_mode_var = tk.BooleanVar(value=data['is_mixer_mode'])
|
|
self.mdx_batch_size_var = tk.StringVar(value=data['mdx_batch_size'])
|
|
self.mdx_voc_inst_secondary_model_var = tk.StringVar(value=data['mdx_voc_inst_secondary_model'])
|
|
self.mdx_other_secondary_model_var = tk.StringVar(value=data['mdx_other_secondary_model'])
|
|
self.mdx_bass_secondary_model_var = tk.StringVar(value=data['mdx_bass_secondary_model'])
|
|
self.mdx_drums_secondary_model_var = tk.StringVar(value=data['mdx_drums_secondary_model'])
|
|
self.mdx_is_secondary_model_activate_var = tk.BooleanVar(value=data['mdx_is_secondary_model_activate'])
|
|
self.mdx_voc_inst_secondary_model_scale_var = tk.StringVar(value=data['mdx_voc_inst_secondary_model_scale'])
|
|
self.mdx_other_secondary_model_scale_var = tk.StringVar(value=data['mdx_other_secondary_model_scale'])
|
|
self.mdx_bass_secondary_model_scale_var = tk.StringVar(value=data['mdx_bass_secondary_model_scale'])
|
|
self.mdx_drums_secondary_model_scale_var = tk.StringVar(value=data['mdx_drums_secondary_model_scale'])
|
|
self.is_mdxnet_c_model_var = tk.BooleanVar(value=False)
|
|
|
|
#Ensemble Vars
|
|
self.is_save_all_outputs_ensemble_var = tk.BooleanVar(value=data['is_save_all_outputs_ensemble'])
|
|
self.is_append_ensemble_name_var = tk.BooleanVar(value=data['is_append_ensemble_name'])
|
|
|
|
#Audio Tool Vars
|
|
self.chosen_audio_tool_var = tk.StringVar(value=data['chosen_audio_tool'])
|
|
self.choose_algorithm_var = tk.StringVar(value=data['choose_algorithm'])
|
|
self.time_stretch_rate_var = tk.StringVar(value=data['time_stretch_rate'])
|
|
self.pitch_rate_var = tk.StringVar(value=data['pitch_rate'])
|
|
self.is_time_correction_var = tk.BooleanVar(value=data['is_time_correction'])
|
|
|
|
#Shared Vars
|
|
self.semitone_shift_var = tk.StringVar(value=data['semitone_shift'])
|
|
self.mp3_bit_set_var = tk.StringVar(value=data['mp3_bit_set'])
|
|
self.save_format_var = tk.StringVar(value=data['save_format'])
|
|
self.wav_type_set_var = tk.StringVar(value=data['wav_type_set'])#
|
|
self.device_set_var = tk.StringVar(value=data['device_set'])#
|
|
self.user_code_var = tk.StringVar(value=data['user_code'])
|
|
self.is_gpu_conversion_var = tk.BooleanVar(value=data['is_gpu_conversion'])
|
|
self.is_primary_stem_only_var = tk.BooleanVar(value=data['is_primary_stem_only'])
|
|
self.is_secondary_stem_only_var = tk.BooleanVar(value=data['is_secondary_stem_only'])
|
|
self.is_testing_audio_var = tk.BooleanVar(value=data['is_testing_audio'])#
|
|
self.is_auto_update_model_params_var = tk.BooleanVar(value=True)#
|
|
self.is_auto_update_model_params = data['is_auto_update_model_params']
|
|
self.is_add_model_name_var = tk.BooleanVar(value=data['is_add_model_name'])
|
|
self.is_accept_any_input_var = tk.BooleanVar(value=data['is_accept_any_input'])
|
|
self.is_task_complete_var = tk.BooleanVar(value=data['is_task_complete'])
|
|
self.is_normalization_var = tk.BooleanVar(value=data['is_normalization'])#
|
|
self.is_use_opencl_var = tk.BooleanVar(value=False)#True if is_opencl_only else data['is_use_opencl'])#
|
|
self.is_wav_ensemble_var = tk.BooleanVar(value=data['is_wav_ensemble'])#
|
|
self.is_create_model_folder_var = tk.BooleanVar(value=data['is_create_model_folder'])
|
|
self.help_hints_var = tk.BooleanVar(value=data['help_hints_var'])
|
|
self.model_sample_mode_var = tk.BooleanVar(value=data['model_sample_mode'])
|
|
self.model_sample_mode_duration_var = tk.StringVar(value=data['model_sample_mode_duration'])
|
|
self.model_sample_mode_duration_checkbox_var = tk.StringVar(value=SAMPLE_MODE_CHECKBOX(self.model_sample_mode_duration_var.get()))
|
|
self.model_sample_mode_duration_label_var = tk.StringVar(value=f'{self.model_sample_mode_duration_var.get()} Seconds')
|
|
self.set_vocal_splitter_var = tk.StringVar(value=data['set_vocal_splitter'])
|
|
self.is_set_vocal_splitter_var = tk.BooleanVar(value=data['is_set_vocal_splitter'])#
|
|
self.is_save_inst_set_vocal_splitter_var = tk.BooleanVar(value=data['is_save_inst_set_vocal_splitter'])#
|
|
|
|
#Path Vars
|
|
self.export_path_var = tk.StringVar(value=data['export_path'])
|
|
self.inputPaths = data['input_paths']
|
|
self.lastDir = data['lastDir']
|
|
|
|
#DualPaths-Align
|
|
self.time_window_var = tk.StringVar(value=data['time_window'])#
|
|
self.intro_analysis_var = tk.StringVar(value=data['intro_analysis'])
|
|
self.db_analysis_var = tk.StringVar(value=data['db_analysis'])
|
|
|
|
self.fileOneEntry_var = tk.StringVar(value=data['fileOneEntry'])
|
|
self.fileOneEntry_Full_var = tk.StringVar(value=data['fileOneEntry_Full'])
|
|
self.fileTwoEntry_var = tk.StringVar(value=data['fileTwoEntry'])
|
|
self.fileTwoEntry_Full_var = tk.StringVar(value=data['fileTwoEntry_Full'])
|
|
self.DualBatch_inputPaths = data['DualBatch_inputPaths']
|
|
|
|
def load_saved_settings(self, loaded_setting: dict, process_method=None, is_default_reset=False):
|
|
"""Loads user saved application settings or resets to default"""
|
|
|
|
for key, value in DEFAULT_DATA.items():
|
|
if not key in loaded_setting.keys():
|
|
loaded_setting = {**loaded_setting, **{key:value}}
|
|
loaded_setting['batch_size'] = DEF_OPT
|
|
|
|
is_default_reset = True if process_method == ENSEMBLE_MODE or is_default_reset else False
|
|
|
|
if process_method == VR_ARCH_PM or is_default_reset:
|
|
self.vr_model_var.set(loaded_setting['vr_model'])
|
|
self.aggression_setting_var.set(loaded_setting['aggression_setting'])
|
|
self.window_size_var.set(loaded_setting['window_size'])
|
|
self.mdx_segment_size_var.set(loaded_setting['mdx_segment_size'])
|
|
self.batch_size_var.set(loaded_setting['batch_size'])
|
|
self.crop_size_var.set(loaded_setting['crop_size'])
|
|
self.is_tta_var.set(loaded_setting['is_tta'])
|
|
self.is_output_image_var.set(loaded_setting['is_output_image'])
|
|
self.is_post_process_var.set(loaded_setting['is_post_process'])
|
|
self.is_high_end_process_var.set(loaded_setting['is_high_end_process'])
|
|
self.post_process_threshold_var.set(loaded_setting['post_process_threshold'])
|
|
self.vr_voc_inst_secondary_model_var.set(loaded_setting['vr_voc_inst_secondary_model'])
|
|
self.vr_other_secondary_model_var.set(loaded_setting['vr_other_secondary_model'])
|
|
self.vr_bass_secondary_model_var.set(loaded_setting['vr_bass_secondary_model'])
|
|
self.vr_drums_secondary_model_var.set(loaded_setting['vr_drums_secondary_model'])
|
|
self.vr_is_secondary_model_activate_var.set(loaded_setting['vr_is_secondary_model_activate'])
|
|
self.vr_voc_inst_secondary_model_scale_var.set(loaded_setting['vr_voc_inst_secondary_model_scale'])
|
|
self.vr_other_secondary_model_scale_var.set(loaded_setting['vr_other_secondary_model_scale'])
|
|
self.vr_bass_secondary_model_scale_var.set(loaded_setting['vr_bass_secondary_model_scale'])
|
|
self.vr_drums_secondary_model_scale_var.set(loaded_setting['vr_drums_secondary_model_scale'])
|
|
|
|
if process_method == DEMUCS_ARCH_TYPE or is_default_reset:
|
|
self.demucs_model_var.set(loaded_setting['demucs_model'])
|
|
self.segment_var.set(loaded_setting['segment'])
|
|
self.overlap_var.set(loaded_setting['overlap'])
|
|
self.shifts_var.set(loaded_setting['shifts'])
|
|
self.chunks_demucs_var.set(loaded_setting['chunks_demucs'])
|
|
self.margin_demucs_var.set(loaded_setting['margin_demucs'])
|
|
self.is_chunk_demucs_var.set(loaded_setting['is_chunk_demucs'])
|
|
self.is_chunk_mdxnet_var.set(loaded_setting['is_chunk_mdxnet'])
|
|
self.is_primary_stem_only_Demucs_var.set(loaded_setting['is_primary_stem_only_Demucs'])
|
|
self.is_secondary_stem_only_Demucs_var.set(loaded_setting['is_secondary_stem_only_Demucs'])
|
|
self.is_split_mode_var.set(loaded_setting['is_split_mode'])
|
|
self.is_demucs_combine_stems_var.set(loaded_setting['is_demucs_combine_stems'])#
|
|
self.is_mdx23_combine_stems_var.set(loaded_setting['is_mdx23_combine_stems'])#
|
|
self.demucs_voc_inst_secondary_model_var.set(loaded_setting['demucs_voc_inst_secondary_model'])
|
|
self.demucs_other_secondary_model_var.set(loaded_setting['demucs_other_secondary_model'])
|
|
self.demucs_bass_secondary_model_var.set(loaded_setting['demucs_bass_secondary_model'])
|
|
self.demucs_drums_secondary_model_var.set(loaded_setting['demucs_drums_secondary_model'])
|
|
self.demucs_is_secondary_model_activate_var.set(loaded_setting['demucs_is_secondary_model_activate'])
|
|
self.demucs_voc_inst_secondary_model_scale_var.set(loaded_setting['demucs_voc_inst_secondary_model_scale'])
|
|
self.demucs_other_secondary_model_scale_var.set(loaded_setting['demucs_other_secondary_model_scale'])
|
|
self.demucs_bass_secondary_model_scale_var.set(loaded_setting['demucs_bass_secondary_model_scale'])
|
|
self.demucs_drums_secondary_model_scale_var.set(loaded_setting['demucs_drums_secondary_model_scale'])
|
|
self.demucs_stems_var.set(loaded_setting['demucs_stems'])
|
|
self.mdxnet_stems_var.set(loaded_setting['mdx_stems'])
|
|
self.update_stem_checkbox_labels(self.demucs_stems_var.get(), demucs=True)
|
|
self.demucs_pre_proc_model_var.set(loaded_setting['demucs_pre_proc_model'])
|
|
self.is_demucs_pre_proc_model_activate_var.set(loaded_setting['is_demucs_pre_proc_model_activate'])
|
|
self.is_demucs_pre_proc_model_inst_mix_var.set(loaded_setting['is_demucs_pre_proc_model_inst_mix'])
|
|
|
|
if process_method == MDX_ARCH_TYPE or is_default_reset:
|
|
self.mdx_net_model_var.set(loaded_setting['mdx_net_model'])
|
|
self.chunks_var.set(loaded_setting['chunks'])
|
|
self.margin_var.set(loaded_setting['margin'])
|
|
self.compensate_var.set(loaded_setting['compensate'])
|
|
self.denoise_option_var.set(loaded_setting['denoise_option'])
|
|
self.is_match_frequency_pitch_var.set(loaded_setting['is_match_frequency_pitch'])#
|
|
self.overlap_mdx_var.set(loaded_setting['overlap_mdx'])
|
|
self.overlap_mdx23_var.set(loaded_setting['overlap_mdx23'])
|
|
self.is_mdx_c_seg_def_var.set(loaded_setting['is_mdx_c_seg_def'])#
|
|
self.is_invert_spec_var.set(loaded_setting['is_invert_spec'])#
|
|
self.is_mixer_mode_var.set(loaded_setting['is_mixer_mode'])
|
|
self.mdx_batch_size_var.set(loaded_setting['mdx_batch_size'])
|
|
self.mdx_voc_inst_secondary_model_var.set(loaded_setting['mdx_voc_inst_secondary_model'])
|
|
self.mdx_other_secondary_model_var.set(loaded_setting['mdx_other_secondary_model'])
|
|
self.mdx_bass_secondary_model_var.set(loaded_setting['mdx_bass_secondary_model'])
|
|
self.mdx_drums_secondary_model_var.set(loaded_setting['mdx_drums_secondary_model'])
|
|
self.mdx_is_secondary_model_activate_var.set(loaded_setting['mdx_is_secondary_model_activate'])
|
|
self.mdx_voc_inst_secondary_model_scale_var.set(loaded_setting['mdx_voc_inst_secondary_model_scale'])
|
|
self.mdx_other_secondary_model_scale_var.set(loaded_setting['mdx_other_secondary_model_scale'])
|
|
self.mdx_bass_secondary_model_scale_var.set(loaded_setting['mdx_bass_secondary_model_scale'])
|
|
self.mdx_drums_secondary_model_scale_var.set(loaded_setting['mdx_drums_secondary_model_scale'])
|
|
|
|
if is_default_reset:
|
|
self.is_save_all_outputs_ensemble_var.set(loaded_setting['is_save_all_outputs_ensemble'])
|
|
self.is_append_ensemble_name_var.set(loaded_setting['is_append_ensemble_name'])
|
|
self.choose_algorithm_var.set(loaded_setting['choose_algorithm'])
|
|
self.time_stretch_rate_var.set(loaded_setting['time_stretch_rate'])
|
|
self.pitch_rate_var.set(loaded_setting['pitch_rate'])#
|
|
self.is_time_correction_var.set(loaded_setting['is_time_correction'])#
|
|
self.is_primary_stem_only_var.set(loaded_setting['is_primary_stem_only'])
|
|
self.is_secondary_stem_only_var.set(loaded_setting['is_secondary_stem_only'])
|
|
self.is_testing_audio_var.set(loaded_setting['is_testing_audio'])#
|
|
self.is_auto_update_model_params_var.set(loaded_setting['is_auto_update_model_params'])
|
|
self.is_add_model_name_var.set(loaded_setting['is_add_model_name'])
|
|
self.is_accept_any_input_var.set(loaded_setting["is_accept_any_input"])
|
|
self.is_task_complete_var.set(loaded_setting['is_task_complete'])
|
|
self.is_create_model_folder_var.set(loaded_setting['is_create_model_folder'])
|
|
self.mp3_bit_set_var.set(loaded_setting['mp3_bit_set'])
|
|
self.semitone_shift_var.set(loaded_setting['semitone_shift'])#
|
|
self.save_format_var.set(loaded_setting['save_format'])
|
|
self.wav_type_set_var.set(loaded_setting['wav_type_set'])#
|
|
self.device_set_var.set(loaded_setting['device_set'])#
|
|
self.user_code_var.set(loaded_setting['user_code'])
|
|
self.phase_option_var.set(loaded_setting['phase_option'])#
|
|
self.phase_shifts_var.set(loaded_setting['phase_shifts'])#
|
|
self.is_save_align_var.set(loaded_setting['is_save_align'])#i
|
|
self.time_window_var.set(loaded_setting['time_window'])#
|
|
self.is_match_silence_var.set(loaded_setting['is_match_silence'])#
|
|
self.is_spec_match_var.set(loaded_setting['is_spec_match'])#
|
|
self.intro_analysis_var.set(loaded_setting['intro_analysis'])#
|
|
self.db_analysis_var.set(loaded_setting['db_analysis'])#
|
|
self.fileOneEntry_var.set(loaded_setting['fileOneEntry'])#
|
|
self.fileOneEntry_Full_var.set(loaded_setting['fileOneEntry_Full'])#
|
|
self.fileTwoEntry_var.set(loaded_setting['fileTwoEntry'])#
|
|
self.fileTwoEntry_Full_var.set(loaded_setting['fileTwoEntry_Full'])#
|
|
self.DualBatch_inputPaths = []
|
|
|
|
self.is_gpu_conversion_var.set(loaded_setting['is_gpu_conversion'])
|
|
self.is_normalization_var.set(loaded_setting['is_normalization'])#
|
|
self.is_use_opencl_var.set(False)#True if is_opencl_only else loaded_setting['is_use_opencl'])#
|
|
self.is_wav_ensemble_var.set(loaded_setting['is_wav_ensemble'])#
|
|
self.help_hints_var.set(loaded_setting['help_hints_var'])
|
|
self.is_wav_ensemble_var.set(loaded_setting['is_wav_ensemble'])
|
|
self.set_vocal_splitter_var.set(loaded_setting['set_vocal_splitter'])
|
|
self.is_set_vocal_splitter_var.set(loaded_setting['is_set_vocal_splitter'])#
|
|
self.is_save_inst_set_vocal_splitter_var.set(loaded_setting['is_save_inst_set_vocal_splitter'])#
|
|
self.deverb_vocal_opt_var.set(loaded_setting['deverb_vocal_opt'])#
|
|
self.voc_split_save_opt_var.set(loaded_setting['voc_split_save_opt'])#
|
|
self.is_deverb_vocals_var.set(loaded_setting['is_deverb_vocals'])#
|
|
|
|
self.model_sample_mode_var.set(loaded_setting['model_sample_mode'])
|
|
self.model_sample_mode_duration_var.set(loaded_setting['model_sample_mode_duration'])
|
|
self.model_sample_mode_duration_checkbox_var.set(SAMPLE_MODE_CHECKBOX(self.model_sample_mode_duration_var.get()))
|
|
self.model_sample_mode_duration_label_var.set(f'{self.model_sample_mode_duration_var.get()} Seconds')
|
|
|
|
def save_values(self, app_close=True, is_restart=False, is_auto_save=False):
|
|
"""Saves application data"""
|
|
|
|
# -Save Data-
|
|
main_settings={
|
|
'vr_model': self.vr_model_var.get(),
|
|
'aggression_setting': self.aggression_setting_var.get(),
|
|
'window_size': self.window_size_var.get(),
|
|
'mdx_segment_size': self.mdx_segment_size_var.get(),
|
|
'batch_size': self.batch_size_var.get(),
|
|
'crop_size': self.crop_size_var.get(),
|
|
'is_tta': self.is_tta_var.get(),
|
|
'is_output_image': self.is_output_image_var.get(),
|
|
'is_post_process': self.is_post_process_var.get(),
|
|
'is_high_end_process': self.is_high_end_process_var.get(),
|
|
'post_process_threshold': self.post_process_threshold_var.get(),
|
|
'vr_voc_inst_secondary_model': self.vr_voc_inst_secondary_model_var.get(),
|
|
'vr_other_secondary_model': self.vr_other_secondary_model_var.get(),
|
|
'vr_bass_secondary_model': self.vr_bass_secondary_model_var.get(),
|
|
'vr_drums_secondary_model': self.vr_drums_secondary_model_var.get(),
|
|
'vr_is_secondary_model_activate': self.vr_is_secondary_model_activate_var.get(),
|
|
'vr_voc_inst_secondary_model_scale': self.vr_voc_inst_secondary_model_scale_var.get(),
|
|
'vr_other_secondary_model_scale': self.vr_other_secondary_model_scale_var.get(),
|
|
'vr_bass_secondary_model_scale': self.vr_bass_secondary_model_scale_var.get(),
|
|
'vr_drums_secondary_model_scale': self.vr_drums_secondary_model_scale_var.get(),
|
|
'demucs_model': self.demucs_model_var.get(),
|
|
'segment': self.segment_var.get(),
|
|
'overlap': self.overlap_var.get(),
|
|
'overlap_mdx': self.overlap_mdx_var.get(),
|
|
'overlap_mdx23': self.overlap_mdx23_var.get(),
|
|
'shifts': self.shifts_var.get(),
|
|
'chunks_demucs': self.chunks_demucs_var.get(),
|
|
'margin_demucs': self.margin_demucs_var.get(),
|
|
'is_chunk_demucs': self.is_chunk_demucs_var.get(),
|
|
'is_chunk_mdxnet': self.is_chunk_mdxnet_var.get(),
|
|
'is_primary_stem_only_Demucs': self.is_primary_stem_only_Demucs_var.get(),
|
|
'is_secondary_stem_only_Demucs': self.is_secondary_stem_only_Demucs_var.get(),
|
|
'is_split_mode': self.is_split_mode_var.get(),
|
|
'is_demucs_combine_stems': self.is_demucs_combine_stems_var.get(),#
|
|
'is_mdx23_combine_stems': self.is_mdx23_combine_stems_var.get(),#
|
|
'demucs_voc_inst_secondary_model': self.demucs_voc_inst_secondary_model_var.get(),
|
|
'demucs_other_secondary_model': self.demucs_other_secondary_model_var.get(),
|
|
'demucs_bass_secondary_model': self.demucs_bass_secondary_model_var.get(),
|
|
'demucs_drums_secondary_model': self.demucs_drums_secondary_model_var.get(),
|
|
'demucs_is_secondary_model_activate': self.demucs_is_secondary_model_activate_var.get(),
|
|
'demucs_voc_inst_secondary_model_scale': self.demucs_voc_inst_secondary_model_scale_var.get(),
|
|
'demucs_other_secondary_model_scale': self.demucs_other_secondary_model_scale_var.get(),
|
|
'demucs_bass_secondary_model_scale': self.demucs_bass_secondary_model_scale_var.get(),
|
|
'demucs_drums_secondary_model_scale': self.demucs_drums_secondary_model_scale_var.get(),
|
|
'demucs_pre_proc_model': self.demucs_pre_proc_model_var.get(),
|
|
'is_demucs_pre_proc_model_activate': self.is_demucs_pre_proc_model_activate_var.get(),
|
|
'is_demucs_pre_proc_model_inst_mix': self.is_demucs_pre_proc_model_inst_mix_var.get(),
|
|
'mdx_net_model': self.mdx_net_model_var.get(),
|
|
'chunks': self.chunks_var.get(),
|
|
'margin': self.margin_var.get(),
|
|
'compensate': self.compensate_var.get(),
|
|
'denoise_option': self.denoise_option_var.get(),#
|
|
'is_match_frequency_pitch': self.is_match_frequency_pitch_var.get(),#
|
|
'phase_option': self.phase_option_var.get(),#
|
|
'phase_shifts': self.phase_shifts_var.get(),#
|
|
'is_save_align': self.is_save_align_var.get(),#
|
|
'is_match_silence': self.is_match_silence_var.get(),#
|
|
'is_spec_match': self.is_spec_match_var.get(),#
|
|
'is_mdx_c_seg_def': self.is_mdx_c_seg_def_var.get(),#
|
|
'is_invert_spec': self.is_invert_spec_var.get(),#
|
|
'is_deverb_vocals': self.is_deverb_vocals_var.get(),##,
|
|
'deverb_vocal_opt': self.deverb_vocal_opt_var.get(),#
|
|
'voc_split_save_opt': self.voc_split_save_opt_var.get(),##,
|
|
'is_mixer_mode': self.is_mixer_mode_var.get(),
|
|
'mdx_batch_size':self.mdx_batch_size_var.get(),
|
|
'mdx_voc_inst_secondary_model': self.mdx_voc_inst_secondary_model_var.get(),
|
|
'mdx_other_secondary_model': self.mdx_other_secondary_model_var.get(),
|
|
'mdx_bass_secondary_model': self.mdx_bass_secondary_model_var.get(),
|
|
'mdx_drums_secondary_model': self.mdx_drums_secondary_model_var.get(),
|
|
'mdx_is_secondary_model_activate': self.mdx_is_secondary_model_activate_var.get(),
|
|
'mdx_voc_inst_secondary_model_scale': self.mdx_voc_inst_secondary_model_scale_var.get(),
|
|
'mdx_other_secondary_model_scale': self.mdx_other_secondary_model_scale_var.get(),
|
|
'mdx_bass_secondary_model_scale': self.mdx_bass_secondary_model_scale_var.get(),
|
|
'mdx_drums_secondary_model_scale': self.mdx_drums_secondary_model_scale_var.get(),
|
|
'is_save_all_outputs_ensemble': self.is_save_all_outputs_ensemble_var.get(),
|
|
'is_append_ensemble_name': self.is_append_ensemble_name_var.get(),
|
|
'chosen_audio_tool': self.chosen_audio_tool_var.get(),
|
|
'choose_algorithm': self.choose_algorithm_var.get(),
|
|
'time_stretch_rate': self.time_stretch_rate_var.get(),
|
|
'pitch_rate': self.pitch_rate_var.get(),#
|
|
'is_time_correction': self.is_time_correction_var.get(),#
|
|
'is_gpu_conversion': self.is_gpu_conversion_var.get(),
|
|
'is_primary_stem_only': self.is_primary_stem_only_var.get(),
|
|
'is_secondary_stem_only': self.is_secondary_stem_only_var.get(),
|
|
'is_testing_audio': self.is_testing_audio_var.get(),#
|
|
'is_auto_update_model_params': self.is_auto_update_model_params_var.get(),
|
|
'is_add_model_name': self.is_add_model_name_var.get(),
|
|
'is_accept_any_input': self.is_accept_any_input_var.get(),
|
|
'is_task_complete': self.is_task_complete_var.get(),
|
|
'is_normalization': self.is_normalization_var.get(),#
|
|
'is_use_opencl': self.is_use_opencl_var.get(),#
|
|
'is_wav_ensemble': self.is_wav_ensemble_var.get(),#
|
|
'is_create_model_folder': self.is_create_model_folder_var.get(),
|
|
'mp3_bit_set': self.mp3_bit_set_var.get(),
|
|
'semitone_shift': self.semitone_shift_var.get(),#
|
|
'save_format': self.save_format_var.get(),
|
|
'wav_type_set': self.wav_type_set_var.get(),#
|
|
'device_set': self.device_set_var.get(),#
|
|
'user_code': self.user_code_var.get(),
|
|
'help_hints_var': self.help_hints_var.get(),
|
|
'set_vocal_splitter': self.set_vocal_splitter_var.get(),
|
|
'is_set_vocal_splitter': self.is_set_vocal_splitter_var.get(),#
|
|
'is_save_inst_set_vocal_splitter': self.is_save_inst_set_vocal_splitter_var.get(),#
|
|
'model_sample_mode': self.model_sample_mode_var.get(),
|
|
'model_sample_mode_duration': self.model_sample_mode_duration_var.get()
|
|
}
|
|
|
|
other_data = {
|
|
'chosen_process_method': self.chosen_process_method_var.get(),
|
|
'input_paths': self.inputPaths,
|
|
'lastDir': self.lastDir,
|
|
'export_path': self.export_path_var.get(),
|
|
'time_window': self.time_window_var.get(),
|
|
'intro_analysis': self.intro_analysis_var.get(),
|
|
'db_analysis': self.db_analysis_var.get(),
|
|
'fileOneEntry': self.fileOneEntry_var.get(),
|
|
'fileOneEntry_Full': self.fileOneEntry_Full_var.get(),
|
|
'fileTwoEntry': self.fileTwoEntry_var.get(),
|
|
'fileTwoEntry_Full': self.fileTwoEntry_Full_var.get(),
|
|
'DualBatch_inputPaths': self.DualBatch_inputPaths,
|
|
#'model_hash_table': model_hash_table,
|
|
}
|
|
|
|
user_saved_extras = {
|
|
'demucs_stems': self.demucs_stems_var.get(),
|
|
'mdx_stems': self.mdxnet_stems_var.get()}
|
|
|
|
if app_close:
|
|
save_data(data={**main_settings, **other_data})
|
|
|
|
if self.thread_check(self.active_download_thread):
|
|
self.error_dialoge(EXIT_DOWNLOAD_ERROR)
|
|
return
|
|
|
|
if self.thread_check(self.active_processing_thread):
|
|
if self.is_process_stopped:
|
|
self.error_dialoge(EXIT_HALTED_PROCESS_ERROR)
|
|
else:
|
|
self.error_dialoge(EXIT_PROCESS_ERROR)
|
|
return
|
|
|
|
remove_temps(ENSEMBLE_TEMP_PATH)
|
|
remove_temps(SAMPLE_CLIP_PATH)
|
|
self.delete_temps()
|
|
|
|
if is_restart:
|
|
try:
|
|
subprocess.Popen(f'UVR_Launcher.exe')
|
|
except Exception:
|
|
subprocess.Popen(f'python "{__file__}"', shell=True)
|
|
|
|
self.destroy()
|
|
|
|
elif is_auto_save:
|
|
save_data(data={**main_settings, **other_data})
|
|
else:
|
|
return {**main_settings, **user_saved_extras}
|
|
|
|
def get_settings_list(self):
|
|
|
|
settings_dict = self.save_values(app_close=False)
|
|
settings_list = '\n'.join(''.join(f"{key}: {value}") for key, value in settings_dict.items() if not key == 'user_code')
|
|
|
|
return f"\n{FULL_APP_SET_TEXT}:\n\n{settings_list}"
|
|
|
|
def read_bulliten_text_mac(path, data):
|
|
try:
|
|
with open(path, 'w') as f:
|
|
f.write(data)
|
|
|
|
if os.path.isfile(path):
|
|
with open(path, 'r') as file :
|
|
data = file.read().replace("~", "•")
|
|
except Exception as e:
|
|
data = 'No information available.'
|
|
|
|
return data
|
|
|
|
def open_link(event, link=None):
|
|
webbrowser.open(link)
|
|
|
|
def auto_hyperlink(text_widget:tk.Text):
|
|
content = text_widget.get('1.0', tk.END)
|
|
|
|
# Regular expression to identify URLs
|
|
urls = re.findall(r'http[s]?://(?:[a-zA-Z]|[0-9]|[$-_@.&+]|[!*\\(\\),]|(?:%[0-9a-fA-F][0-9a-fA-F]))+', content)
|
|
|
|
for url in urls:
|
|
start_idx = content.find(url)
|
|
end_idx = start_idx + len(url)
|
|
|
|
# Convert indices to tk.Text widget format
|
|
start_line = content.count('\n', 0, start_idx) + 1
|
|
start_char = start_idx - content.rfind('\n', 0, start_idx) - 1
|
|
end_line = content.count('\n', 0, end_idx) + 1
|
|
end_char = end_idx - content.rfind('\n', 0, end_idx) - 1
|
|
|
|
start_tag = f"{start_line}.{start_char}"
|
|
end_tag = f"{end_line}.{end_char}"
|
|
|
|
# Tag the hyperlink text and configure it
|
|
text_widget.tag_add(url, start_tag, end_tag)
|
|
text_widget.tag_configure(url, foreground=FG_COLOR, underline=True)
|
|
text_widget.tag_bind(url, "<Button-1>", lambda e, link=url: open_link(e, link))
|
|
text_widget.tag_bind(url, "<Enter>", lambda e: text_widget.config(cursor="hand2"))
|
|
text_widget.tag_bind(url, "<Leave>", lambda e: text_widget.config(cursor="arrow"))
|
|
|
|
def vip_downloads(password, link_type=VIP_REPO):
|
|
"""Attempts to decrypt VIP model link with given input code"""
|
|
|
|
try:
|
|
kdf = PBKDF2HMAC(
|
|
algorithm=hashes.SHA256(),
|
|
length=32,
|
|
salt=link_type[0],
|
|
iterations=390000,)
|
|
|
|
key = base64.urlsafe_b64encode(kdf.derive(bytes(password, 'utf-8')))
|
|
f = Fernet(key)
|
|
|
|
return str(f.decrypt(link_type[1]), 'UTF-8')
|
|
except Exception:
|
|
return NO_CODE
|
|
|
|
def extract_stems(audio_file_base, export_path):
|
|
|
|
filenames = [file for file in os.listdir(export_path) if file.startswith(audio_file_base)]
|
|
|
|
pattern = r'\(([^()]+)\)(?=[^()]*\.wav)'
|
|
stem_list = []
|
|
|
|
for filename in filenames:
|
|
match = re.search(pattern, filename)
|
|
if match:
|
|
stem_list.append(match.group(1))
|
|
|
|
counter = Counter(stem_list)
|
|
filtered_lst = [item for item in stem_list if counter[item] > 1]
|
|
|
|
return list(set(filtered_lst))
|
|
|
|
if __name__ == "__main__":
|
|
|
|
try:
|
|
windll.user32.SetThreadDpiAwarenessContext(wintypes.HANDLE(-1))
|
|
except Exception as e:
|
|
if OPERATING_SYSTEM == 'Windows':
|
|
print(e)
|
|
|
|
root = MainWindow()
|
|
root.update_checkbox_text()
|
|
root.is_root_defined_var.set(True)
|
|
root.is_check_splash = True
|
|
|
|
root.update() if is_windows else root.update_idletasks()
|
|
root.deiconify()
|
|
root.configure(bg=BG_COLOR)
|
|
root.mainloop()
|