Upload app.py

app.py

@@ -490,7 +490,7 @@ with gr.Blocks(css=custom_css) as demo:
                 layer_buttons_html = gr.HTML("<div style='padding: 10px; text-align: center; color: #888;'>No layers created</div>")
                 
                 # 레이어 선택
-                layer_selector = gr.Dropdown(label="Select Layer to Add Points", choices=[], interactive=True)
+                layer_selector = gr.Dropdown(label="Select Layer to Add Points", choices=[], interactive=True, value=None)
                 
                 # 포인트 모드 선택
                 gr.Markdown("### Point Mode")
@@ -544,16 +544,18 @@ with gr.Blocks(css=custom_css) as demo:
         def on_layer_select(layer_id, mgr):
             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), f"Layer '{mgr.layers[layer_id]['name']}' selected"
-            return mgr, create_layer_status_html(mgr), "Please select a layer"
-        
+                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"
+
         layer_selector.change(
             on_layer_select,
             inputs=[layer_selector, st_layer_manager],
-            outputs=[st_layer_manager, layer_buttons_html, status_text]
+            outputs=[st_layer_manager, layer_buttons_html, layer_selector, status_text]
         )
         
         # 포인트 모드 변경
@@ -617,7 +619,7 @@ 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=[], value=None), "positive", "Include Point (Red)", "New image uploaded"
+            return new_manager, img, pd.DataFrame(), empty_html, gr.Dropdown(choices=[], interactive=True, value=None), "positive", "Include Point (Red)", "New image uploaded"
         
         img_input.change(
             on_image_upload,
@@ -625,5 +627,17 @@ with gr.Blocks(css=custom_css) as demo:
             outputs=[st_layer_manager, img_output, area_table, layer_buttons_html, layer_selector, point_mode_state, point_mode_text, status_text]
         )
 
-if __name__ == "__main__":
+@spaces.GPU
+def run_spaces():
+    demo.launch()
+
+def run_local():
     demo.launch(show_error=True)
+
+if __name__ == "__main__":
+    # Hugging Spaces 환경 감지
+    import os
+    if os.getenv("SPACE_ID"):
+        run_spaces()
+    else:
+        run_local()