Cool_Tools/ultimatevocalremovergui
1
0
Fork 0
mirror of https://github.com/Anjok07/ultimatevocalremovergui.git synced 2024-11-14 19:07:37 +01:00
Code Issues Packages Projects Releases Wiki Activity

Revert Drag & Drop feature

Browse Source
This commit is contained in:
DilanBoskan 2020-11-10 10:27:28 +01:00
parent 259f7a3c3e
commit 3127c1f06b
1 changed files with 55 additions and 121 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

174
VocalRemover.py
View File

@ -4,7 +4,6 @@ import tkinter.ttk as ttk
import tkinter.messagebox import tkinter.messagebox
import tkinter.filedialog import tkinter.filedialog
import tkinter.font import tkinter.font
from tkinterdnd2 import TkinterDnD, DND_FILES # Enable Drag & Drop
from datetime import datetime from datetime import datetime
# Images # Images
from PIL import Image from PIL import Image
@ -94,7 +93,6 @@ def open_image(path: str, size: tuple = None, keep_aspect: bool = True, rotate:
def save_data(data): def save_data(data):
""" """
Saves given data as a .pkl (pickle) file Saves given data as a .pkl (pickle) file
Paramters: Paramters:
data(dict): data(dict):
Dictionary containing all the necessary data to save Dictionary containing all the necessary data to save
@ -107,7 +105,6 @@ def save_data(data):
def load_data() -> dict: def load_data() -> dict:
""" """
Loads saved pkl file and returns the stored data Loads saved pkl file and returns the stored data
Returns(dict): Returns(dict):
Dictionary containing all the saved data Dictionary containing all the saved data
""" """
@ -169,30 +166,6 @@ def get_model_values(model_name):
return model_values return model_values
def drop(var, event, accept_mode: str = 'files'):
"""
Drag & Drop verification process
"""
path = event.data
if accept_mode == 'folder':
path = path.replace('{', '').replace('}', '')
if not os.path.isdir(path):
tk.messagebox.showerror(title='Invalid Folder',
message='Your given export path is not a valid folder!')
return
elif accept_mode == 'files':
# Clean path text and set path to the list of paths
path = path[:-1]
path = path.replace('{', '')
path = path.split('} ')
else:
# Invalid accept mode
return
var.set(path)
class ThreadSafeConsole(tk.Text): class ThreadSafeConsole(tk.Text):
""" """
Text Widget which is thread safe for tkinter Text Widget which is thread safe for tkinter
@ -226,7 +199,7 @@ class ThreadSafeConsole(tk.Text):
self.after(100, self.update_me) self.after(100, self.update_me)
class MainWindow(TkinterDnD.Tk): class MainWindow(tk.Tk):
# --Constants-- # --Constants--
# Layout # Layout
IMAGE_HEIGHT = 140 IMAGE_HEIGHT = 140
@ -258,7 +231,6 @@ class MainWindow(TkinterDnD.Tk):
xpad=int(self.winfo_screenwidth()/2 - 550/2), xpad=int(self.winfo_screenwidth()/2 - 550/2),
ypad=int(self.winfo_screenheight()/2 - height/2 - 30))) ypad=int(self.winfo_screenheight()/2 - height/2 - 30)))
self.configure(bg='#000000') # Set background color to black self.configure(bg='#000000') # Set background color to black
self.protocol("WM_DELETE_WINDOW", self.save_values)
self.resizable(False, False) self.resizable(False, False)
self.update() self.update()
@ -275,7 +247,7 @@ class MainWindow(TkinterDnD.Tk):
data = load_data() data = load_data()
# Paths # Paths
self.exportPath_var = tk.StringVar(value=data['export_path']) self.exportPath_var = tk.StringVar(value=data['export_path'])
self.inputPaths_var = tk.StringVar(value='') self.inputPaths = []
# Processing Options # Processing Options
self.gpuConversion_var = tk.BooleanVar(value=data['gpu']) self.gpuConversion_var = tk.BooleanVar(value=data['gpu'])
self.postprocessing_var = tk.BooleanVar(value=data['postprocess']) self.postprocessing_var = tk.BooleanVar(value=data['postprocess'])
@ -299,7 +271,6 @@ class MainWindow(TkinterDnD.Tk):
self.aiModel_var = tk.StringVar(value=data['aiModel']) self.aiModel_var = tk.StringVar(value=data['aiModel'])
self.last_aiModel = self.aiModel_var.get() self.last_aiModel = self.aiModel_var.get()
# Other # Other
self.inputPathsEntry_var = tk.StringVar(value='')
self.lastDir = data['lastDir'] # nopep8 self.lastDir = data['lastDir'] # nopep8
self.progress_var = tk.IntVar(value=0) self.progress_var = tk.IntVar(value=0)
# Font # Font
@ -307,18 +278,12 @@ class MainWindow(TkinterDnD.Tk):
# --Widgets-- # --Widgets--
self.create_widgets() self.create_widgets()
self.configure_widgets() self.configure_widgets()
self.bind_widgets()
self.place_widgets() self.place_widgets()
self.update_available_models() self.update_available_models()
self.update_states() self.update_states()
self.update_loop() self.update_loop()
# Display the multiple selected music files more visually understandable
self.inputPaths_var.trace_add('write',
lambda *args: self.inputPathsEntry_var.set('; '.join(list(self.inputPaths_var.get()))))
# -Widget Methods- # -Widget Methods-
def create_widgets(self): def create_widgets(self):
"""Create window widgets""" """Create window widgets"""
@ -354,21 +319,6 @@ class MainWindow(TkinterDnD.Tk):
font=self.font, foreground='white') font=self.font, foreground='white')
ttk.Style().configure('T', font=self.font, foreground='white') ttk.Style().configure('T', font=self.font, foreground='white')
def bind_widgets(self):
"""Bind widgets to the drag & drop mechanic"""
self.filePaths_saveTo_Button.drop_target_register(DND_FILES)
self.filePaths_saveTo_Entry.drop_target_register(DND_FILES)
self.filePaths_musicFile_Button.drop_target_register(DND_FILES)
self.filePaths_musicFile_Entry.drop_target_register(DND_FILES)
self.filePaths_saveTo_Button.dnd_bind('<<Drop>>',
lambda e, var=self.exportPath_var: drop(var, e, accept_mode='folder'))
self.filePaths_saveTo_Entry.dnd_bind('<<Drop>>',
lambda e, var=self.exportPath_var: drop(var, e, accept_mode='folder'))
self.filePaths_musicFile_Button.dnd_bind('<<Drop>>',
lambda e, var=self.inputPaths_var: drop(var, e, accept_mode='files'))
self.filePaths_musicFile_Entry.dnd_bind('<<Drop>>',
lambda e, var=self.inputPaths_var: drop(var, e, accept_mode='files'))
def place_widgets(self): def place_widgets(self):
"""Place main widgets""" """Place main widgets"""
self.title_Label.place(x=-2, y=-2) self.title_Label.place(x=-2, y=-2)
@ -402,7 +352,7 @@ class MainWindow(TkinterDnD.Tk):
text='Select Your Audio File(s)', text='Select Your Audio File(s)',
command=self.open_file_filedialog) command=self.open_file_filedialog)
self.filePaths_musicFile_Entry = ttk.Entry(master=self.filePaths_Frame, self.filePaths_musicFile_Entry = ttk.Entry(master=self.filePaths_Frame,
textvariable=self.inputPathsEntry_var, text=self.inputPaths,
state=tk.DISABLED state=tk.DISABLED
) )
# -Place Widgets- # -Place Widgets-
@ -605,7 +555,13 @@ class MainWindow(TkinterDnD.Tk):
initialdir=self.lastDir, initialdir=self.lastDir,
) )
if paths: # Path selected if paths: # Path selected
self.inputPaths_var.set(paths) self.inputPaths = paths
# Change the entry text
self.filePaths_musicFile_Entry.configure(state=tk.NORMAL)
self.filePaths_musicFile_Entry.delete(0, tk.END)
self.filePaths_musicFile_Entry.insert(0, self.inputPaths)
self.filePaths_musicFile_Entry.configure(state=tk.DISABLED)
self.lastDir = os.path.dirname(paths[0]) self.lastDir = os.path.dirname(paths[0])
def open_export_filedialog(self): def open_export_filedialog(self):
@ -626,7 +582,6 @@ class MainWindow(TkinterDnD.Tk):
""" """
# -Get all variables- # -Get all variables-
export_path = self.exportPath_var.get() export_path = self.exportPath_var.get()
input_paths = list(self.inputPaths_var.get())
instrumentalModel_path = self.instrumentalLabel_to_path[self.instrumentalModel_var.get()] # nopep8 instrumentalModel_path = self.instrumentalLabel_to_path[self.instrumentalModel_var.get()] # nopep8
stackedModel_path = self.stackedLabel_to_path[self.stackedModel_var.get()] # nopep8 stackedModel_path = self.stackedLabel_to_path[self.stackedModel_var.get()] # nopep8
# Get constants # Get constants
@ -656,13 +611,12 @@ class MainWindow(TkinterDnD.Tk):
return return
# -Check for invalid inputs- # -Check for invalid inputs-
for path in input_paths: if not any([(os.path.isfile(path) and path.endswith(('.mp3', '.mp4', '.m4a', '.flac', '.wav')))
if not os.path.isfile(path): for path in self.inputPaths]):
tk.messagebox.showwarning(master=self, tk.messagebox.showwarning(master=self,
title='Invalid Music File', title='Invalid Music File',
message='You have selected an invalid music file! Please make sure that the file still exists!', message='You have selected an invalid music file!\nPlease make sure that your files still exist and ends with either ".mp3", ".mp4", ".m4a", ".flac", ".wav"')
detail=f'File path: {path}') return
return
if not os.path.isdir(export_path): if not os.path.isdir(export_path):
tk.messagebox.showwarning(master=self, tk.messagebox.showwarning(master=self,
title='Invalid Export Directory', title='Invalid Export Directory',
@ -682,6 +636,27 @@ class MainWindow(TkinterDnD.Tk):
message='You have selected an invalid stacked model file!\nPlease make sure that your model file still exists!') message='You have selected an invalid stacked model file!\nPlease make sure that your model file still exists!')
return return
# -Save Data-
save_data(data={
'export_path': export_path,
'gpu': self.gpuConversion_var.get(),
'postprocess': self.postprocessing_var.get(),
'tta': self.tta_var.get(),
'output_image': self.outputImage_var.get(),
'stack': self.stack_var.get(),
'stackOnly': self.stackOnly_var.get(),
'stackPasses': stackPasses,
'saveAllStacked': self.saveAllStacked_var.get(),
'sr': sr,
'hop_length': hop_length,
'window_size': window_size,
'n_fft': n_fft,
'useModel': 'instrumental', # Always instrumental
'lastDir': self.lastDir,
'modelFolder': self.modelFolder_var.get(),
'aiModel': self.aiModel_var.get(),
})
if self.aiModel_var.get() == 'v2': if self.aiModel_var.get() == 'v2':
inference = inference_v2 inference = inference_v2
elif self.aiModel_var.get() == 'v4': elif self.aiModel_var.get() == 'v4':
@ -693,7 +668,7 @@ class MainWindow(TkinterDnD.Tk):
threading.Thread(target=inference.main, threading.Thread(target=inference.main,
kwargs={ kwargs={
# Paths # Paths
'input_paths': input_paths, 'input_paths': self.inputPaths,
'export_path': export_path, 'export_path': export_path,
# Processing Options # Processing Options
'gpu': 0 if self.gpuConversion_var.get() else -1, 'gpu': 0 if self.gpuConversion_var.get() else -1,
@ -749,19 +724,20 @@ class MainWindow(TkinterDnD.Tk):
# Loop through each constant (key) and its widgets # Loop through each constant (key) and its widgets
for key, (widget, var) in widgetsVars.items(): for key, (widget, var) in widgetsVars.items():
if stacked_selectable: if stacked_selectable:
if instrumental_selectable: # Stacked model can be selected
if (key in instrumental.keys() and if key in stacked.keys():
key in stacked.keys()): if (key in stacked.keys() and
# Both models have set constants not instrumental_selectable):
widget.configure(state=tk.DISABLED)
var.set('%d/%d' % (instrumental[key], stacked[key]))
continue
else:
if key in stacked.keys():
# Only stacked selectable # Only stacked selectable
widget.configure(state=tk.DISABLED) widget.configure(state=tk.DISABLED)
var.set(stacked[key]) var.set(stacked[key])
continue continue
elif (key in instrumental.keys() and
instrumental_selectable):
# Both models have set constants
widget.configure(state=tk.DISABLED)
var.set('%d/%d' % (instrumental[key], stacked[key]))
continue
else: else:
# Stacked model can not be selected # Stacked model can not be selected
if (key in instrumental.keys() and if (key in instrumental.keys() and
@ -859,6 +835,7 @@ class MainWindow(TkinterDnD.Tk):
# Instrumental Model # Instrumental Model
self.options_instrumentalModel_Label.configure(foreground='#000') self.options_instrumentalModel_Label.configure(foreground='#000')
self.options_instrumentalModel_Optionmenu.configure(state=tk.NORMAL) # nopep8 self.options_instrumentalModel_Optionmenu.configure(state=tk.NORMAL) # nopep8
self.instrumentalModel_var.set('')
# Stack Model # Stack Model
if stackLoops > 0: if stackLoops > 0:
@ -906,54 +883,11 @@ class MainWindow(TkinterDnD.Tk):
""" """
Restart the application after asking for confirmation Restart the application after asking for confirmation
""" """
save = tk.messagebox.askyesno(title='Confirmation', proceed = tk.messagebox.askyesno(title='Confirmation',
message='The application will restart. Do you want to save the data?') message='The application will restart and lose unsaved data. Do you wish to proceed?')
if save: if proceed:
self.save_values() subprocess.Popen(f'python "{__file__}"', shell=True)
subprocess.Popen(f'python "{__file__}"', shell=True) exit()
exit()
def save_values(self):
"""
Save the data of the application
"""
export_path = self.exportPath_var.get()
# Get constants
instrumental = get_model_values(self.instrumentalModel_var.get())
stacked = get_model_values(self.stackedModel_var.get())
if [bool(instrumental), bool(stacked)].count(True) == 2:
sr = DEFAULT_DATA['sr']
hop_length = DEFAULT_DATA['hop_length']
window_size = DEFAULT_DATA['window_size']
n_fft = DEFAULT_DATA['n_fft']
else:
sr = self.srValue_var.get()
hop_length = self.hopValue_var.get()
window_size = self.winSize_var.get()
n_fft = self.nfft_var.get()
# -Save Data-
save_data(data={
'export_path': export_path,
'gpu': self.gpuConversion_var.get(),
'postprocess': self.postprocessing_var.get(),
'tta': self.tta_var.get(),
'output_image': self.outputImage_var.get(),
'stack': self.stack_var.get(),
'stackOnly': self.stackOnly_var.get(),
'stackPasses': self.stackLoops_var.get(),
'saveAllStacked': self.saveAllStacked_var.get(),
'sr': sr,
'hop_length': hop_length,
'window_size': window_size,
'n_fft': n_fft,
'useModel': 'instrumental',
'lastDir': self.lastDir,
'modelFolder': self.modelFolder_var.get(),
'aiModel': self.aiModel_var.get(),
})
self.destroy()
if __name__ == "__main__": if __name__ == "__main__":