add gradio_imageslider

app.py

@@ -5,6 +5,7 @@ import gradio as gr
 import httpx
 from environs import Env
 from gradio_image_annotation import image_annotator
+from gradio_imageslider import ImageSlider
 from PIL import Image
 
 env = Env()
@@ -18,7 +19,10 @@ with env.prefixed("ERASER_"):
 auth = None if API_KEY is None else httpx.BasicAuth("hf", API_KEY)
 
 
-def process_bbox(prompts: dict[str, Any], request: gr.Request | None) -> Image.Image:
+def process_bbox(
+    prompts: dict[str, Any],
+    request: gr.Request | None,
+) -> tuple[Image.Image, Image.Image]:
     assert isinstance(img := prompts["image"], Image.Image)
     assert isinstance(boxes := prompts["boxes"], list)
     assert len(boxes) == 1
@@ -40,7 +44,9 @@ def process_bbox(prompts: dict[str, Any], request: gr.Request | None) -> Image.I
             headers=headers,
         )
     r.raise_for_status()
-    return Image.open(io.BytesIO(r.content))
+
+    output_image = Image.open(io.BytesIO(r.content))
+    return (img, output_image)
 
 
 def on_change_bbox(prompts: dict[str, Any]):
@@ -51,7 +57,7 @@ def process_prompt(
     img: Image.Image,
     prompt: str,
     request: gr.Request | None,
-) -> Image.Image:
+) -> tuple[Image.Image, Image.Image]:
     data = {"prompt": prompt}
     headers = {}
     if request:  # avoid DOS - can be None despite type hint!
@@ -69,7 +75,9 @@ def process_prompt(
             headers=headers,
         )
     r.raise_for_status()
-    return Image.open(io.BytesIO(r.content))
+
+    output_image = Image.open(io.BytesIO(r.content))
+    return (img, output_image)
 
 
 def on_change_prompt(img: Image.Image | None, prompt: str | None):
@@ -125,7 +133,7 @@ with gr.Blocks() as demo:
                 iimg = gr.Image(type="pil", label="Input")
                 prompt = gr.Textbox(label="What should we erase?")
             with gr.Column():
-                oimg = gr.Image(show_label=False, label="Output")
+                oimg = ImageSlider(label="Output")
         with gr.Row():
             btn = gr.Button("Erase Object", interactive=False)
         for inp in [iimg, prompt]:
@@ -195,7 +203,7 @@ with gr.Blocks() as demo:
                     label="Input",
                 )
             with gr.Column():
-                oimg = gr.Image(show_label=False, label="Output")
+                oimg = ImageSlider(label="Output")
         with gr.Row():
             btn = gr.Button("Erase Object", interactive=False)
         annotator.change(

requirements.txt

@@ -1,4 +1,5 @@
 gradio_image_annotation==0.2.3
+gradio_imageslider>=0.0.20
 environs>=11.0.0
 httpx>=0.27.0
 pillow>=10.4.0