kohya-sd-scripts-webui/scripts/tabs/tools/detect_face_rotate.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 "Detect face rotate"


def create_ui():
    options = {}

    templates, script_file = load_args_template("tools", "detect_face_rotate.py")

    with gr.Column():
        init_runner = initialize_runner(script_file, templates, options)
        with gr.Box():
            with gr.Row():
                init_ui = presets.create_ui(
                    "tools.detect_face_rotate", templates, options
                )
        with gr.Box():
            ui.title("Options")
            with gr.Column():
                options_to_gradio(templates, options)

    init_runner()
    init_ui()