Update app.py

app.py

@@ -183,16 +183,30 @@ def _process(img: Image.Image, prompt: str | BoundingBox | None, bg_prompt: str
 
     return (img, combined, masked_alpha), gr.DownloadButton(value=temp.name, interactive=True)
 
-def process_bbox(prompts: dict[str, Any]) -> tuple[tuple[Image.Image, Image.Image], gr.DownloadButton]:
-    assert isinstance(img := prompts["image"], Image.Image)
-    assert isinstance(boxes := prompts["boxes"], list)
-    if len(boxes) == 1:
-        assert isinstance(box := boxes[0], dict)
-        bbox = tuple(box[k] for k in ["xmin", "ymin", "xmax", "ymax"])
-    else:
-        assert len(boxes) == 0
-        bbox = None
-    return _process(img, bbox)
+def process_bbox(img: Image.Image, box_input: str) -> tuple[list[Image.Image], str]:
+    try:
+        if img is None or box_input.strip() == "":
+            raise gr.Error("Please provide both image and bounding box coordinates")
+        
+        # Parse box coordinates
+        try:
+            coords = eval(box_input)
+            if not isinstance(coords, list) or len(coords) != 4:
+                raise ValueError("Invalid box format")
+            bbox = tuple(int(x) for x in coords)
+        except:
+            raise gr.Error("Invalid box format. Please provide [xmin, ymin, xmax, ymax]")
+        
+        # Process the image
+        results, download_path = _process(img, bbox)
+        
+        # Convert results to list for gallery
+        gallery_images = list(results)
+        
+        return gallery_images, download_path
+    except Exception as e:
+        raise gr.Error(str(e))
+
 
 def on_change_bbox(prompts: dict[str, Any] | None):
     return gr.update(interactive=prompts is not None)
@@ -201,15 +215,28 @@ def on_change_bbox(prompts: dict[str, Any] | None):
 def on_change_prompt(img: Image.Image | None, prompt: str | None, bg_prompt: str | None = None):
     return gr.update(interactive=bool(img and prompt))
 
+def process_prompt(img: Image.Image, prompt: str, bg_prompt: str | None = None) -> tuple[list[Image.Image], str]:
+    try:
+        if img is None or prompt.strip() == "":
+            raise gr.Error("Please provide both image and prompt")
+        
+        # Process the image
+        results, download_path = _process(img, prompt, bg_prompt)
+        
+        # Convert results to list for gallery
+        gallery_images = list(results)
+        
+        return gallery_images, download_path
+    except Exception as e:
+        raise gr.Error(str(e))
 
 
-# Event handler functions
 def update_process_button(img, prompt):
     return gr.Button.update(
         interactive=bool(img and prompt),
         variant="primary" if bool(img and prompt) else "secondary"
     )
-
+    
 def update_box_button(img, box_input):
     try:
         if img and box_input:
@@ -219,6 +246,8 @@ def update_box_button(img, box_input):
         return gr.Button.update(interactive=False, variant="secondary")
     except:
         return gr.Button.update(interactive=False, variant="secondary")
+
+
 # 맨 앞부분에 CSS 정의 추가
 css = """
 footer {display: none}
@@ -283,7 +312,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
     """)
 
     with gr.Tabs(selected=0):
-        # Tab for Text-based Object Extraction
+        # Text-based extraction tab
         with gr.TabItem("Extract by Text"):
             with gr.Row():
                 with gr.Column(scale=1):
@@ -319,8 +348,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
                         visible=True
                     )
 
-
-        # Tab for Box-based Object Extraction
+        # Box-based extraction tab
         with gr.TabItem("Extract by Box"):
             with gr.Row():
                 with gr.Column(scale=1):
@@ -331,7 +359,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
                     )
                     box_input = gr.Textbox(
                         label="Bounding Box (xmin, ymin, xmax, ymax)",
-                        placeholder="Enter bounding box coordinates as a list",
+                        placeholder="Enter coordinates as [x1, y1, x2, y2]",
                         interactive=True
                     )
                     box_btn = gr.Button(
@@ -347,7 +375,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
                         visible=True
                     )
                     box_download = gr.DownloadButton(
-                        "Download Box Result",
+                        "Download Result",
                         visible=True
                     )