Update app.py

app.py

@@ -197,16 +197,28 @@ def process_bbox(prompts: dict[str, Any]) -> tuple[tuple[Image.Image, Image.Imag
 def on_change_bbox(prompts: dict[str, Any] | None):
     return gr.update(interactive=prompts is not None)
 
-def process_prompt(img: Image.Image, prompt: str, bg_prompt: str | None = 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))
 
-def update_button_state(img, prompt):
-    return gr.Button.update(interactive=bool(img and prompt))
 
 
+# Event handler functions
+def update_process_button(img, prompt):
+    return gr.Button.update(
+        interactive=bool(img and prompt),
+        variant="primary" if bool(img and prompt) else "secondary"
+    )
+
+def update_box_button(img, box_input):
+    try:
+        if img and box_input:
+            coords = eval(box_input)
+            if isinstance(coords, list) and len(coords) == 4:
+                return gr.Button.update(interactive=True, variant="primary")
+        return gr.Button.update(interactive=False, variant="secondary")
+    except:
+        return gr.Button.update(interactive=False, variant="secondary")
 # 맨 앞부분에 CSS 정의 추가
 css = """
 footer {display: none}
@@ -362,7 +374,14 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
     )
 
     box_image.change(
-        fn=update_process_button,
+        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
@@ -374,6 +393,6 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
         outputs=[box_output, box_download],
         queue=True
     )
-    
+
 demo.queue(max_size=30, api_open=False)
-demo.launch()
+demo.launch()