import gradio as gr

import beta_fn
import fns
import generate
import login_discord
import preset_fn
import rekord
import pre

js_func = """
function refresh() {
    const url = new URL(window.location);

    if (url.searchParams.get('__theme') !== 'dark') {
        url.searchParams.set('__theme', 'dark');
        window.location.href = url.href;
    }
}
"""

animation = """
function createGradioAnimation() {
    var container = document.createElement('div');
    container.id = 'gradio-animation';
    container.style.fontSize = '2em';
    container.style.fontWeight = 'bold';
    container.style.textAlign = 'center';
    container.style.marginTop = '0px';
    container.style.marginBottom = '5px';

    var text = 'Objektify';
    for (var i = 0; i < text.length; i++) {
        (function(i){
            setTimeout(function(){
                var letter = document.createElement('span');
                letter.style.opacity = '0';
                letter.style.transition = 'opacity 0.5s';
                letter.innerText = text[i];

                container.appendChild(letter);

                setTimeout(function() {
                    letter.style.opacity = '1';
                }, 50);
            }, i * 250);
        })(i);
    }

    var gradioContainer = document.querySelector('.gradio-container');
    gradioContainer.insertBefore(container, gradioContainer.firstChild);

    setTimeout(function() {
        container.style.transition = 'all 2s';
        container.style.fontSize = '1em';
        container.style.letterSpacing = '0.1em';
    }, 3000);

    return 'Animation created';
}

"""


ga_script = """
<script async src="https://www.googletagmanager.com/gtag/js?id=G-J2J6MBSCQB"></script>
"""
ga_load = """
function() {
  window.dataLayer = window.dataLayer || [];
  function gtag(){dataLayer.push(arguments);}
  gtag('js', new Date());

  gtag('config', 'G-J2J6MBSCQB');
}
"""

confirm_horizontal = "(x) => confirm('Apply horizontal mode?')"
def rst(a):
    if a == None:
        return False
    return gr.Checkbox()

def tests():
    print('tests')
def chk_size(img, dummy):
    x, y = img.size
    if x <= y:
        return dummy
    else:
        return not dummy

theme = gr.themes.Soft(
    secondary_hue="violet",
    text_size="sm",
    spacing_size="sm",
    radius_size="lg",
).set(
    block_background_fill='*secondary_50'
)


login_uri = pre.login_uri


