kohya-sd-scripts-webui/scripts/tabs/preparation/merge_captions.py

import gradio as gr

from scripts import presets, ui
from scripts.runner import initialize_runner
from scripts.utilities import load_args_template, options_to_gradio


def title():
    return "Merge captions"


def create_ui():
    options = {}
    templates, script_file = load_args_template(
        "finetune", "merge_captions_to_metadata.py"
    )

    with gr.Column():
        inti_runner = initialize_runner(script_file, templates, options)
        with gr.Box():
            with gr.Row():
                init_ui = presets.create_ui(
                    "finetune.merge_captions_to_metadata", templates, options
                )
        with gr.Box():
            ui.title("Options")
            with gr.Column():
                options_to_gradio(templates, options)

    inti_runner()
    init_ui()