Update app.py

app.py

@@ -678,6 +678,28 @@ def add_text_to_image(
 # 전역 변수 설정
 current_position = None  # position 상태를 전역으로 관리하기 위한 변수
 
+# 위치 업데이트 함수 수정
+def update_position(new_position):
+    print(f"Position updated to: {new_position}")  # 디버깅용
+    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")
+        ]
+    }
+    
 with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
     gr.HTML("""
         <div class="main-title">
@@ -844,39 +866,19 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
         </style>
     """)
 
-# 위치 업데이트 함수 수정
-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)
+    # 버튼 클릭 이벤트 바인딩
+    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(
         fn=update_process_button,
         inputs=[input_image, text_prompt],
@@ -898,7 +900,6 @@ def update_button_styles():
         queue=False
     )
 
-    # Process 버튼 클릭 이벤트 수정
     process_btn.click(
         fn=process_prompt,
         inputs=[
@@ -906,24 +907,13 @@ def update_button_styles():
             text_prompt,
             bg_prompt,
             aspect_ratio,
-            position,  # position 상태 전달
+            position,
             scale_slider
         ],
         outputs=[combined_image, extracted_image],
         queue=True
     )
 
-
-    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
-        ]
-
-
-    # 텍스트 추가 버튼 이벤트 연결 수정
     add_text_btn.click(
         fn=add_text_to_image,
         inputs=[
@@ -936,16 +926,15 @@ def update_button_styles():
             y_position,
             thickness,
             text_position_type,
-            font_choice  # 새로운 입력 추가
+            font_choice
         ],
         outputs=combined_image
     )
 
-
-demo.queue(max_size=5)  # 큐 크기 제한
+demo.queue(max_size=5)
 demo.launch(
     server_name="0.0.0.0",
     server_port=7860,
     share=False,
-    max_threads=2  # 스레드 수 제한
+    max_threads=2
 )