kohya_gui/class_metadata.py

import gradio as gr

from .class_gui_config import KohyaSSGUIConfig


class MetaData:
    def __init__(
        self,
        config: KohyaSSGUIConfig = {},
    ) -> None:
        self.config = config

        with gr.Row():
            self.metadata_title = gr.Textbox(
                label="Metadata title",
                placeholder="(optional) title for model metadata (default is output_name)",
                interactive=True,
                value=self.config.get("metadata.title", ""),
            )
            self.metadata_author = gr.Textbox(
                label="Metadata author",
                placeholder="(optional) author name for model metadata",
                interactive=True,
                value=self.config.get("metadata.author", ""),
            )
        self.metadata_description = gr.Textbox(
            label="Metadata description",
            placeholder="(optional) description for model metadata",
            interactive=True,
            value=self.config.get("metadata.description", ""),
        )
        with gr.Row():
            self.metadata_license = gr.Textbox(
                label="Metadata license",
                placeholder="(optional) license for model metadata",
                interactive=True,
                value=self.config.get("metadata.license", ""),
            )
            self.metadata_tags = gr.Textbox(
                label="Metadata tags",
                placeholder="(optional) tags for model metadata, separated by comma",
                interactive=True,
                value=self.config.get("metadata.tags", ""),
            )

    def run_cmd(run_cmd: list, **kwargs):
        if "metadata_title" in kwargs and kwargs.get("metadata_title") != "":
            run_cmd.append("--metadata_title")
            run_cmd.append(kwargs["metadata_title"])

        if "metadata_author" in kwargs and kwargs.get("metadata_author") != "":
            run_cmd.append("--metadata_author")
            run_cmd.append(kwargs["metadata_author"])

        if "metadata_description" in kwargs and kwargs.get("metadata_description") != "":
            run_cmd.append("--metadata_description")
            run_cmd.append(kwargs["metadata_description"])

        if "metadata_license" in kwargs and kwargs.get("metadata_license") != "":
            run_cmd.append("--metadata_license")
            run_cmd.append(kwargs["metadata_license"])

        if "metadata_tags" in kwargs and kwargs.get("metadata_tags") != "":
            run_cmd.append("--metadata_tags")
            run_cmd.append(kwargs["metadata_tags"])

        return run_cmd