with gr.Blocks(theme=theme, js=animation, head=ga_script) as demo: # , js=js_func
    with gr.Column():
        with gr.Row():
            gr.Markdown(
                """
                ### [Join objektify official discord!](https://discord.com/invite/KAuTCUGXpH)
                """
            )
            login = gr.Button(size="sm", value="Login to Discord",
                              link=f"https://discord.com/oauth2/authorize?client_id=1239473790732472370&response_type=code&redirect_uri={login_uri}&scope=identify",
                              scale=0)

        with gr.Row():
            gr.Markdown(
                """
                ## Input image and Select Options
                """
            )
            preset = gr.Dropdown(scale=0, label='Preset', interactive=True, allow_custom_value=False, type='value', visible=True, info='Login to create preset')


    with gr.Tab("Front"):
        with gr.Row():
            with gr.Column():       #1
                img_type = gr.Radio(["Image", "Gif"], value="Image", scale=1, label="type")
                img_static = gr.Image(label="Image", interactive=True, sources=['upload', 'clipboard'], type="pil", visible=True, scale=10)
                img_dynamic_file = gr.File(file_types=[".gif"], visible=False, scale=1, interactive=True, label="Gif File", type="filepath")
                img_dynamic_show = gr.Image(label="Gif", interactive=False, sources=['upload', 'clipboard'], type="pil", visible=False, scale=10)


                horizontal = gr.Checkbox(visible=False, label='horizontal')
                dummy = gr.Checkbox(visible=False, label='dummy')

                img_type.change(fn=fns.change_img_type, inputs=[img_type], outputs=[img_static, img_dynamic_file, img_dynamic_show])


                img_static.upload(fn=fns.img2cache_resize, inputs=[img_static, dummy], outputs=[img_static, dummy])

                img_static.change(fn=rst, inputs=img_static, outputs=[horizontal])
                dummy.change(fn=None, js=confirm_horizontal, outputs=horizontal)
                horizontal.change(fn=fns.rotate, inputs=img_static, outputs=img_static)

                img_dynamic_file.upload(fn=fns.gif2cache, inputs=img_dynamic_file, outputs=img_dynamic_show)
                img_dynamic_file.clear(fn=fns.gif_clear, outputs=img_dynamic_show)


            with gr.Column():       #2
                with gr.Accordion(label='Colors'):        #objekt color
                    obj_color = gr.Radio(["Atom01", "Binary01", "Cream01", "Divine01", "Special", "OMA1", "OMA2", "Other color", "AI Color"], value="Divine01", label='Objekt Color')

                    with gr.Row():
                        ai_num = gr.Slider(1, 1000, 500, label="AI Serial", visible=False, step=1)
                        change_ai = gr.Button(value="Change AI", visible=False)
                    obj_color_picker = gr.ColorPicker(label="Other Objekt Color", visible=False, interactive=True, value="#ffffff")


                    obj_color.change(fn=fns.obj_color_update, inputs=obj_color, outputs=[obj_color_picker, ai_num, change_ai])



                    txt_color = gr.Radio(["White", "Black", "Other"], label="Text Color", value="Black")
                    txt_color_picker = gr.ColorPicker(label="Other Text Color", value="#ffffff", visible=False, interactive=True)
                    txt_color.change(fns.txt_color_update, txt_color, txt_color_picker)


                with gr.Accordion('Names'):
                    group_name_radio = gr.Radio(choices=pre.groups_Other, label="Group Name")
                    group_name_txt = gr.Textbox(label="Other Group name", placeholder="CLASSY", visible=False, interactive=True)


                    member_name_dropdown = gr.Dropdown(choices=pre.names, label="Member Name", allow_custom_value=True)



                with gr.Accordion("Add Objekt Number", open=False):
                    with gr.Row():
                        obj_num = gr.Textbox(label="Objekt Number", placeholder="100", interactive=True, visible=True, scale=1)
                        obj_alphabet = gr.Radio(["Z", "A", "Other"], label="Z or A", interactive=True, visible=True, scale=1, value='Z')
                        obj_alphabet_txt = gr.Textbox(label="Alphabet", placeholder="S", interactive=True, visible=False, scale=1)

                        obj_alphabet.change(fns.alphabet_txt_update, obj_alphabet, obj_alphabet_txt)


                        with gr.Row():
                            obj_serial = gr.Textbox(label="Objekt Serial", placeholder="00001", interactive=True, visible=True, scale=1)
                            obj_serial_random = gr.Button(value="Random Serial", interactive=True, visible=True, scale=1)


                            obj_serial_random.click(fn=fns.serial_random, outputs=obj_serial)


                with gr.Group():
                    generate_btn = gr.Button(value="Generate")

            with gr.Column():
                edited_img = gr.Image(label="Edited Image", scale=10)


    with gr.Tab("Back") as back_tab:
        with gr.Row():
            preview = gr.Image(label="Image", interactive=False)

            with gr.Column():
                with gr.Accordion('Contents'):
                    obj_outline_color = gr.Radio(["White", "Black", "Other"], label="Frame color", value="White")
                    obj_outline_color_picker = gr.ColorPicker(label="Other Outline Color", value="#ffffff", visible=False, interactive=True)

                    obj_outline_color.change(fns.frame_color_update, obj_outline_color, obj_outline_color_picker)

                    logo_radio = gr.Radio(choices=pre.groups, label="Add Logo", interactive=True, value="None")


                with gr.Accordion('Texts'):
                    class_radio = gr.Radio(["Zero", "First", "Double", "Special", "Other"], label="CLASS", visible=True)
                    class_txt = gr.Textbox(label="Other class", placeholder="Third", visible=False)

                    class_radio.change(fns.txt_class_visibility, class_radio, class_txt)


                    season_radio = gr.Radio(["Atom01", "Binary01", "Cream01", "Divine01", "Other"], label="SEASON", type="value")

                    with gr.Row():
                        season_txt = gr.Textbox(label="Other SEASON", interactive=True, visible=False)
                        season_txt_outline = gr.Textbox(label="Other SEASON Outline", interactive=True, visible=False)

                    season_radio.change(fns.txt_season_visibility, season_radio, [season_txt, season_txt_outline])


                sign = gr.Dropdown(choices=pre.names, label="Sign", type='value')



                with gr.Accordion('QR Options'):
                    qr_url_dropdown = gr.Dropdown(["tripleS website", "tripleS youtube", "tripleS 𝕏", "tripleS discord"], label="QR url", info="Can write any url", allow_custom_value=True, value='https://objektify.wav.haus')

                    qr_Logo_radio = gr.Radio(["tripleS", "None"], label="QR Logo", value="None", interactive=True)


                right_btn = gr.Checkbox(label="Add Rights", value=False, interactive=True)
                generate_btn_back = gr.Button(value="Generate")

            with gr.Column():
                outputs2 = gr.Image(label="Edited Image", scale=10)

    with gr.Tab("Export") as ex:
        with gr.Row():
            ex_img_f = gr.Image(label="Front", type="pil", interactive=False)
            ex_file_f = gr.File(render=True, visible=False)
            ex_img_b = gr.Image(label="Back", type="pil", interactive=False)
            with gr.Column():
                with gr.Group():
                    with gr.Row():
                        ex_hide_blur = gr.Checkbox(label="Blur")
                        ex_hide_blur_range = gr.Slider(1, 20, 5, label="Blur Range", step=1, interactive=False, scale=10)

                        ex_hide_blur.select(fns.ex_blur, ex_hide_blur, ex_hide_blur_range)

                with gr.Group():
                    watermark = gr.Textbox(label="Add watermark")

                ex_gen_file_count = gr.Radio(["One File", "Two Files"], label="Export Type", value="One File")
                ex_btn = gr.Button(value="Export")
        with gr.Row():
            ex_img_f_out = gr.Image(label="Front", interactive=False)
            ex_img_b_out = gr.Image(label="Back", visible=False, interactive=False)


    with gr.Tab("Preset", visible=False) as beta:
        gr.Markdown(
            """
            ## This is the beta channel for testing new features.
            ### Please join [here](https://discord.com/invite/KAuTCUGXpH) to report bugs and provide feedback.
            """
        )
        with gr.Row():
            with gr.Column():
                preset_name = gr.Textbox(label="Set preset name", placeholder="aespa winter")
                artist_name = gr.Dropdown(label="Artist name", multiselect=True, allow_custom_value=True, info="Press Enter to add more than 1 people (Max 12)", max_choices=12)
                group_name = gr.Textbox(label="Group name", placeholder="aespa")
                company_name = gr.Textbox(label="Company name", placeholder="SM ent")

            top_logo_img = gr.Image(label="Top Logo", type='pil', image_mode='RGBA')
            qr_logo_img = gr.Image(label="QR Logo", type='pil', image_mode='RGBA')

            with gr.Accordion('Signs', open=True):
                with gr.Column():
                    sign_img1 = gr.Image(label="Sign", visible=False, interactive=True, type='pil', image_mode='RGBA')
                    sign_img2 = gr.Image(label="Sign", visible=False, interactive=True, type='pil', image_mode='RGBA')
                    sign_img3 = gr.Image(label="Sign", visible=False, interactive=True, type='pil', image_mode='RGBA')
                    sign_img4 = gr.Image(label="Sign", visible=False, interactive=True, type='pil', image_mode='RGBA')
                    sign_img5 = gr.Image(label="Sign", visible=False, interactive=True, type='pil', image_mode='RGBA')
                    sign_img6 = gr.Image(label="Sign", visible=False, interactive=True, type='pil', image_mode='RGBA')
                    sign_img7 = gr.Image(label="Sign", visible=False, interactive=True, type='pil', image_mode='RGBA')
                    sign_img8 = gr.Image(label="Sign", visible=False, interactive=True, type='pil', image_mode='RGBA')
                    sign_img9 = gr.Image(label="Sign", visible=False, interactive=True, type='pil', image_mode='RGBA')
                    sign_img10 = gr.Image(label="Sign", visible=False, interactive=True, type='pil', image_mode='RGBA')
                    sign_img11 = gr.Image(label="Sign", visible=False, interactive=True, type='pil', image_mode='RGBA')
                    sign_img12 = gr.Image(label="Sign", visible=False, interactive=True, type='pil', image_mode='RGBA')

        save = gr.Button(value="Save and Apply all")


        artist_name.change(fn=fns.custom_name, inputs=artist_name, outputs=[sign_img1, sign_img2, sign_img3, sign_img4, sign_img5, sign_img6, sign_img7, sign_img8, sign_img9, sign_img10, sign_img11, sign_img12])

    with gr.Tab("ReKord", visible=False):
        rekord_img = gr.Image(type="filepath")

        rekord_member = gr.Dropdown(choices=pre.names, label="Sign", type='index')
        rekord_submit = gr.Button()
        rekord_out = gr.Textbox()
        rekord_submit.click(fn=rekord.upload, inputs=[rekord_img, rekord_member], outputs=rekord_out)

    with gr.Tab(label='Dev_preset', visible=False) as rec:
        preset_file = gr.File()
        preset_file.upload(fn=preset_fn.unzip_file_to_directory, inputs=preset_file)
        preset_zip = gr.Button()
        preset_zip.click(fn=preset_fn.zip_directory)
    with gr.Tab("Dev", visible=False) as dev:

        dev_btn = gr.Button(value="refresh", visible=True, interactive=True)
        dev_out = gr.Textbox()
        dev_drop = gr.Dropdown()
        dev_file = gr.File()
        dev_btn2 = gr.Button(value="move")
        dev_btn3 = gr.Button(value="zip")
        dev_btn4 = gr.Button(value="download")
        dev_btn5 = gr.Button(value="remove")

        dev_btn2.click(fn=fns.dev2, outputs=dev_out)
        dev_btn3.click(fn=fns.dev3, inputs=dev_drop, outputs=dev_out)
        dev_btn4.click(fn=fns.dev4, inputs=dev_drop, outputs=dev_file)
        dev_btn5.click(fn=fns.dev5, inputs=dev_drop, outputs=dev_out)
        dev_btn.click(fns.dev, outputs=[dev_out, dev_drop])
    gr.Markdown(
        """
        ## Made by
        # Discord : hj_sss
        ## official discord
        ### [Click here and Join](https://discord.com/invite/KAuTCUGXpH)
        
        #### Check the objektify official discord for licensing and other data processing policies.
        ### Members or group presets can be added and advertised. Check the official discord.
        """)
    userid = gr.Textbox(visible=False)

    preset.change(fn=preset_fn.apply, inputs=preset,
                  outputs=[group_name_radio, member_name_dropdown, logo_radio, sign, qr_Logo_radio, preset])

    group_name_radio.change(fns.group_member_name_update, [group_name_radio, artist_name, preset], [group_name_txt, member_name_dropdown])

    save.click(fn=beta_fn.save, inputs=[preset_name, artist_name, group_name, company_name, top_logo_img, qr_logo_img, sign_img1, sign_img2, sign_img3, sign_img4, sign_img5, sign_img6, sign_img7, sign_img8, sign_img9, sign_img10, sign_img11, sign_img12, userid, login], outputs=[group_name_radio, member_name_dropdown, logo_radio, sign, qr_Logo_radio, preset])


    generate_btn.click(fn=generate.sorting,
                       inputs=[img_type, img_static, img_dynamic_file, obj_color, obj_color_picker, ai_num, txt_color,
                               txt_color_picker, member_name_dropdown, group_name_radio, group_name_txt, obj_num,
                               obj_alphabet, obj_alphabet_txt, obj_serial, horizontal], outputs=[edited_img, ex_img_f, ex_file_f])

    ex_gen_file_count.change(fns.ex_gen_count, ex_gen_file_count, ex_img_b_out)


    change_ai.click(fns.change_ran, obj_color, ai_num)

    generate_btn_back.click(fn=generate.back,
                            inputs=[obj_color, obj_color_picker, ai_num, txt_color, txt_color_picker, group_name_radio,
                                    group_name_txt, member_name_dropdown, obj_num, obj_alphabet, obj_alphabet_txt,
                                    obj_serial, obj_outline_color, obj_outline_color_picker, logo_radio, class_radio,
                                    class_txt, season_radio, season_txt, season_txt_outline, sign, qr_url_dropdown,
                                    qr_Logo_radio, right_btn, preset], outputs=[outputs2, ex_img_b])

    back_tab.select(fn=generate.back,
                    inputs=[obj_color, obj_color_picker, ai_num, txt_color, txt_color_picker, group_name_radio,
                            group_name_txt, member_name_dropdown, obj_num, obj_alphabet, obj_alphabet_txt, obj_serial,
                            obj_outline_color, obj_outline_color_picker, logo_radio, class_radio, class_txt,
                            season_radio, season_txt, season_txt_outline, sign, qr_url_dropdown, qr_Logo_radio,
                            right_btn, preset], outputs=[preview, ex_img_b])

    ex_btn.click(fn=generate.ex_gen, inputs=[ex_img_f, ex_file_f, ex_img_b, ex_gen_file_count, ex_hide_blur, ex_hide_blur_range, watermark], outputs=[ex_img_f_out, ex_img_b_out])
    demo.load(fn=login_discord.id_check, outputs=[login, dev, rec, userid, beta, preset])
    demo.load(fn=preset_fn.refresh, outputs=preset)
    demo.load(None, js=ga_load)



if __name__ == "__main__":
    print("http://localhost:7860")
    demo.launch(server_name="0.0.0.0")