import gradio as gr
import os

pix2pix = gr.Blocks.load(name="spaces/patrickvonplaten/instruct-pix2pix")

def infer(prompt,image):
    result = pix2pix(prompt,
        5.5,
        1.5,
        image,
        15,
        "",
        512,
        512,
        0,
        fn_index=0)

    print(result)
    return [os.path.join(result[0], img) for img in os.listdir(result[0])]

with gr.Blocks() as demo:
    with gr.Column():
        prompt = gr.Textbox(placeholder="enter prompt")
        image = gr.Image(source="upload", type="filepath")
        gallery_out = gr.Gallery()
        submit_btn = gr.Button("go")

    inputs = [
        prompt,
        image
    ]
    outputs = [gallery_out]
    
    submit_btn.click(infer, inputs, outputs)

demo.launch().queue()