Upload app.py

app.py

@@ -249,6 +249,9 @@ def draw_points_on_image(image, layer_manager):
 # ============ UI FUNCTIONS ============
 def create_new_layer(name, current_manager):
     """새 레이어 생성"""
+    if current_manager is None:
+        current_manager = LayerManager()
+    
     if not name.strip():
         return current_manager, create_layer_status_html(current_manager), gr.Dropdown(choices=[]), "Please enter a layer name"
 
@@ -325,6 +328,9 @@ def create_layer_status_html(current_manager):
 
 def click_on_image(current_manager, image, point_mode, evt: gr.SelectData):
     """이미지 클릭 처리 - Include/Exclude 모드에 따라 포인트 추가"""
+    if current_manager is None:
+        current_manager = LayerManager()
+    
     if image is None or current_manager.current_layer_id is None:
         return image, current_manager, create_layer_status_html(current_manager), "Please select image and layer"
 
@@ -349,6 +355,9 @@ def click_on_image(current_manager, image, point_mode, evt: gr.SelectData):
 
 def segment_all_layers(current_manager, image, opacity, border_width):
     """모든 레이어를 순서대로 세그멘테이션 실행"""
+    if current_manager is None:
+        current_manager = LayerManager()
+    
     if image is None:
         return None, current_manager, create_layer_status_html(current_manager), "Please upload an image", pd.DataFrame()
     
@@ -414,6 +423,9 @@ def segment_all_layers(current_manager, image, opacity, border_width):
 
 def clear_current_layer(current_manager, image, opacity, border_width):
     """현재 레이어 초기화"""
+    if current_manager is None:
+        current_manager = LayerManager()
+    
     if current_manager.current_layer_id:
         current_manager.clear_current_layer()
 
@@ -433,6 +445,9 @@ def clear_current_layer(current_manager, image, opacity, border_width):
 
 def refresh_visualization(current_manager, image, opacity, border_width):
     """시각화 새로고침"""
+    if current_manager is None:
+        current_manager = LayerManager()
+    
     if image is None:
         return None, "Please upload an image", pd.DataFrame()
 
@@ -456,9 +471,6 @@ custom_css="""
 # No custom JavaScript needed anymore
 custom_js = ""
 
-# 전역 레이어 매니저
-layer_manager = LayerManager()
-
 with gr.Blocks(css=custom_css) as demo:
     with gr.Column(elem_id="col-container"):
         gr.Markdown("# **SAM3 Layer Segmentation Tool**", elem_id="main-title")
@@ -505,7 +517,7 @@ with gr.Blocks(css=custom_css) as demo:
 
                 # 상태
                 status_text = gr.Textbox(label="Status", interactive=False)
-                st_layer_manager = gr.State(layer_manager)
+                st_layer_manager = gr.State(None)  # LayerManager는 함수 내에서 생성
                 point_mode_state = gr.State("positive")  # "positive" or "negative"
 
             with gr.Column(scale=2):
@@ -530,6 +542,9 @@ with gr.Blocks(css=custom_css) as demo:
         
         # 레이어 선택
         def on_layer_select(layer_id, mgr):
+            if mgr is None:
+                mgr = LayerManager()
+            
             if layer_id:
                 mgr.set_current_layer(layer_id)
                 return mgr, create_layer_status_html(mgr), f"Layer '{mgr.layers[layer_id]['name']}' selected"