Update app.py

app.py

@@ -38,7 +38,12 @@ from transformers import pipeline
 from controlnet_union import ControlNetModel_Union
 from pipeline_fill_sd_xl import StableDiffusionXLFillPipeline
 
-
+def debug_event(event_name, *args):
+    """이벤트 디버깅 유틸리티"""
+    print(f"Event '{event_name}' triggered at {time.strftime('%H:%M:%S')}")
+    print(f"Arguments: {args}")
+    return args
+    
 def clear_memory():
     """메모리 정리 함수"""
     gc.collect()
@@ -936,6 +941,21 @@ def update_position(new_position):
     print(f"Position updated to: {new_position}")
     return new_position
 
+def update_position_and_ui(pos):
+    """위치 업데이트 및 UI 반영"""
+    updates = {btn: gr.update(value="selected" if pos_val == pos else "") 
+              for btn, pos_val in position_mapping.items()}
+    updates['position'] = pos
+    return [pos] + [updates[btn] for btn in position_mapping.keys()]
+    
+def process_inpainting_with_feedback(image, mask, prompt):
+    """인페인팅 처리 및 피드백"""
+    try:
+        result = process_inpainting(image, mask, prompt)
+        return result, update_ui_state("inpainting", "Inpainting completed successfully!")
+    except Exception as e:
+        return None, update_ui_state("inpainting", f"Error: {str(e)}", is_error=True)
+        
 def update_controls(bg_prompt):
     """배경 프롬프트 입력 여부에 따라 컨트롤 표시 업데이트"""
     is_visible = bool(bg_prompt)
@@ -1155,35 +1175,22 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
             visible=True
         )
 
-# 버튼 이벤트 바인딩
+# 위치 버튼 이벤트 바인딩
     for btn, pos in position_mapping.items():
         btn.click(
-            fn=lambda p=pos: debug_event("position_update", p),
-            inputs=[],
-            outputs=[status_message]
-        ).then(
-            fn=update_position_and_ui,
-            inputs=[gr.State(pos)],
+            fn=lambda p=pos: update_position_and_ui(p),
             outputs=[position] + list(position_mapping.keys())
         )
 
-# 인페인팅 프로세스
+# 인페인팅 버튼 이벤트
     inpaint_btn.click(
-        fn=lambda: debug_event("inpainting_start"),
-        inputs=[],
-        outputs=[status_message]
-    ).then(
         fn=process_inpainting_with_feedback,
         inputs=[input_image, mask_input, inpaint_prompt],
         outputs=[input_image, status_message]
     )
 
-# 프로세스 버튼
+# 프로세스 버튼 이벤트
     process_btn.click(
-        fn=lambda: debug_event("process_start"),
-        inputs=[],
-        outputs=[status_message]
-    ).then(
         fn=process_prompt,
         inputs=[
             input_image,
@@ -1193,7 +1200,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
             position,
             scale_slider
         ],
-        outputs=[combined_image, extracted_image, status_message]
+        outputs=[combined_image, extracted_image]
     )