import gradio as gr

with gr.Blocks() as demo:
    gr.Markdown("""
    # IP-Adapter Playground

    by [Tony Assi](https://www.tonyassi.com/)
    """)
    with gr.Row():
        with gr.Tab("Text-to-Image"):
            text_ip = gr.Image(label='IP-Adapter Image')
            text_textbox = gr.Textbox(label='Prompt')
            text_button = gr.Button("Generate")
        with gr.Tab("Image-to-Image"):
            image_ip = gr.Image(label='IP-Adapter Image')
            image_image = gr.Image(label='Image')
            image_textbox = gr.Textbox(label='Prompt')
            image_button = gr.Button("Generate")
        with gr.Tab("Inpainting"):
            inpaint_ip = gr.Image(label='IP-Adapter Image')
            inpaint_editor = gr.ImageEditor(label='Image + Mask')
            inpaint_button = gr.Button("Generate")
        output_image = gr.Image(label='Result')

demo.launch()