Update app.py

app.py

@@ -678,32 +678,21 @@ def add_text_to_image(
 # 전역 변수 설정
 current_position = None  # position 상태를 전역으로 관리하기 위한 변수
 
-# 위치 업데이트 함수 수정
-def update_position(new_position):
+def update_position(new_position, *buttons):
     print(f"Position updated to: {new_position}")  # 디버깅용
-    # 모든 버튼의 스타일 업데이트
-    for btn in [btn_top_left, btn_top_center, btn_top_right,
-                btn_middle_left, btn_middle_center, btn_middle_right,
-                btn_bottom_left, btn_bottom_center, btn_bottom_right]:
-        btn.elem_classes = ["position-btn", "selected"] if btn == new_position else ["position-btn"]
-    return new_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")
-        ]
-    }
-    
+    updates = []
+    for btn in buttons:
+        updates.append(gr.update(elem_classes=["position-btn", "selected"] if btn == new_position else ["position-btn"]))
+    return [new_position] + updates
+
+def update_controls(bg_prompt):
+    """배경 프롬프트 입력 여부에 따라 컨트롤 표시 업데이트"""
+    is_visible = bool(bg_prompt)
+    return [
+        gr.update(visible=is_visible),  # aspect_ratio
+        gr.update(visible=is_visible),  # object_controls
+    ]
+
 with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
     gr.HTML("""
         <div class="main-title">
@@ -870,27 +859,21 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
         </style>
     """)
 
-
     # 버튼 클릭 이벤트 바인딩
-    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_controls(bg_prompt):
-        """배경 프롬프트 입력 여부에 따라 컨트롤 표시 업데이트"""
-        is_visible = bool(bg_prompt)
-        return [
-            gr.update(visible=is_visible),  # aspect_ratio
-            gr.update(visible=is_visible),  # object_controls
-        ]
-
-# 이벤트 바인딩에서도 필요
+    all_position_buttons = [btn_top_left, btn_top_center, btn_top_right,
+                          btn_middle_left, btn_middle_center, btn_middle_right,
+                          btn_bottom_left, btn_bottom_center, btn_bottom_right]
+
+    for btn, pos in zip(all_position_buttons, 
+                       ["top-left", "top-center", "top-right",
+                        "middle-left", "middle-center", "middle-right",
+                        "bottom-left", "bottom-center", "bottom-right"]):
+        btn.click(
+            fn=lambda p=pos: update_position(p),
+            outputs=[position] + all_position_buttons
+        )
+
+    # 이벤트 바인딩
     bg_prompt.change(
         fn=update_controls,
         inputs=bg_prompt,
@@ -898,7 +881,6 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
         queue=False
     )
     
-    # 기타 이벤트 바인딩
     input_image.change(
         fn=update_process_button,
         inputs=[input_image, text_prompt],
@@ -913,8 +895,6 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
         queue=False
     )
 
-
-
     process_btn.click(
         fn=process_prompt,
         inputs=[