import gradio as gr
import numpy as np
from options import Banner, Video
from huggingface_hub import login
import os
login(token=os.getenv("TOKEN"))

MAX_SEED = np.iinfo(np.int32).max
MAX_IMAGE_SIZE = 2048

with gr.Blocks() as demo:
    gr.Markdown("# Create your own Advertisement")
    with gr.Tab("Banner"):
        gr.Markdown("# Take your banner to the next LEVEL!")
        with gr.TabItem("Create your Banner"):
            textInput = gr.Textbox(label="Enter the text to get a good start")
            with gr.Accordion("Advanced Settings", open=False):
                with gr.Row():
                    width = gr.Slider(
                        label="Width",
                        minimum=256,
                        maximum=MAX_IMAGE_SIZE,
                        step=8,
                        value=1024,
                    )

                    height = gr.Slider(
                        label="Height",
                        minimum=256,
                        maximum=MAX_IMAGE_SIZE,
                        step=32,
                        value=1024,
                    )

                guidance_scale = gr.Slider(
                    label="Guidance Scale",
                    minimum=1,
                    maximum=15,
                    step=0.1,
                    value=3.5,
                )

                num_inference_steps = gr.Slider(
                    label="Number of Inference Steps",
                    minimum=1,
                    maximum=50,
                    step=1,
                    value=28,
                )

            submit = gr.Button("Submit")
            submit.click(
                fn=Banner.TextImage,
                inputs=[textInput, width, height, guidance_scale, num_inference_steps],
                outputs=gr.Image()
            )

        with gr.TabItem("Edit your Banner"):
            img = gr.ImageMask(sources=["upload"], layers=False, transforms=[], format="png", label="base image",
                               show_label=True)
            prompt = gr.Textbox(label="Enter the text to get a good start")
            out_img=gr.Image()
            btn = gr.Button()
            btn.click(Banner.Image2Image, [prompt,img], out_img)

        with gr.TabItem("Upgrade your Banner"):
            img = gr.Image()
            prompt = gr.Textbox(label="Enter the text to get a good start")
            btn = gr.Button()
            size = gr.Slider(label="Size", minimum=256, maximum=MAX_IMAGE_SIZE, step=8, value=1024)
            out_img = gr.Image()
            btn.click(Banner.Image2Image_2, [prompt, img,size,num_inference_steps], out_img)


    with gr.Tab("Video"):
        gr.Markdown("# Create your own Video")
        img=gr.Image()
        btn = gr.Button()
        video=gr.PlayableVideo()
        btn.click(Video.Video, img, video)

demo.launch()