Update app.py

app.py

@@ -272,6 +272,12 @@ footer {visibility: hidden}
 .example-region {margin-top: 2em; padding: 20px; background: #f0f0f0; border-radius: 10px;}
 """
 
+def process_prompt(img: Image.Image, prompt: str, bg_prompt: str = None) -> tuple[tuple[Image.Image, Image.Image], gr.DownloadButton]:
+    return _process(img, prompt, bg_prompt)
+
+def on_change_prompt(img: Image.Image | None, prompt: str | None, bg_prompt: str | None = None):
+    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">
@@ -364,23 +370,31 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
                     {
                         "image": "examples/text.jpg",
                         "boxes": [{"xmin": 51, "ymin": 511, "xmax": 639, "ymax": 1255}],
+                        "white background"  # bg_prompt 예제 추가
+
                     },            
                     {
                         "image": "examples/potted-plant.jpg",
                         "boxes": [{"xmin": 51, "ymin": 511, "xmax": 639, "ymax": 1255}],
+                        "natural garden background"                        
                     },
                     {
                         "image": "examples/chair.jpg",
                         "boxes": [{"xmin": 98, "ymin": 330, "xmax": 973, "ymax": 1468}],
+                        "modern living room"                        
                     },
                     {
                         "image": "examples/black-lamp.jpg",
                         "boxes": [{"xmin": 88, "ymin": 148, "xmax": 700, "ymax": 1414}],
+                        "minimalist interior"
+               
                     },
                 ]
                 ex_bb = gr.Examples(
                     examples=examples_bb,
-                    inputs=[annotator],
+
+                    inputs=[iimg, prompt, bg_prompt],  # bg_prompt 추가
+                    
                     outputs=[oimg_bb, dlbt_bb],
                     fn=process_bbox,
                     cache_examples=True
@@ -391,7 +405,9 @@ 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],
+
+            inputs=[iimg, prompt, bg_prompt],  # bg_prompt 추가
+
             outputs=[btn],
         )
     btn.click(