from scripts.api import baidu_netdisk_api, send_img_path
from modules import script_callbacks, generation_parameters_copypaste as send
from scripts.bin import (
    bin_file_name,
    get_matched_summary,
    check_bin_exists,
    download_bin_file,
)
from scripts.tool import cwd

"""
api函数声明和启动分离方便另外一边被外部调用
"""

not_exists_msg = (
    f"找不到{bin_file_name},尝试手动从 {get_matched_summary()[1]} 下载,下载后放到 {cwd} 文件夹下,重启界面"
)


def on_ui_tabs():
    import gradio as gr

    exists = check_bin_exists()
    if not exists:
        try:
            print("缺少必要的二进制文件，开始下载")
            download_bin_file()
            print("done")
        except Exception as e:
            print("下载二进制文件时出错：", str(e))
        exists = check_bin_exists()
        if not exists:
            print(f"\033[31m{not_exists_msg}\033[0m")
    with gr.Blocks(analytics_enabled=False) as baidu_netdisk:
        img = gr.Image(
            type="pil",
            elem_id="bd_hidden_img",
        )

        img_update_trigger = gr.Button("button", elem_id="bd_hidden_img_update_trigger")

        def img_update_func():
            return send_img_path.get("value")

        img_update_trigger.click(img_update_func, outputs=img)
        img_file_info = gr.Textbox(elem_id="bd_hidden_img_file_info")
        for tab in ["txt2img", "img2img", "inpaint", "extras"]:
            btn = gr.Button(f"Send to {tab}", elem_id=f"bd_hidden_tab_{tab}")
            send.register_paste_params_button(
                send.ParamBinding(
                    paste_button=btn,
                    tabname=tab,
                    source_image_component=img,
                    source_text_component=img_file_info,
                )
            )

        gr.Textbox(not_exists_msg, visible=not exists)
        with gr.Row(visible=bool(exists)):
            with gr.Column():
                gr.HTML(
                    "如果你看到这个那说明此项那说明出现了问题", elem_id="baidu_netdisk_container_wrapper"
                )

        return ((baidu_netdisk, "百度云", "baiduyun"),)


script_callbacks.on_ui_tabs(on_ui_tabs)
script_callbacks.on_app_started(baidu_netdisk_api)