From c38c018197b0f7c33640734c3ca33fe194899169 Mon Sep 17 00:00:00 2001 From: bmaltais Date: Sat, 25 Mar 2023 20:29:04 -0400 Subject: [PATCH] Prevent the file selection GUI from running --- README.md | 3 + library/common_gui.py | 226 +++++++++++++++++------------------------- macos_setup.sh | 2 +- 3 files changed, 96 insertions(+), 135 deletions(-) diff --git a/README.md b/README.md index 30beda2..9002afa 100644 --- a/README.md +++ b/README.md @@ -213,6 +213,9 @@ This will store your a backup file with your current locally installed pip packa ## Change History +* 2023/03/26 (v21.3.5) + - Fix for https://github.com/bmaltais/kohya_ss/issues/230 + - Added detection for Google Colab to not bring up the GUI file/folder window on the platform. Instead it will only use the file/folder path provided in the input field. * 2023/03/25 (v21.3.4) - Added untested support for MacOS base on this gist: https://gist.github.com/jstayco/9f5733f05b9dc29de95c4056a023d645 diff --git a/library/common_gui.py b/library/common_gui.py index 8ff6a27..442ca3a 100644 --- a/library/common_gui.py +++ b/library/common_gui.py @@ -118,54 +118,58 @@ def get_dir_and_file(file_path): def get_file_path( file_path='', default_extension='.json', extension_name='Config files' ): - current_file_path = file_path - # print(f'current file path: {current_file_path}') + if not any(var in os.environ for var in ['COLAB_GPU', 'RUNPOD_ENVIRONMENT']): + current_file_path = file_path + # print(f'current file path: {current_file_path}') - initial_dir, initial_file = get_dir_and_file(file_path) + initial_dir, initial_file = get_dir_and_file(file_path) - # Create a hidden Tkinter root window - root = Tk() - root.wm_attributes('-topmost', 1) - root.withdraw() + # Create a hidden Tkinter root window + root = Tk() + root.wm_attributes('-topmost', 1) + root.withdraw() - # Show the open file dialog and get the selected file path - file_path = filedialog.askopenfilename( - filetypes=( - (extension_name, f'*{default_extension}'), - ('All files', '*.*'), - ), - defaultextension=default_extension, - initialfile=initial_file, - initialdir=initial_dir, - ) + # Show the open file dialog and get the selected file path + file_path = filedialog.askopenfilename( + filetypes=( + (extension_name, f'*{default_extension}'), + ('All files', '*.*'), + ), + defaultextension=default_extension, + initialfile=initial_file, + initialdir=initial_dir, + ) - # Destroy the hidden root window - root.destroy() + # Destroy the hidden root window + root.destroy() - # If no file is selected, use the current file path - if not file_path: - file_path = current_file_path + # If no file is selected, use the current file path + if not file_path: + file_path = current_file_path + current_file_path = file_path + # print(f'current file path: {current_file_path}') return file_path def get_any_file_path(file_path=''): - current_file_path = file_path - # print(f'current file path: {current_file_path}') + if not any(var in os.environ for var in ['COLAB_GPU', 'RUNPOD_ENVIRONMENT']): + current_file_path = file_path + # print(f'current file path: {current_file_path}') - initial_dir, initial_file = get_dir_and_file(file_path) + initial_dir, initial_file = get_dir_and_file(file_path) - root = Tk() - root.wm_attributes('-topmost', 1) - root.withdraw() - file_path = filedialog.askopenfilename( - initialdir=initial_dir, - initialfile=initial_file, - ) - root.destroy() + root = Tk() + root.wm_attributes('-topmost', 1) + root.withdraw() + file_path = filedialog.askopenfilename( + initialdir=initial_dir, + initialfile=initial_file, + ) + root.destroy() - if file_path == '': - file_path = current_file_path + if file_path == '': + file_path = current_file_path return file_path @@ -191,18 +195,19 @@ def remove_doublequote(file_path): def get_folder_path(folder_path=''): - current_folder_path = folder_path + if not any(var in os.environ for var in ['COLAB_GPU', 'RUNPOD_ENVIRONMENT']): + current_folder_path = folder_path - initial_dir, initial_file = get_dir_and_file(folder_path) + initial_dir, initial_file = get_dir_and_file(folder_path) - root = Tk() - root.wm_attributes('-topmost', 1) - root.withdraw() - folder_path = filedialog.askdirectory(initialdir=initial_dir) - root.destroy() + root = Tk() + root.wm_attributes('-topmost', 1) + root.withdraw() + folder_path = filedialog.askdirectory(initialdir=initial_dir) + root.destroy() - if folder_path == '': - folder_path = current_folder_path + if folder_path == '': + folder_path = current_folder_path return folder_path @@ -210,34 +215,35 @@ def get_folder_path(folder_path=''): def get_saveasfile_path( file_path='', defaultextension='.json', extension_name='Config files' ): - current_file_path = file_path - # print(f'current file path: {current_file_path}') + if not any(var in os.environ for var in ['COLAB_GPU', 'RUNPOD_ENVIRONMENT']): + current_file_path = file_path + # print(f'current file path: {current_file_path}') - initial_dir, initial_file = get_dir_and_file(file_path) + initial_dir, initial_file = get_dir_and_file(file_path) - root = Tk() - root.wm_attributes('-topmost', 1) - root.withdraw() - save_file_path = filedialog.asksaveasfile( - filetypes=( - (f'{extension_name}', f'{defaultextension}'), - ('All files', '*'), - ), - defaultextension=defaultextension, - initialdir=initial_dir, - initialfile=initial_file, - ) - root.destroy() + root = Tk() + root.wm_attributes('-topmost', 1) + root.withdraw() + save_file_path = filedialog.asksaveasfile( + filetypes=( + (f'{extension_name}', f'{defaultextension}'), + ('All files', '*'), + ), + defaultextension=defaultextension, + initialdir=initial_dir, + initialfile=initial_file, + ) + root.destroy() - # print(save_file_path) + # print(save_file_path) - if save_file_path == None: - file_path = current_file_path - else: - print(save_file_path.name) - file_path = save_file_path.name + if save_file_path == None: + file_path = current_file_path + else: + print(save_file_path.name) + file_path = save_file_path.name - # print(file_path) + # print(file_path) return file_path @@ -245,27 +251,28 @@ def get_saveasfile_path( def get_saveasfilename_path( file_path='', extensions='*', extension_name='Config files' ): - current_file_path = file_path - # print(f'current file path: {current_file_path}') + if not any(var in os.environ for var in ['COLAB_GPU', 'RUNPOD_ENVIRONMENT']): + current_file_path = file_path + # print(f'current file path: {current_file_path}') - initial_dir, initial_file = get_dir_and_file(file_path) + initial_dir, initial_file = get_dir_and_file(file_path) - root = Tk() - root.wm_attributes('-topmost', 1) - root.withdraw() - save_file_path = filedialog.asksaveasfilename( - filetypes=((f'{extension_name}', f'{extensions}'), ('All files', '*')), - defaultextension=extensions, - initialdir=initial_dir, - initialfile=initial_file, - ) - root.destroy() + root = Tk() + root.wm_attributes('-topmost', 1) + root.withdraw() + save_file_path = filedialog.asksaveasfilename( + filetypes=((f'{extension_name}', f'{extensions}'), ('All files', '*')), + defaultextension=extensions, + initialdir=initial_dir, + initialfile=initial_file, + ) + root.destroy() - if save_file_path == '': - file_path = current_file_path - else: - # print(save_file_path) - file_path = save_file_path + if save_file_path == '': + file_path = current_file_path + else: + # print(save_file_path) + file_path = save_file_path return file_path @@ -316,33 +323,6 @@ def add_pre_postfix( ) -# def add_pre_postfix( -# folder='', prefix='', postfix='', caption_file_ext='.caption' -# ): -# if not has_ext_files(folder, caption_file_ext): -# msgbox( -# f'No files with extension {caption_file_ext} were found in {folder}...' -# ) -# return - -# if prefix == '' and postfix == '': -# return - -# files = [f for f in os.listdir(folder) if f.endswith(caption_file_ext)] -# if not prefix == '': -# prefix = f'{prefix} ' -# if not postfix == '': -# postfix = f' {postfix}' - -# for file in files: -# with open(os.path.join(folder, file), 'r+') as f: -# content = f.read() -# content = content.rstrip() -# f.seek(0, 0) -# f.write(f'{prefix} {content} {postfix}') -# f.close() - - def has_ext_files(folder_path: str, file_extension: str) -> bool: """ Check if there are any files with the specified extension in the given folder. @@ -402,28 +382,6 @@ def find_replace( f.write(content) -# def find_replace(folder='', caption_file_ext='.caption', find='', replace=''): -# print('Running caption find/replace') -# if not has_ext_files(folder, caption_file_ext): -# msgbox( -# f'No files with extension {caption_file_ext} were found in {folder}...' -# ) -# return - -# if find == '': -# return - -# files = [f for f in os.listdir(folder) if f.endswith(caption_file_ext)] -# for file in files: -# with open(os.path.join(folder, file), 'r', errors='ignore') as f: -# content = f.read() -# f.close -# content = content.replace(find, replace) -# with open(os.path.join(folder, file), 'w') as f: -# f.write(content) -# f.close() - - def color_aug_changed(color_aug): if color_aug: msgbox( diff --git a/macos_setup.sh b/macos_setup.sh index d429c65..4de8417 100755 --- a/macos_setup.sh +++ b/macos_setup.sh @@ -30,7 +30,7 @@ if command -v python3.10 >/dev/null; then pip install torch==2.0.0 torchvision==0.15.1 -f https://download.pytorch.org/whl/cpu/torch_stable.html python -m pip install --use-pep517 --upgrade -r requirements_macos.txt accelerate config - echo -e "Setup finished! Run ./gui.sh to start." + echo -e "Setup finished! Run ./gui_macos.sh to start." else echo "Python not found. Please ensure you install Python." echo "The brew command for Python 3.10 is: brew install python@3.10"