Update app.py

app.py

@@ -220,7 +220,7 @@ def txt2img(prompt, negative_prompt, model, steps, sampler, cfg_scale, width, he
     progress(0.75, desc="Opening image")
     job = prodia_client.wait(result)
     progress(0.99, desc="Sending image")
-    return job["imageUrl"]
+    return [job["imageUrl"]], job["imageUrl"]
 
 
 def img2img(input_image, denoising, prompt, negative_prompt, model, steps, sampler, cfg_scale, width, height, seed, progress=gr.Progress()):
@@ -247,7 +247,7 @@ def img2img(input_image, denoising, prompt, negative_prompt, model, steps, sampl
     job = prodia_client.wait(result)
     progress(0.99, desc="Sending image")
     time.sleep(0.5)
-    return job["imageUrl"]
+    return [job["imageUrl"]], job["imageUrl"]
 
 
 css = """
@@ -296,13 +296,13 @@ with gr.Blocks(css=css) as demo:
                         seed = gr.Number(label="Seed", value=-1)
 
                 with gr.Column(scale=2):
-                    image_output = gr.Image(show_label=False, type="filepath", interactive=False)
+                    image_output = gr.Gallery(show_label=False, rows=2)
                     send_to_img2img = gr.Button(value="Send OUTPUT IMAGE to img2img")
                     send_to_png = gr.Button(value="Send OUTPUT IMAGE to PNG Info")
-                    
+                    past_url = gr.Textbox(visible=False, interactive=False)
 
             text_button.click(txt2img, inputs=[prompt, negative_prompt, model, steps, sampler, cfg_scale, width, height,
-                                               seed], outputs=image_output, concurrency_limit=64)
+                                               seed], outputs=[image_output, past_url], concurrency_limit=64)
 
         with gr.Tab("img2img", id='i2i'):
             with gr.Row():
@@ -316,7 +316,7 @@ with gr.Blocks(css=css) as demo:
             with gr.Row():
                 with gr.Column(scale=3):
                     with gr.Tab("Generation"):
-                        i2i_image_input = gr.Image(type="pil")
+                        i2i_image_input = gr.Image(type="filepath", interactive=True)
 
                         with gr.Row():
                             with gr.Column(scale=1):
@@ -340,13 +340,14 @@ 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(show_label=False, type="filepath", interactive=False)
+                    i2i_image_output = gr.Gallery(show_label=False, rows=2)
                     send_to_png_i2i = gr.Button(value="Send INPUT IMAGE to PNG Info")
+                    i2i_past_url = gr.Textbox(visible=False, interactive=False)
 
             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,
-                                                   i2i_seed], outputs=i2i_image_output, concurrency_limit=64)
-        send_to_img2img.click(send_to_img2img_def, inputs=image_output, outputs=i2i_image_input)
+                                                   i2i_seed], outputs=[i2i_image_output, i2i_past_url], concurrency_limit=64)
+        send_to_img2img.click(send_to_img2img_def, inputs=past_url, outputs=i2i_image_input)
 
         with gr.Tab("PNG Info"):
             def plaintext_to_html(text, classname=None):
@@ -376,7 +377,7 @@ with gr.Blocks(css=css) as demo:
 
             with gr.Row():
                 with gr.Column():
-                    image_input = gr.Image(type="pil")
+                    image_input = gr.Image(type="filepath", interactive=True)
             png_button = gr.Button("Get Info")
             with gr.Row():
                 with gr.Column():
@@ -390,9 +391,9 @@ with gr.Blocks(css=css) as demo:
                                                                                       steps, seed, model, sampler,
                                                                                       width, height, cfg_scale],
                                       concurrency_limit=64)
-        send_to_png.click(send_to_img2img_def, inputs=image_output, outputs=image_input)
-        send_to_img2img_png.click(send_to_img2img_def, inputs=image_input, outputs=i2i_image_input)
-        send_to_png_i2i.click(send_to_img2img_def, inputs=i2i_image_output, outputs=image_input)
+        send_to_png.click(send_to_img2img_def, inputs=past_url, outputs=image_input)
+        send_to_img2img_png.click(send_to_img2img_def, inputs=past_url, outputs=i2i_image_input)
+        send_to_png_i2i.click(send_to_img2img_def, inputs=i2i_past_url, outputs=image_input)
         with gr.Tab("HuggingFace Inference"):
             with gr.Row():
                 gr.Markdown("Add your model from HF.co, enter model ID.")