Update app.py

app.py

@@ -124,7 +124,7 @@ def _gpu_process(
 def _process(
     img: Image.Image,
     prompt: str | BoundingBox | None,
-) -> tuple[tuple[Image.Image, Image.Image], gr.DownloadButton]:
+) -> tuple[tuple[Image.Image, Image.Image], str]:  # Changed return type
     # enforce max dimensions for pymatting performance reasons
     if img.width > 2048 or img.height > 2048:
         orig_res = max(img.width, img.height)
@@ -150,10 +150,11 @@ def _process(
     to_dl.save(temp, format="PNG")
     temp.close()
 
-    return (img, masked_rgb), gr.DownloadButton(value=temp.name, interactive=True)
+    # Return the file path instead of a DownloadButton
+    return (img, masked_rgb), temp.name
 
 
-def process_bbox(prompts: dict[str, Any]) -> tuple[tuple[Image.Image, Image.Image], gr.DownloadButton]:
+def process_bbox(prompts: dict[str, Any]) -> tuple[tuple[Image.Image, Image.Image], str]:
     assert isinstance(img := prompts["image"], Image.Image)
     assert isinstance(boxes := prompts["boxes"], list)
     if len(boxes) == 1:
@@ -169,7 +170,7 @@ def on_change_bbox(prompts: dict[str, Any] | None):
     return gr.update(interactive=prompts is not None)
 
 
-def process_prompt(img: Image.Image, prompt: str) -> tuple[tuple[Image.Image, Image.Image], gr.DownloadButton]:
+def process_prompt(img: Image.Image, prompt: str) -> tuple[tuple[Image.Image, Image.Image], str]:
     return _process(img, prompt)
 
 
@@ -177,6 +178,14 @@ def on_change_prompt(img: Image.Image | None, prompt: str | None):
     return gr.update(interactive=bool(img and prompt))
 
 
+# Function to update download button with file path
+def update_download_button(file_path: str | None) -> gr.update:
+    if file_path:
+        return gr.update(value=file_path, interactive=True)
+    else:
+        return gr.update(value=None, interactive=False)
+
+
 css = """
 footer {
     visibility: hidden;
@@ -194,20 +203,26 @@ with gr.Blocks(css=css) as demo:
                 btn = gr.Button("Cut Out Object", interactive=False)
             with gr.Column():
                 oimg = ImageSlider(label="Before / After", show_download_button=False, interactive=False)
+                # Use a File component for download instead of DownloadButton
+                file_output = gr.File(label="Download Cutout", visible=False)
                 dlbt = gr.DownloadButton("Download Cutout", interactive=False)
 
-        # btn.add(oimg)  <- 제거됨
-
         for inp in [iimg, prompt]:
             inp.change(
                 fn=on_change_prompt,
                 inputs=[iimg, prompt],
                 outputs=[btn],
             )
+        
+        # Process and update both outputs
+        def process_and_update_prompt(img, prompt):
+            result, file_path = process_prompt(img, prompt)
+            return result, file_path, update_download_button(file_path)
+        
         btn.click(
-            fn=process_prompt,
+            fn=process_and_update_prompt,
             inputs=[iimg, prompt],
-            outputs=[oimg, dlbt],
+            outputs=[oimg, file_output, dlbt],
             api_name=False,
         )
 
@@ -230,12 +245,13 @@ with gr.Blocks(css=css) as demo:
             ],
         ]
 
+        # Don't cache examples to avoid the error
         ex = gr.Examples(
             examples=examples,
             inputs=[iimg, prompt],
-            outputs=[oimg, dlbt],
+            outputs=[oimg, file_output],  # Remove dlbt from outputs
             fn=process_prompt,
-            cache_examples=True,
+            cache_examples=False,  # Disable caching
         )
 
     with gr.Tab("By bounding box", id="tab_bb"):
@@ -249,22 +265,27 @@ with gr.Blocks(css=css) as demo:
                     single_box=True,
                     label="Input",
                 )
-                btn = gr.Button("Cut Out Object", interactive=False)
+                btn2 = gr.Button("Cut Out Object", interactive=False)
             with gr.Column():
-                oimg = ImageSlider(label="Before / After", show_download_button=False)
-                dlbt = gr.DownloadButton("Download Cutout", interactive=False)
-
-        # btn.add(oimg)  <- 제거됨
+                oimg2 = ImageSlider(label="Before / After", show_download_button=False)
+                file_output2 = gr.File(label="Download Cutout", visible=False)
+                dlbt2 = gr.DownloadButton("Download Cutout", interactive=False)
 
         annotator.change(
             fn=on_change_bbox,
             inputs=[annotator],
-            outputs=[btn],
+            outputs=[btn2],
         )
-        btn.click(
-            fn=process_bbox,
+        
+        # Process and update both outputs
+        def process_and_update_bbox(prompts):
+            result, file_path = process_bbox(prompts)
+            return result, file_path, update_download_button(file_path)
+        
+        btn2.click(
+            fn=process_and_update_bbox,
             inputs=[annotator],
-            outputs=[oimg, dlbt],
+            outputs=[oimg2, file_output2, dlbt2],
             api_name=False,
         )
 
@@ -287,14 +308,14 @@ with gr.Blocks(css=css) as demo:
             },
         ]
 
-        ex = gr.Examples(
+        ex2 = gr.Examples(
             examples=examples,
             inputs=[annotator],
-            outputs=[oimg, dlbt],
+            outputs=[oimg2, file_output2],  # Remove dlbt2 from outputs
             fn=process_bbox,
-            cache_examples=True,
+            cache_examples=False,  # Disable caching
         )
 
 
 demo.queue(max_size=30, api_open=False)
-demo.launch(show_api=False)
+demo.launch(show_api=False)