Update app.py

app.py

@@ -281,6 +281,8 @@ with gr.Blocks(css=css) as demo:
                 with gr.Column(scale=2):
                     image_output = gr.Image(show_label=False, type="filepath", interactive=False)
                     send_to_img2img = gr.Button(value="Send to img2img")
+                    send_to_png = gr.Button(value="Send to PNG Info")
+                    
 
             text_button.click(txt2img, inputs=[prompt, negative_prompt, model, steps, sampler, cfg_scale, width, height,
                                                seed], outputs=image_output, concurrency_limit=64)
@@ -323,6 +325,7 @@ with gr.Blocks(css=css) as demo:
 
                 with gr.Column(scale=2):
                     i2i_image_output = gr.Image(show_label=False, type="filepath", interactive=False)
+                    send_to_png_i2i = gr.Button(value="Send to PNG Info")
 
             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,
@@ -368,6 +371,8 @@ 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_png_i2i.click(send_to_img2img_def, inputs=i2i_image_output, outputs=image_input)
         with gr.Tab("HuggingFace Inference"):
             with gr.Row():
                 gr.Markdown("Add your model from HF.co, enter model ID.")