adding more outputs

app.py

@@ -165,10 +165,19 @@ def run_open_gen_fill(image, edit_prompt):
     masks = run_segmentation(image, to_replace, device=device)
 
     # Diffusion pipeline for inpainting
-    return run_inpainting(
+    output = run_inpainting(
         image=image, replaced_caption=replaced_caption, masks=masks, device=device
     )
 
+    return (
+        to_replace,
+        replace_with,
+        caption,
+        replaced_caption,
+        masks,
+        output,
+    )
+
 
 def setup_gradio_interface():
     block = gr.Blocks()
@@ -183,6 +192,13 @@ def setup_gradio_interface():
                 run_button_placeholder = gr.Button(value="Run")
 
             with gr.Column():
+                with gr.Row():
+                    to_replace_placeholder = gr.Textbox(label="to_replace")
+                    replace_with_placeholder = gr.Textbox(label="replace_with")
+                image_caption_placeholder = gr.Textbox(label="Image Caption")
+                replace_caption_placeholder = gr.Textbox(label="Replaced Caption")
+                # object_detection_placeholder = gr.Image(type="pil", label="Object Detection")
+                segmentation_placeholder = gr.Image(type="pil", label="Segmentation")
                 output_image_placeholder = gr.Image(type="pil", label="Output Image")
 
         run_button_placeholder.click(
@@ -191,13 +207,29 @@ def setup_gradio_interface():
                 edit_prompt=edit_prompt,
             ),
             inputs=[input_image_placeholder, edit_prompt_placeholder],
-            outputs=[output_image_placeholder],
+            outputs=[
+                to_replace_placeholder,
+                replace_with_placeholder,
+                image_caption_placeholder,
+                replace_caption_placeholder,
+                # object_detection_placeholder,
+                segmentation_placeholder,
+                output_image_placeholder,
+            ],
         )
 
         gr.Examples(
             examples=[["dog.jpeg", "replace the dog with a tiger"]],
             inputs=[input_image_placeholder, edit_prompt_placeholder],
-            outputs=[output_image_placeholder],
+            outputs=[
+                to_replace_placeholder,
+                replace_with_placeholder,
+                image_caption_placeholder,
+                replace_caption_placeholder,
+                # object_detection_placeholder,
+                segmentation_placeholder,
+                output_image_placeholder,
+            ],
             fn=lambda image, edit_prompt: run_open_gen_fill(
                 image=image,
                 edit_prompt=edit_prompt,