Update app.py

app.py

@@ -268,44 +268,45 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
         </div>
     """)
 
-    with gr.Tabs() as tabs:
+    with gr.Tabs(selected=0) as tabs:
         # Text 탭
-        with gr.Tab(label="✨ Extract by Text") as text_tab:
+        with gr.TabItem("Extract by Text") as text_tab:
             with gr.Row():
-                with gr.Column():
+                with gr.Column(scale=1):
                     input_image = gr.Image(
                         type="pil",
                         label="Upload Image",
-                        interactive=True,
-                        elem_id="input_image"
-                    )
-                    text_prompt = gr.Textbox(
-                        label="Object to Extract",
-                        placeholder="Enter what you want to extract...",
-                        interactive=True,
-                        elem_id="text_prompt"
-                    )
-                    bg_prompt = gr.Textbox(
-                        label="Background Prompt (optional)",
-                        placeholder="Describe the background...",
-                        interactive=True,
-                        elem_id="bg_prompt"
+                        tool="select",
+                        height=400,
+                        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
+                        )
                     process_btn = gr.Button(
                         "Process",
                         variant="primary",
-                        elem_id="process_btn"
+                        interactive=False
                     )
 
-                with gr.Column():
+                with gr.Column(scale=1):
                     output_display = ImageSlider(
                         label="Results",
                         show_download_button=False,
-                        elem_id="output_display"
+                        height=400,
+                        visible=True
                     )
                     download_btn = gr.DownloadButton(
                         "Download Result",
-                        elem_id="download_btn"
+                        visible=True
                     )
 
             with gr.Accordion("Examples", open=False):
@@ -321,34 +322,35 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
                 )
 
         # Box 탭
-        with gr.Tab(label="📏 Extract by Box") as box_tab:
+        with gr.TabItem("Extract by Box") as box_tab:
             with gr.Row():
-                with gr.Column():
+                with gr.Column(scale=1):
                     box_annotator = image_annotator(
                         image_type="pil",
+                        height=400,
                         disable_edit_boxes=False,
                         show_download_button=False,
                         show_share_button=False,
                         single_box=True,
                         label="Draw Box Around Object",
-                        interactive=True,
-                        elem_id="box_annotator"
+                        interactive=True
                     )
                     box_btn = gr.Button(
                         "Extract Selection",
                         variant="primary",
-                        elem_id="box_btn"
+                        interactive=False
                     )
 
-                with gr.Column():
+                with gr.Column(scale=1):
                     box_output = ImageSlider(
                         label="Results",
                         show_download_button=False,
-                        elem_id="box_output"
+                        height=400,
+                        visible=True
                     )
                     box_download = gr.DownloadButton(
                         "Download Result",
-                        elem_id="box_download"
+                        visible=True
                     )
 
             with gr.Accordion("Examples", open=False):
@@ -371,38 +373,49 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
 
     # Event handlers
     def update_process_button(img, prompt):
-        return gr.Button.update(interactive=bool(img and prompt))
+        return gr.Button.update(
+            interactive=bool(img and prompt),
+            variant="primary" if bool(img and prompt) else "secondary"
+        )
 
     # Text tab events
     input_image.change(
         fn=update_process_button,
         inputs=[input_image, text_prompt],
-        outputs=process_btn
+        outputs=process_btn,
+        queue=False
     )
+    
     text_prompt.change(
         fn=update_process_button,
         inputs=[input_image, text_prompt],
-        outputs=process_btn
+        outputs=process_btn,
+        queue=False
     )
+    
     process_btn.click(
         fn=process_prompt,
         inputs=[input_image, text_prompt, bg_prompt],
-        outputs=[output_display, download_btn]
+        outputs=[output_display, download_btn],
+        queue=True
     )
 
     # Box tab events
     box_annotator.change(
         fn=lambda x: gr.Button.update(interactive=bool(x)),
         inputs=[box_annotator],
-        outputs=box_btn
+        outputs=box_btn,
+        queue=False
     )
+    
     box_btn.click(
         fn=process_bbox,
         inputs=[box_annotator],
-        outputs=[box_output, box_download]
+        outputs=[box_output, box_download],
+        queue=True
     )
 
-# Launch the app
+# Launch settings
 demo.queue(max_size=30, api_open=False)
 demo.launch(
     show_api=False,