Update app.py

app.py

@@ -684,17 +684,66 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
                 height=256
             )
 
-    # 이벤트 핸들러
-    def get_text_params():
-        return {
-            'text': text_input.value,
-            'font_size': font_size.value,
-            'thickness': thickness.value,
-            'color': color_dropdown.value,
-            'opacity': opacity_slider.value,
-            'x_position': text_x_position.value,
-            'y_position': text_y_position.value
-        }
+
+# UI 구성 부분에서 이벤트 핸들러 함수들을 먼저 정의
+def update_controls(bg_prompt):
+    """배경 프롬프트 입력 여부에 따라 컨트롤 표시 업데이트"""
+    is_visible = bool(bg_prompt)
+    return [
+        gr.update(visible=True),  # aspect_ratio는 항상 표시
+        gr.update(visible=is_visible)  # object_controls
+    ]
+
+def update_process_button(img, prompt):
+    """프로세스 버튼 상태 업데이트"""
+    return gr.update(
+        interactive=bool(img and prompt),
+        variant="primary" if bool(img and prompt) else "secondary"
+    )
+
+def get_text_params():
+    """텍스트 파라미터 가져오기"""
+    return {
+        'text': text_input.value,
+        'font_size': font_size.value,
+        'thickness': thickness.value,
+        'color': color_dropdown.value,
+        'opacity': opacity_slider.value,
+        'x_position': text_x_position.value,
+        'y_position': text_y_position.value
+    }
+
+
+    
+    # 위치 버튼 이벤트
+    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")
+    ]:
+        btn.click(fn=lambda p=pos: p, outputs=position)
+
+    # 이벤트 바인딩
+    input_image.change(
+        fn=update_process_button,
+        inputs=[input_image, text_prompt],
+        outputs=process_btn,
+        queue=False
+    )
+    
+    text_prompt.change(
+        fn=update_process_button,
+        inputs=[input_image, text_prompt],
+        outputs=process_btn,
+        queue=False
+    )
+
+    bg_prompt.change(
+        fn=update_controls,
+        inputs=[bg_prompt],
+        outputs=[aspect_ratio, object_controls],
+        queue=False
+    )
 
     # 위치 버튼 이벤트
     for btn, pos in [
@@ -716,29 +765,20 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
             scale_slider,
             gr.State(get_text_params)
         ],
-        outputs=[combined_image, extracted_image]
+        outputs=[combined_image, extracted_image],
+        queue=True
     )
 
-    # UI 업데이트 이벤트
-    input_image.change(
-        fn=update_process_button,
-        inputs=[input_image, text_prompt],
-        outputs=process_btn
-    )
-    
-    text_prompt.change(
-        fn=update_process_button,
-        inputs=[input_image, text_prompt],
-        outputs=process_btn
-    )
+if HF_TOKEN:
+    login(token=HF_TOKEN, add_to_git_credential=False)
 
-    bg_prompt.change(
-        fn=update_controls,
-        inputs=[bg_prompt],
-        outputs=[aspect_ratio, object_controls]
-    )
+# FLUX 파이프라인 초기화 수정
+pipe = FluxPipeline.from_pretrained(
+    "black-forest-labs/FLUX.1-dev",
+    torch_dtype=torch.float16,
+    token=HF_TOKEN  # use_auth_token 대신 token 사용
+)
 
-# 런처 설정
 demo.queue(max_size=5)
 demo.launch(
     server_name="0.0.0.0",