Upload app.py

app.py

@@ -247,26 +247,28 @@ def draw_points_on_image(image, layer_manager):
     return draw_img
 
 # ============ UI FUNCTIONS ============
+def update_layer_selector_choices(manager):
+    """레이어 선택 드롭다운의 choices 업데이트"""
+    choices = [(layer['name'], lid) for lid, layer in manager.layers.items()]
+    return gr.Dropdown(choices=choices, interactive=True, value=manager.current_layer_id)
+
 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"
+        return current_manager, create_layer_status_html(current_manager), update_layer_selector_choices(current_manager), "Please enter a layer name"
 
     # 중복 이름 체크
     for layer_id, layer in current_manager.layers.items():
         if layer['name'] == name.strip():
-            return current_manager, create_layer_status_html(current_manager), gr.Dropdown(choices=[(layer['name'], lid) for lid, layer in current_manager.layers.items()]), f"Layer name '{name}' already exists"
+            return current_manager, create_layer_status_html(current_manager), update_layer_selector_choices(current_manager), f"Layer name '{name}' already exists"
 
     layer_id = current_manager.create_layer(name.strip())
     current_manager.set_current_layer(layer_id)
-    
-    # 드롭다운 선택지 업데이트
-    choices = [(layer['name'], lid) for lid, layer in current_manager.layers.items()]
 
-    return current_manager, create_layer_status_html(current_manager), gr.Dropdown(choices=choices, value=layer_id), f"Layer '{name}' created"
+    return current_manager, create_layer_status_html(current_manager), update_layer_selector_choices(current_manager), f"Layer '{name}' created"
 
 def create_layer_status_html(current_manager):
     """레이어 상태 표시 HTML 생성 (시각적 표시만)"""
@@ -462,16 +464,10 @@ def refresh_visualization(current_manager, image, opacity, border_width):
 
 
 # ============ GRADIO INTERFACE ============
-custom_css="""
-#col-container { margin: 0 auto; max-width: 1200px; }
-#main-title h1 { font-size: 2.1em !important; }
-.layer-button { margin: 2px; }
-"""
-
 # No custom JavaScript needed anymore
 custom_js = ""
 
-with gr.Blocks(css=custom_css) as demo:
+with gr.Blocks() as demo:
     with gr.Column(elem_id="col-container"):
         gr.Markdown("# **SAM3 Layer Segmentation Tool**", elem_id="main-title")
         gr.Markdown("**Layer-based object separation and area analysis tool** | 1. Create layers 2. Select point mode and click 3. Run segmentation (processes all layers)")
@@ -545,17 +541,15 @@ with gr.Blocks(css=custom_css) as demo:
             if mgr is None:
                 mgr = LayerManager()
 
-            choices = [(layer['name'], lid) for lid, layer in mgr.layers.items()]
-
             if layer_id:
                 mgr.set_current_layer(layer_id)
-                return mgr, create_layer_status_html(mgr), gr.Dropdown(choices=choices, value=layer_id), f"Layer '{mgr.layers[layer_id]['name']}' selected"
-            return mgr, create_layer_status_html(mgr), gr.Dropdown(choices=choices, value=None), "Please select a layer"
+                return mgr, create_layer_status_html(mgr), f"Layer '{mgr.layers[layer_id]['name']}' selected"
+            return mgr, create_layer_status_html(mgr), "Please select a layer"
 
         layer_selector.change(
             on_layer_select,
             inputs=[layer_selector, st_layer_manager],
-            outputs=[st_layer_manager, layer_buttons_html, layer_selector, status_text]
+            outputs=[st_layer_manager, layer_buttons_html, status_text]
         )
         
         # 포인트 모드 변경
@@ -619,8 +613,8 @@ with gr.Blocks(css=custom_css) as demo:
             new_manager = LayerManager()
             empty_html = "<div style='padding: 10px; text-align: center; color: #888;'>No layers created</div>"
             # 업로드한 이미지를 출력에도 표시
-            return new_manager, img, pd.DataFrame(), empty_html, gr.Dropdown(choices=[], interactive=True, value=None), "positive", "Include Point (Red)", "New image uploaded"
-        
+            return new_manager, img, pd.DataFrame(), empty_html, update_layer_selector_choices(new_manager), "positive", "Include Point (Red)", "New image uploaded"
+
         img_input.change(
             on_image_upload,
             inputs=[img_input],
@@ -629,10 +623,18 @@ with gr.Blocks(css=custom_css) as demo:
 
 @spaces.GPU
 def run_spaces():
-    demo.launch()
+    demo.launch(ssr=False, css="""
+#col-container { margin: 0 auto; max-width: 1200px; }
+#main-title h1 { font-size: 2.1em !important; }
+.layer-button { margin: 2px; }
+""")
 
 def run_local():
-    demo.launch(show_error=True)
+    demo.launch(show_error=True, ssr=False, css="""
+#col-container { margin: 0 auto; max-width: 1200px; }
+#main-title h1 { font-size: 2.1em !important; }
+.layer-button { margin: 2px; }
+""")
 
 if __name__ == "__main__":
     # Hugging Spaces 환경 감지