Update app.py

app.py

@@ -279,13 +279,6 @@ def on_change_prompt(img: Image.Image | None, prompt: str | None, bg_prompt: str
     return gr.update(interactive=bool(img and prompt))
     
 with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
-    gr.HTML("""
-        <div class="main-title">
-            <h1>🎨 Advanced Image Object Extractor</h1>
-            <p>Extract objects from images using text prompts or bounding boxes</p>
-        </div>
-    """)
-
     with gr.Tabs() as tabs:
         with gr.Tab("✨ Extract by Text", id="tab_prompt"):
             with gr.Row(equal_height=True):
@@ -323,19 +316,36 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
 
             with gr.Accordion("📚 Examples", open=False):
                 examples = [
-                    ["examples/text.jpg", "text"],            
-                    ["examples/potted-plant.jpg", "potted plant"],
-                    ["examples/chair.jpg", "chair"],
-                    ["examples/black-lamp.jpg", "black lamp"],
+                    {
+                        "image": "examples/text.jpg",
+                        "prompt": "text",
+                        "bg_prompt": "white background"
+                    },            
+                    {
+                        "image": "examples/potted-plant.jpg",
+                        "prompt": "potted plant",
+                        "bg_prompt": "natural garden background"
+                    },
+                    {
+                        "image": "examples/chair.jpg",
+                        "prompt": "chair",
+                        "bg_prompt": "modern living room"
+                    },
+                    {
+                        "image": "examples/black-lamp.jpg",
+                        "prompt": "black lamp",
+                        "bg_prompt": "minimalist interior"
+                    }
                 ]
                 ex = gr.Examples(
                     examples=examples,
-                    inputs=[iimg, prompt],
+                    inputs=[iimg, prompt, bg_prompt],
                     outputs=[oimg, dlbt],
                     fn=process_prompt,
                     cache_examples=True
                 )
 
+        # Bounding Box 탭
         with gr.Tab("📏 Extract by Box", id="tab_bb"):
             with gr.Row(equal_height=True):
                 with gr.Column(scale=1, min_width=400):
@@ -369,32 +379,24 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
                 examples_bb = [
                     {
                         "image": "examples/text.jpg",
-                        "boxes": [{"xmin": 51, "ymin": 511, "xmax": 639, "ymax": 1255}],
-                        "white background"  # bg_prompt 예제 추가
-
+                        "boxes": [{"xmin": 51, "ymin": 511, "xmax": 639, "ymax": 1255}]
                     },            
                     {
                         "image": "examples/potted-plant.jpg",
-                        "boxes": [{"xmin": 51, "ymin": 511, "xmax": 639, "ymax": 1255}],
-                        "natural garden background"                        
+                        "boxes": [{"xmin": 51, "ymin": 511, "xmax": 639, "ymax": 1255}]
                     },
                     {
                         "image": "examples/chair.jpg",
-                        "boxes": [{"xmin": 98, "ymin": 330, "xmax": 973, "ymax": 1468}],
-                        "modern living room"                        
+                        "boxes": [{"xmin": 98, "ymin": 330, "xmax": 973, "ymax": 1468}]
                     },
                     {
                         "image": "examples/black-lamp.jpg",
-                        "boxes": [{"xmin": 88, "ymin": 148, "xmax": 700, "ymax": 1414}],
-                        "minimalist interior"
-               
-                    },
+                        "boxes": [{"xmin": 88, "ymin": 148, "xmax": 700, "ymax": 1414}]
+                    }
                 ]
                 ex_bb = gr.Examples(
                     examples=examples_bb,
-
-                    inputs=[iimg, prompt, bg_prompt],  # bg_prompt 추가
-                    
+                    inputs=[annotator],
                     outputs=[oimg_bb, dlbt_bb],
                     fn=process_bbox,
                     cache_examples=True
@@ -405,14 +407,12 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
     for inp in [iimg, prompt]:
         inp.change(
             fn=on_change_prompt,
-
-            inputs=[iimg, prompt, bg_prompt],  # bg_prompt 추가
-
+            inputs=[iimg, prompt, bg_prompt],
             outputs=[btn],
         )
     btn.click(
         fn=process_prompt,
-        inputs=[iimg, prompt, bg_prompt],  # bg_prompt 추가
+        inputs=[iimg, prompt, bg_prompt],
         outputs=[oimg, dlbt],
         api_name=False,
     )
@@ -430,35 +430,6 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
         api_name=False,
     )
 
-# CSS 스타일 정의
-css = """
-footer {display: none}
-.main-title {
-    text-align: center;
-    margin: 2em 0;
-}
-.main-title h1 {
-    color: #2196F3;
-    font-size: 2.5em;
-}
-.container {
-    max-width: 1200px;
-    margin: auto;
-    padding: 20px;
-}
-"""
-
-# 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
-)
-
-# Launch settings
 demo.queue(max_size=30, api_open=False)
 demo.launch(
     show_api=False,
@@ -466,4 +437,4 @@ demo.launch(
     server_name="0.0.0.0",
     server_port=7860,
     show_error=True
-)
+)