Update app.py

app.py

@@ -9,7 +9,7 @@ import threading
 import sys
 import os
 
-# 모델 초기화 상태 플래그
+# 전역 변수 정의
 model_initialized = False
 flux_generator = None
 initialization_message = "모델 로딩 중... 잠시만 기다려주세요."
@@ -56,6 +56,8 @@ class FluxGenerator:
         self.pulid_model = None
         
     def initialize(self):
+        global initialization_message
+        
         try:
             # 필요한 모듈 지연 임포트
             from pulid.pipeline_flux import PuLIDPipeline
@@ -71,6 +73,7 @@ class FluxGenerator:
             if None in [self.model, self.ae, self.t5, self.clip_model]:
                 print("모델 초기화 실패: 하나 이상의 모델 컴포넌트를 로드할 수 없습니다.")
                 self.initialized = False
+                initialization_message = "모델 로드 실패: 일부 컴포넌트를 로드할 수 없습니다."
                 return
             
             self.pulid_model = PuLIDPipeline(
@@ -83,7 +86,6 @@ class FluxGenerator:
             print("모델 초기화 완료!")
             
             # UI 메시지 업데이트
-            global initialization_message
             initialization_message = "모델 로딩 완료! 이제 이미지를 생성할 수 있습니다."
             
         except Exception as e:
@@ -93,13 +95,12 @@ class FluxGenerator:
             self.initialized = False
             
             # UI 메시지 업데이트
-            global initialization_message
             initialization_message = f"모델 로딩 실패: {str(e)}"
 
 
 # 지연 로딩을 위한 백그라운드 초기화 함수
 def initialize_models_in_background():
-    global flux_generator, model_initialized
+    global flux_generator, model_initialized, initialization_message
     
     print("백그라운드에서 모델 초기화 시작...")
     
@@ -123,11 +124,11 @@ def initialize_models_in_background():
         error_msg = f"백그라운드 초기화 중 오류 발생: {str(e)}\n{traceback.format_exc()}"
         print(error_msg)
         model_initialized = False
+        initialization_message = f"모델 초기화 오류: {str(e)}"
 
 
 # 모델 상태 확인 함수
 def check_model_status():
-    global initialization_message
     return initialization_message