Update app.py

app.py

@@ -844,23 +844,37 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
         </style>
     """)
 
-    # 위치 업데이트 함수
-    def update_position(new_position, current):
-        print(f"Position updated from {current} to {new_position}")  # 디버깅용
-        return new_position
-
-    # 버튼 클릭 이벤트 바인딩
-    btn_top_left.click(fn=update_position, inputs=["top-left", position], outputs=position)
-    btn_top_center.click(fn=update_position, inputs=["top-center", position], outputs=position)
-    btn_top_right.click(fn=update_position, inputs=["top-right", position], outputs=position)
-    btn_middle_left.click(fn=update_position, inputs=["middle-left", position], outputs=position)
-    btn_middle_center.click(fn=update_position, inputs=["middle-center", position], outputs=position)
-    btn_middle_right.click(fn=update_position, inputs=["middle-right", position], outputs=position)
-    btn_bottom_left.click(fn=update_position, inputs=["bottom-left", position], outputs=position)
-    btn_bottom_center.click(fn=update_position, inputs=["bottom-center", position], outputs=position)
-    btn_bottom_right.click(fn=update_position, inputs=["bottom-right", position], outputs=position)
-
-
+# 위치 업데이트 함수 수정
+def update_position(new_position):
+    print(f"Position updated to: {new_position}")  # 디버깅용
+    return new_position
+
+# 버튼 클릭 이벤트 바인딩 수정
+btn_top_left.click(fn=lambda: update_position("top-left"), outputs=position)
+btn_top_center.click(fn=lambda: update_position("top-center"), outputs=position)
+btn_top_right.click(fn=lambda: update_position("top-right"), outputs=position)
+btn_middle_left.click(fn=lambda: update_position("middle-left"), outputs=position)
+btn_middle_center.click(fn=lambda: update_position("middle-center"), outputs=position)
+btn_middle_right.click(fn=lambda: update_position("middle-right"), outputs=position)
+btn_bottom_left.click(fn=lambda: update_position("bottom-left"), outputs=position)
+btn_bottom_center.click(fn=lambda: update_position("bottom-center"), outputs=position)
+btn_bottom_right.click(fn=lambda: update_position("bottom-right"), outputs=position)
+
+def update_button_styles():
+    return {
+        btn: gr.update(elem_classes=["position-btn", "selected"] if pos == current_position else ["position-btn"])
+        for btn, pos in [
+            (btn_top_left, "top-left"),
+            (btn_top_center, "top-center"),
+            (btn_top_right, "top-right"),
+            (btn_middle_left, "middle-left"),
+            (btn_middle_center, "middle-center"),
+            (btn_middle_right, "middle-right"),
+            (btn_bottom_left, "bottom-left"),
+            (btn_bottom_center, "bottom-center"),
+            (btn_bottom_right, "bottom-right")
+        ]
+    }
 
     # 기존 이벤트 바인딩
     input_image.change(