Update src/app.py

src/app.py

@@ -129,6 +129,29 @@ def process_prompt(img: Image.Image, prompt: str) -> tuple[Image.Image, Image.Im
     )
     return (img, output_image)
 
+# --- NEW BATCH PROCESSING FUNCTIONS ---
+def process_batch_prompt(files: list[Any], prompt: str):
+    """Processes multiple images sequentially with the same prompt."""
+    results = []
+    if not files:
+        return []
+    
+    for file_obj in files:
+        # Open the image from the file path
+        img = Image.open(file_obj.name)
+        resized_img = resize(img)
+        
+        # Run the eraser
+        output_image = _run(
+            ProcessParams(image=resized_img, prompt=prompt),
+        )
+        results.append(output_image)
+        
+    return results
+
+def on_change_batch(files: list[Any] | None, prompt: str | None):
+    return gr.update(interactive=bool(files and prompt))
+# --------------------------------------
 
 def on_change_prompt(img: Image.Image | None, prompt: str | None):
     return gr.update(interactive=bool(img and prompt))
@@ -149,7 +172,9 @@ TITLE = """
 
 with gr.Blocks() as demo:
     gr.HTML(TITLE)
-    with gr.Tab("By prompt", id="tab_prompt"):
+    
+    # --- TAB 1: SINGLE IMAGE (PROMPT) ---
+    with gr.Tab("By prompt (Single)", id="tab_prompt"):
         with gr.Row():
             with gr.Column():
                 iimg = gr.Image(type="pil", label="Input")
@@ -177,30 +202,7 @@ with gr.Blocks() as demo:
                 "examples/white-towels-rattan-basket-white-table-with-bright-room-background.jpg",
                 "soap",
             ],
-            [
-                "examples/interior-decor-with-mirror-potted-plant.jpg",
-                "potted plant",
-            ],
-            [
-                "examples/detail-ball-basketball-court-sunset.jpg",
-                "basketball",
-            ],
-            [
-                "examples/still-life-device-table_23-2150994394.jpg",
-                "glass of water",
-            ],
-            [
-                "examples/knife-fork-green-checkered-napkin_140725-63576.jpg",
-                "knife and fork",
-            ],
-            [
-                "examples/city-night-with-architecture-vibrant-lights_23-2149836930.jpg",
-                "frontmost black car on right lane",
-            ],
-            [
-                "examples/close-up-coffee-latte-wooden-table_23-2147893063.jpg",
-                "coffee cup on plate",
-            ],
+            # ... (Example list abbreviated for brevity, keep your original list here) ...
             [
                 "examples/empty-chair-with-vase-plant_74190-2078.jpg",
                 "chair",
@@ -215,6 +217,8 @@ with gr.Blocks() as demo:
             cache_examples=True,
             cache_mode="eager",
         )
+
+    # --- TAB 2: SINGLE IMAGE (BOUNDING BOX) ---
     with gr.Tab("By bounding box", id="tab_bb"):
         with gr.Row():
             with gr.Column():
@@ -227,65 +231,72 @@ with gr.Blocks() as demo:
                     label="Input",
                 )
             with gr.Column():
-                oimg = gr.ImageSlider(label="Output", max_height=1500, show_fullscreen_button=False)
+                oimg_bb = gr.ImageSlider(label="Output", max_height=1500, show_fullscreen_button=False)
         with gr.Row():
-            btn = gr.ClearButton(components=[oimg], value="Erase Object", interactive=False)
+            btn_bb = gr.ClearButton(components=[oimg_bb], value="Erase Object", interactive=False)
 
         annotator.change(
             fn=on_change_bbox,
             inputs=[annotator],
-            outputs=[btn],
+            outputs=[btn_bb],
         )
-        btn.click(
+        btn_bb.click(
             fn=process_bbox,
             inputs=[annotator],
-            outputs=[oimg],
+            outputs=[oimg_bb],
             api_name=False,
         )
 
-        examples = [
+        examples_bb = [
             {
                 "image": "examples/white-towels-rattan-basket-white-table-with-bright-room-background.jpg",
                 "boxes": [{"xmin": 836, "ymin": 475, "xmax": 1125, "ymax": 1013}],
             },
-            {
-                "image": "examples/interior-decor-with-mirror-potted-plant.jpg",
-                "boxes": [{"xmin": 47, "ymin": 907, "xmax": 397, "ymax": 1633}],
-            },
-            {
-                "image": "examples/detail-ball-basketball-court-sunset.jpg",
-                "boxes": [{"xmin": 673, "ymin": 954, "xmax": 911, "ymax": 1186}],
-            },
-            {
-                "image": "examples/still-life-device-table_23-2150994394.jpg",
-                "boxes": [{"xmin": 429, "ymin": 586, "xmax": 571, "ymax": 834}],
-            },
-            {
-                "image": "examples/knife-fork-green-checkered-napkin_140725-63576.jpg",
-                "boxes": [{"xmin": 972, "ymin": 226, "xmax": 1092, "ymax": 1023}],
-            },
-            {
-                "image": "examples/city-night-with-architecture-vibrant-lights_23-2149836930.jpg",
-                "boxes": [{"xmin": 215, "ymin": 637, "xmax": 411, "ymax": 855}],
-            },
-            {
-                "image": "examples/close-up-coffee-latte-wooden-table_23-2147893063.jpg",
-                "boxes": [{"xmin": 255, "ymin": 456, "xmax": 1080, "ymax": 1064}],
-            },
-            {
+            # ... (Example list abbreviated for brevity, keep your original list here) ...
+             {
                 "image": "examples/empty-chair-with-vase-plant_74190-2078.jpg",
                 "boxes": [{"xmin": 35, "ymin": 320, "xmax": 383, "ymax": 983}],
             },
         ]
 
-        ex = gr.Examples(
-            examples=examples,
+        ex_bb = gr.Examples(
+            examples=examples_bb,
             inputs=[annotator],
-            outputs=[oimg],
+            outputs=[oimg_bb],
             fn=process_bbox,
             cache_examples=True,
             cache_mode="eager",
         )
 
+    # --- TAB 3: BATCH PROCESSING (NEW) ---
+    with gr.Tab("Batch by prompt", id="tab_batch"):
+        gr.Markdown("Upload multiple images and apply the same erase prompt to all of them.")
+        with gr.Row():
+            with gr.Column():
+                # file_count="multiple" allows selecting many images
+                ibatch = gr.File(file_count="multiple", file_types=["image"], label="Upload Images")
+                batch_prompt = gr.Textbox(label="What should we erase from all these images?")
+            with gr.Column():
+                # Gallery displays multiple results
+                obatch = gr.Gallery(label="Batch Results", show_label=True, columns=[2], object_fit="contain", height="auto")
+        
+        with gr.Row():
+            btn_batch = gr.Button("Erase from All Images", interactive=False, variant="primary")
+
+        # Enable button only if files and prompt exist
+        for inp in [ibatch, batch_prompt]:
+            inp.change(
+                fn=on_change_batch,
+                inputs=[ibatch, batch_prompt],
+                outputs=[btn_batch],
+            )
+            
+        btn_batch.click(
+            fn=process_batch_prompt,
+            inputs=[ibatch, batch_prompt],
+            outputs=[obatch],
+            api_name=False,
+        )
+
 demo.queue(max_size=30, api_open=False)
-demo.launch(show_api=False, ssr_mode=False)
+demo.launch(show_api=False, ssr_mode=False)