Update app.py

app.py

@@ -261,6 +261,7 @@ button.primary:hover {
 """
 
 # UI 부분만 수정
+# Main Gradio app
 with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
     gr.HTML("""
         <div class="main-title">
@@ -269,9 +270,9 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
         </div>
     """)
 
-    with gr.Tabs(selected=0) as tabs:
-        # Text 탭
-        with gr.TabItem("Extract by Text") as text_tab:
+    with gr.Tabs(selected=0):
+        # Tab for Text-based Object Extraction
+        with gr.TabItem("Extract by Text"):
             with gr.Row():
                 with gr.Column(scale=1):
                     input_image = gr.Image(
@@ -279,17 +280,16 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
                         label="Upload Image",
                         interactive=True
                     )
-                    with gr.Row():
-                        text_prompt = gr.Textbox(
-                            label="Object to Extract",
-                            placeholder="Enter what you want to extract...",
-                            interactive=True
-                        )
-                        bg_prompt = gr.Textbox(
-                            label="Background Prompt (optional)",
-                            placeholder="Describe the background...",
-                            interactive=True
-                        )
+                    text_prompt = gr.Textbox(
+                        label="Object to Extract",
+                        placeholder="Enter what you want to extract...",
+                        interactive=True
+                    )
+                    bg_prompt = gr.Textbox(
+                        label="Background Prompt (optional)",
+                        placeholder="Describe the background...",
+                        interactive=True
+                    )
                     process_btn = gr.Button(
                         "Process",
                         variant="primary",
@@ -297,7 +297,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
                     )
 
                 with gr.Column(scale=1):
-                    output_display = ImageSlider(
+                    output_display = gr.Gallery(
                         label="Results",
                         show_download_button=False,
                         visible=True
@@ -307,29 +307,19 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
                         visible=True
                     )
 
-            with gr.Accordion("Examples", open=False):
-                gr.Examples(
-                    examples=[
-                        ["examples/text.jpg", "text", "white background"],
-                        ["examples/black-lamp.jpg", "black lamp", "minimalist interior"]
-                    ],
-                    inputs=[input_image, text_prompt, bg_prompt],
-                    outputs=[output_display, download_btn],
-                    fn=process_prompt,
-                    cache_examples=True
-                )
-
-        # Box 탭
-        with gr.TabItem("Extract by Box") as box_tab:
+
+        # Tab for Box-based Object Extraction
+        with gr.TabItem("Extract by Box"):
             with gr.Row():
                 with gr.Column(scale=1):
-                    box_annotator = image_annotator(
-                        image_type="pil",
-                        disable_edit_boxes=False,
-                        show_download_button=False,
-                        show_share_button=False,
-                        single_box=True,
-                        label="Draw Box Around Object",
+                    box_image = gr.Image(
+                        type="pil",
+                        label="Upload Image for Box",
+                        interactive=True
+                    )
+                    box_input = gr.Textbox(
+                        label="Bounding Box (xmin, ymin, xmax, ymax)",
+                        placeholder="Enter bounding box coordinates as a list",
                         interactive=True
                     )
                     box_btn = gr.Button(
@@ -339,42 +329,17 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
                     )
 
                 with gr.Column(scale=1):
-                    box_output = ImageSlider(
+                    box_output = gr.Gallery(
                         label="Results",
                         show_download_button=False,
                         visible=True
                     )
                     box_download = gr.DownloadButton(
-                        "Download Result",
+                        "Download Box Result",
                         visible=True
                     )
 
-            with gr.Accordion("Examples", open=False):
-                gr.Examples(
-                    examples=[
-                        {
-                            "image": "examples/text.jpg",
-                            "boxes": [{"xmin": 51, "ymin": 511, "xmax": 639, "ymax": 1255}]
-                        },
-                        {
-                            "image": "examples/black-lamp.jpg",
-                            "boxes": [{"xmin": 88, "ymin": 148, "xmax": 700, "ymax": 1414}]
-                        }
-                    ],
-                    inputs=[box_annotator],
-                    outputs=[box_output, box_download],
-                    fn=process_bbox,
-                    cache_examples=True
-                )
-
-    # Event handlers
-    def update_process_button(img, prompt):
-        return gr.Button.update(
-            interactive=bool(img and prompt),
-            variant="primary" if bool(img and prompt) else "secondary"
-        )
-
-    # Text tab events
+    # Event bindings
     input_image.change(
         fn=update_process_button,
         inputs=[input_image, text_prompt],
@@ -388,7 +353,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
         outputs=process_btn,
         queue=False
     )
-    
+
     process_btn.click(
         fn=process_prompt,
         inputs=[input_image, text_prompt, bg_prompt],
@@ -396,27 +361,19 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
         queue=True
     )
 
-    # Box tab events
-    box_annotator.change(
-        fn=lambda x: gr.Button.update(interactive=bool(x)),
-        inputs=[box_annotator],
+    box_image.change(
+        fn=update_process_button,
+        inputs=[box_image, box_input],
         outputs=box_btn,
         queue=False
     )
-    
+
     box_btn.click(
         fn=process_bbox,
-        inputs=[box_annotator],
+        inputs=[box_image, box_input],
         outputs=[box_output, box_download],
         queue=True
     )
-
-# Launch settings
+    
 demo.queue(max_size=30, api_open=False)
-demo.launch(
-    show_api=False,
-    share=False,
-    server_name="0.0.0.0",
-    server_port=7860,
-    show_error=True
-)
+demo.launch()