import modules.scripts as scripts
import gradio as gr
import io
import os

from modules import script_callbacks
from scripts import constants
from PIL import Image
import segno
from segno import helpers

def generate(data, micro, error, scale, boost_error, border, dark, light):
    qrcode = segno.make(data, micro=micro, error=error, boost_error=boost_error)
    out = io.BytesIO()
    qrcode.save(out, scale=scale, kind='png', border=border, dark=dark, light=light)
    return Image.open(out)

def generate_wifi(ssid, password, security, hidden, micro, error, scale, boost_error, border, dark, light):
    if security == "None":
        password = security = None

    data = helpers.make_wifi_data(ssid=ssid, password=password, security=security, hidden=hidden)
    return generate(data, micro, error, scale, boost_error, border, dark, light)

def generate_geo(latitude, longitude, micro, error, scale, boost_error, border, dark, light):
    data = helpers.make_geo_data(latitude, longitude)
    return generate(data, micro, error, scale, boost_error, border, dark, light)

def generate_vcard(name, displayname, nickname, street, city, region, zipcode, country, birthday, email, phone, fax,
                   micro, error, scale, boost_error, border, dark, light):
    data = helpers.make_vcard_data(name=name, displayname=displayname, nickname=nickname, street=street, city=city, region=region, zipcode=zipcode, country=country, birthday=birthday, email=email, phone=phone, fax=fax)
    return generate(data, micro, error, scale, boost_error, border, dark, light)

def generate_email(address, subject, body, micro, error, scale, boost_error, border, dark, light):
    data = helpers.make_make_email_data(to=address, subject=subject, body=body)
    return generate(data, micro, error, scale, boost_error, border, dark, light)

def on_ui_tabs():
    with gr.Blocks() as ui_component:
        with gr.Row():
            with gr.Column():
                with gr.Tab("Text"):
                    text = gr.Textbox(label="Text", lines=3)
                    button_generate_text = gr.Button("Generate", variant="primary")
                with gr.Tab("WiFi"):
                    ssid = gr.Text(label="SSID")
                    hidden = gr.Checkbox(False, label="Hidden SSID")
                    password = gr.Text(label="Password")
                    security = gr.Radio(value="None", label="Security", choices=["None", "WEP", "WPA"])
                    button_generate_wifi = gr.Button("Generate", variant="primary")
                with gr.Tab("vCard"):
                    name = gr.Text(label="Name")
                    displayname = gr.Text(label="Display Name")
                    nickname = gr.Text(label="Nickname")
                    address = gr.Text(label="Address")
                    with gr.Row():
                        city = gr.Text(label="City")
                        state = gr.Text(label="State")
                    with gr.Row():
                        zipcode = gr.Text(label="ZIP Code")
                        country = gr.Dropdown(label="Country", choices=constants.countries)
                    birthday = gr.Text(label="Birthday")
                    email = gr.Text(label="Email")
                    phone = gr.Text(label="Phone")
                    fax = gr.Text(label="Fax")
                    button_generate_vcard = gr.Button("Generate", variant="primary")
                with gr.Tab("Email"):
                    recipient = gr.Text(label="Address")
                    subject = gr.Text(label="Subject")
                    body = gr.Textbox(label="Message", lines=3)
                    button_generate_email = gr.Button("Generate", variant="primary")
                with gr.Tab("Coordinates"):
                    with gr.Row():
                        latitude = gr.Number(0, label="Latitude")
                        longitude = gr.Number(0, label="Longitude")
                    button_generate_geo = gr.Button("Generate", variant="primary")
                with gr.Accordion("Settings", open=False):
                    scale = gr.Slider(label="Scale", minimum=1, maximum=50, value=10, step=1)
                    border = gr.Slider(label="Border", minimum=0, maximum=10, value=4, step=1)
                    with gr.Row():
                        dark_color = gr.ColorPicker("#000000", label="Dark Color")
                        light_color = gr.ColorPicker("#ffffff", label="Light Color")
                    error_correction = gr.Dropdown(value="L", label="Error Correction Level", choices=["L", "M", "Q", "H"])
                    with gr.Row():
                        error_boost = gr.Checkbox(True, label="Boost Error Correction Level")
                        micro_code = gr.Checkbox(False, label="Micro QR Code")

            with gr.Column():
                output = gr.Image(interactive=False, show_label=False, elem_id="qrcode_output").style(height=480)

        common_inputs = [micro_code, error_correction, scale, error_boost, border, dark_color, light_color]

        button_generate_text.click(generate, [text] + common_inputs, output, show_progress=False)
        button_generate_wifi.click(generate_wifi, [ssid, password, security, hidden] + common_inputs, output, show_progress=False)
        button_generate_geo.click(generate_geo, [latitude, longitude] + common_inputs, output, show_progress=False)
        button_generate_vcard.click(generate_vcard, [name, displayname, nickname, address, city, state, zipcode, country, birthday, email, phone, fax] + common_inputs, output, show_progress=False)
        button_generate_email.click(generate_email, [recipient, subject, body] + common_inputs, output, show_progress=False)

        return [(ui_component, "QR Code", "qrcode_tab")]

script_callbacks.on_ui_tabs(on_ui_tabs)