import gradio as gr
from utils import download_file, EN_US, API_QR, TMP_DIR

ZH2EN = {
    "二维码输出尺寸": "Image size",
    "输入文本": "Input text",
    "输出二维码": "QR code",
    "输入文字在线生成二维码": "Enter text to generate a QR code.",
    "状态栏": "Status",
}


def _L(zh_txt: str):
    return ZH2EN[zh_txt] if EN_US else zh_txt


def infer(img_size: int, input_txt: str):
    status = "Success"
    img = None
    try:
        if (not input_txt) or input_txt == "0":
            raise ValueError("Please input valid text!")

        img = download_file(
            f"{API_QR}/?size={img_size}x{img_size}&data={input_txt}",
            f"{TMP_DIR}/qrcode.jpg",
        )

    except Exception as e:
        status = f"{e}"

    return status, img


def qrcode():
    return gr.Interface(
        fn=infer,
        inputs=[
            gr.Slider(35, 1000, 217, label=_L("二维码输出尺寸")),
            gr.Textbox(label=_L("输入文本"), placeholder=_L("输入文字在线生成二维码")),
        ],
        outputs=[
            gr.Textbox(label=_L("状态栏"), show_copy_button=True),
            gr.Image(label=_L("输出二维码"), show_share_button=False),
        ],
        flagging_mode="never",
    )