| import gradio as gr |
| import os |
| import argparse |
| from dreambooth_gui import dreambooth_tab |
| from finetune_gui import finetune_tab |
| from textual_inversion_gui import ti_tab |
| from library.utilities import utilities_tab |
| from lora_gui import lora_tab |
| from library.class_lora_tab import LoRATools |
|
|
| import os |
| from library.custom_logging import setup_logging |
| from library.localization_ext import add_javascript |
|
|
| |
| log = setup_logging() |
|
|
|
|
| def UI(**kwargs): |
| add_javascript(kwargs.get("language")) |
| css = "" |
|
|
| headless = kwargs.get("headless", False) |
| log.info(f"headless: {headless}") |
|
|
| if os.path.exists("./style.css"): |
| with open(os.path.join("./style.css"), "r", encoding="utf8") as file: |
| log.info("Load CSS...") |
| css += file.read() + "\n" |
|
|
| if os.path.exists("./.release"): |
| with open(os.path.join("./.release"), "r", encoding="utf8") as file: |
| release = file.read() |
|
|
| if os.path.exists("./README.md"): |
| with open(os.path.join("./README.md"), "r", encoding="utf8") as file: |
| README = file.read() |
|
|
| interface = gr.Blocks( |
| css=css, title=f"Kohya_ss GUI {release}", theme=gr.themes.Default() |
| ) |
|
|
| with interface: |
| with gr.Tab("Dreambooth"): |
| ( |
| train_data_dir_input, |
| reg_data_dir_input, |
| output_dir_input, |
| logging_dir_input, |
| ) = dreambooth_tab(headless=headless) |
| with gr.Tab("LoRA"): |
| lora_tab(headless=headless) |
| with gr.Tab("Textual Inversion"): |
| ti_tab(headless=headless) |
| with gr.Tab("Finetuning"): |
| finetune_tab(headless=headless) |
| with gr.Tab("Utilities"): |
| utilities_tab( |
| train_data_dir_input=train_data_dir_input, |
| reg_data_dir_input=reg_data_dir_input, |
| output_dir_input=output_dir_input, |
| logging_dir_input=logging_dir_input, |
| enable_copy_info_button=True, |
| headless=headless, |
| ) |
| with gr.Tab("LoRA"): |
| _ = LoRATools(headless=headless) |
| with gr.Tab("About"): |
| gr.Markdown(f"kohya_ss GUI release {release}") |
| with gr.Tab("README"): |
| gr.Markdown(README) |
|
|
| htmlStr = f""" |
| <html> |
| <body> |
| <div class="ver-class">{release}</div> |
| </body> |
| </html> |
| """ |
| gr.HTML(htmlStr) |
| |
| launch_kwargs = {} |
| username = kwargs.get("username") |
| password = kwargs.get("password") |
| server_port = kwargs.get("server_port", 0) |
| inbrowser = kwargs.get("inbrowser", False) |
| share = False |
| server_name = kwargs.get("listen") |
|
|
| launch_kwargs["server_name"] = server_name |
| if username and password: |
| launch_kwargs["auth"] = (username, password) |
| if server_port > 0: |
| launch_kwargs["server_port"] = server_port |
| if inbrowser: |
| launch_kwargs["inbrowser"] = inbrowser |
| if share: |
| launch_kwargs["share"] = False |
| interface.launch(**launch_kwargs, share=False) |
|
|
|
|
| if __name__ == "__main__": |
| |
| parser = argparse.ArgumentParser() |
| parser.add_argument( |
| "--listen", |
| type=str, |
| default="127.0.0.1", |
| help="IP to listen on for connections to Gradio", |
| ) |
| parser.add_argument( |
| "--username", type=str, default="", help="Username for authentication" |
| ) |
| parser.add_argument( |
| "--password", type=str, default="", help="Password for authentication" |
| ) |
| parser.add_argument( |
| "--server_port", |
| type=int, |
| default=0, |
| help="Port to run the server listener on", |
| ) |
| parser.add_argument("--inbrowser", action="store_true", help="Open in browser") |
| parser.add_argument("--share", action="store_true", help="Share the gradio UI") |
| parser.add_argument( |
| "--headless", action="store_true", help="Is the server headless" |
| ) |
| parser.add_argument( |
| "--language", type=str, default=None, help="Set custom language" |
| ) |
|
|
| parser.add_argument("--use-ipex", action="store_true", help="Use IPEX environment") |
|
|
| args = parser.parse_args() |
|
|
| UI( |
| username=args.username, |
| password=args.password, |
| inbrowser=args.inbrowser, |
| server_port=args.server_port, |
| share=False, |
| listen=args.listen, |
| headless=args.headless, |
| language=args.language, |
| ) |
|
|
