import time

import gradio as gr


def image_gen(prompt, progress=gr.Progress()):
    progress(0, desc="Starting")
    time.sleep(1)
    progress(0.5)
    time.sleep(1)
    progress(1)
    return "https://www.gradio.app/_app/immutable/assets/gradio.CHB5adID.svg"


def main():
    prompt = gr.Textbox()
    image = gr.Image()
    generate_btn_1 = gr.Button("Generate Image via model 1")
    generate_btn_1.click(
        image_gen, prompt, image, concurrency_limit=2, concurrency_id="gpu_queue"
    )

    generate_btn_2 = gr.Button("Generate Image via model 2")
    generate_btn_2.click(image_gen, prompt, image, concurrency_id="gpu_queue")

    generate_btn_3 = gr.Button("Generate Image via model 3")
    generate_btn_3.click(image_gen, prompt, image, concurrency_id="gpu_queue")