Update app.py

app.py

@@ -1009,7 +1009,8 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
             with gr.Group(elem_classes="input-panel"):
                 with gr.Tabs():
                     # 첫 번째 탭: 이미지 업로드 및 인페인팅
-                    with gr.Tab("Image Upload & Inpainting"):
+                    with gr.Tab("Image Upload & Inpainting", elem_classes="tab"):
+
                         input_image = gr.Image(
                             type="pil",
                             label="Upload Image",
@@ -1030,7 +1031,8 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
                             inpaint_btn = gr.Button("Apply Inpainting", variant="primary")
 
                     # 두 번째 탭: 배경 제거 및 생성
-                    with gr.Tab("Background Removal"):
+                    with gr.Tab("Background Removal", elem_classes="tab"):
+
                         text_prompt = gr.Textbox(
                             label="Object to Extract",
                             placeholder="Enter what you want to extract...",
@@ -1057,17 +1059,17 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
                         with gr.Group(elem_classes="controls-panel", visible=False) as object_controls:
                             with gr.Column(scale=1):
                                 with gr.Row():
-                                    btn_top_left = gr.Button("↖", elem_classes="position-btn")
-                                    btn_top_center = gr.Button("↑", elem_classes="position-btn")
-                                    btn_top_right = gr.Button("↗", elem_classes="position-btn")
+                                    btn_top_left = gr.Button("↖", elem_classes=["position-btn", "interactive-element"])
+                                    btn_top_center = gr.Button("↑", elem_classes=["position-btn", "interactive-element"])
+                                    btn_top_right = gr.Button("↗", elem_classes=["position-btn", "interactive-element"])
                                 with gr.Row():
-                                    btn_middle_left = gr.Button("←", elem_classes="position-btn")
-                                    btn_middle_center = gr.Button("•", elem_classes="position-btn")
-                                    btn_middle_right = gr.Button("→", elem_classes="position-btn")
+                                    btn_middle_left = gr.Button("←", elem_classes=["position-btn", "interactive-element"])
+                                    btn_middle_center = gr.Button("•", elem_classes=["position-btn", "interactive-element"])
+                                    btn_middle_right = gr.Button("→", elem_classes=["position-btn", "interactive-element"])
                                 with gr.Row():
-                                    btn_bottom_left = gr.Button("↙", elem_classes="position-btn")
-                                    btn_bottom_center = gr.Button("↓", elem_classes="position-btn", value="selected")
-                                    btn_bottom_right = gr.Button("↘", elem_classes="position-btn")
+                                    btn_bottom_left = gr.Button("↙", elem_classes=["position-btn", "interactive-element"])
+                                    btn_bottom_center = gr.Button("↓", elem_classes=["position-btn", value="selected")
+                                    btn_bottom_right = gr.Button("↘", elem_classes=["position-btn", "interactive-element"])
                             with gr.Column(scale=1):
                                 scale_slider = gr.Slider(
                                     minimum=10,
@@ -1203,6 +1205,22 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
             visible=True
         )
 
+
+    def handle_button_click(btn_pos):
+        """버튼 클릭 이벤트 핸들러"""
+        print(f"Button clicked: {btn_pos}")  # 디버깅용
+        return update_position_and_ui(btn_pos)
+    
+    # 버튼 이벤트 바인딩
+    for btn, pos in position_mapping.items():
+        btn.click(
+            fn=handle_button_click,
+            inputs=[gr.State(pos)],
+            outputs=[position] + list(position_mapping.keys()),
+            api_name=f"btn_click_{pos}"
+        )
+
+    
     # 이벤트 바인딩
     for btn, pos in position_mapping.items():
         btn.click(
@@ -1211,6 +1229,8 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
             api_name=f"update_position_{pos}"
         )
 
+
+    
     inpaint_btn.click(
         fn=process_inpainting_with_feedback,
         inputs=[input_image, mask_input, inpaint_prompt],