Update app.py

app.py

@@ -315,94 +315,53 @@ button.primary:hover {
 }
 """
 
-# UI 부분만 수정
-# Main Gradio app
 # UI 부분 수정
 with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
     gr.HTML("""
         <div class="main-title">
             <h1>🎨 Image Object Extractor</h1>
-            <p>Extract objects from images using text prompts or bounding boxes</p>
+            <p>Extract objects from images using text prompts</p>
         </div>
     """)
 
-    with gr.Tabs(selected=0):
-        # Text-based extraction tab
-        with gr.TabItem("Extract by Text"):
+    with gr.Row():
+        with gr.Column(scale=1):
+            input_image = gr.Image(
+                type="pil",
+                label="Upload Image",
+                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",
+                interactive=False
+            )
+
+        with gr.Column(scale=1):
             with gr.Row():
-                with gr.Column(scale=1):
-                    input_image = gr.Image(
-                        type="pil",
-                        label="Upload Image",
-                        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",
-                        interactive=False
-                    )
-
-                with gr.Column(scale=1):
-                    with gr.Row():
-                        combined_image = gr.Image(
-                            label="Combined Result",
-                            show_download_button=True,
-                            type="pil",
-                            height=512
-                        )
-                    with gr.Row():
-                        extracted_image = gr.Image(
-                            label="Extracted Object",
-                            show_download_button=True,
-                            type="pil",
-                            height=256
-                        )
-
-        # Box-based extraction tab
-        with gr.TabItem("Extract by Box"):
+                combined_image = gr.Image(
+                    label="Combined Result",
+                    show_download_button=True,
+                    type="pil",
+                    height=512
+                )
             with gr.Row():
-                with gr.Column(scale=1):
-                    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 coordinates as [x1, y1, x2, y2]",
-                        interactive=True
-                    )
-                    box_btn = gr.Button(
-                        "Extract Selection",
-                        variant="primary",
-                        interactive=False
-                    )
-
-                with gr.Column(scale=1):
-                    with gr.Row():
-                        box_combined_image = gr.Image(
-                            label="Combined Result",
-                            show_download_button=True,
-                            type="pil",
-                            height=512
-                        )
-                    with gr.Row():
-                        box_extracted_image = gr.Image(
-                            label="Extracted Object",
-                            show_download_button=True,
-                            type="pil",
-                            height=256
-                        )
+                extracted_image = gr.Image(
+                    label="Extracted Object",
+                    show_download_button=True,
+                    type="pil",
+                    height=256
+                )
 
     # Event bindings
     input_image.change(
@@ -422,30 +381,9 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
     process_btn.click(
         fn=process_prompt,
         inputs=[input_image, text_prompt, bg_prompt],
-        outputs=[combined_image, extracted_image],  # 출력 변경
-        queue=True
-    )
-
-    box_image.change(
-        fn=update_box_button,
-        inputs=[box_image, box_input],
-        outputs=box_btn,
-        queue=False
-    )
-
-    box_input.change(
-        fn=update_box_button,
-        inputs=[box_image, box_input],
-        outputs=box_btn,
-        queue=False
-    )
-
-    box_btn.click(
-        fn=process_bbox,
-        inputs=[box_image, box_input],
-        outputs=[box_combined_image, box_extracted_image],  # 출력 변경
+        outputs=[combined_image, extracted_image],
         queue=True
     )
 
 demo.queue(max_size=30, api_open=False)
-demo.launch()
+demo.launch()