Update app.py

app.py

@@ -119,6 +119,9 @@ def get_data(text):
         results['h'] = None
     return results
 
+def send_to_img2img(image):
+    result = Image.open(image)
+    return [result]
 
 def send_to_txt2img(image):
 
@@ -171,7 +174,7 @@ def txt2img(prompt, negative_prompt, model, steps, sampler, cfg_scale, width, he
 
     job = prodia_client.wait(result)
 
-    return job["imageUrl"]
+    return [job["imageUrl"]]
 
 
 def img2img(input_image, denoising, prompt, negative_prompt, model, steps, sampler, cfg_scale, width, height, seed):
@@ -191,7 +194,7 @@ def img2img(input_image, denoising, prompt, negative_prompt, model, steps, sampl
 
     job = prodia_client.wait(result)
 
-    return job["imageUrl"]
+    return [job["imageUrl"]]
 
 
 css = """
@@ -240,7 +243,10 @@ with gr.Blocks(css=css) as demo:
                         seed = gr.Number(label="Seed", value=-1)
 
                 with gr.Column(scale=2):
-                    image_output = gr.Image(value="https://images.prodia.xyz/8ede1a7c-c0ee-4ded-987d-6ffed35fc477.png")
+                    image_output = gr.Gallery(show_label=False, columns=4)
+                    send_to_img2img = gr.Button(variant="secondary", value="Send to img2img")
+
+            send_to_img2img.click(send_to_img2img, inputs=image_output, outputs=i2i_image_input)
     
             text_button.click(txt2img, inputs=[prompt, negative_prompt, model, steps, sampler, cfg_scale, width, height,
                                                seed], outputs=image_output, concurrency_limit=64)
@@ -279,7 +285,7 @@ with gr.Blocks(css=css) as demo:
                         i2i_seed = gr.Number(label="Seed", value=-1)
 
                 with gr.Column(scale=2):
-                    i2i_image_output = gr.Image(value="https://images.prodia.xyz/8ede1a7c-c0ee-4ded-987d-6ffed35fc477.png")
+                    i2i_image_output = gr.Gallery(show_label=False, columns=4)
     
             i2i_text_button.click(img2img, inputs=[i2i_image_input, i2i_denoising, i2i_prompt, i2i_negative_prompt,
                                                    model, i2i_steps, i2i_sampler, i2i_cfg_scale, i2i_width, i2i_height,