import gradio as gr
from sd_utils import *



with gr.Blocks() as interface:
    #gr.HTML(value=HTML_TEMPLATE, show_label=False)
    with gr.Row():
        text_input = gr.Textbox(
            label="Enter your prompt here",
            placeholder="A portrait James bond in the style of ...",
        )
        concept_dropdown = gr.Dropdown(
            label="Select an SD Concept",
            choices=["midjourney", "birb", "indian-watercolor", "Marc Allante", "herge"],
            value='Dream'
        )


        seed = gr.Slider(
            label="Random Seed",
            minimum=0,
            maximum=10000,
            step=1,
            value=42
        )
        inputs = [text_input, concept_dropdown, seed]

    with gr.Row():
        outputs = gr.Gallery(
            label="Generated Image", show_label=True,
            columns=[1], rows=[1], object_fit="contain"
        )

    with gr.Row():
        button = gr.Button("Generate")
        button.click(generate_with_embs_custom_loss, inputs=inputs, outputs=outputs)

    with gr.Row():
        gr.Examples(examples=get_examples(), inputs=inputs, outputs=outputs, fn=generate_art, cache_examples=True)


if __name__ == "__main__":
    interface.launch(enable_queue=True)