Update pages/12_🌲_VertXtractor.py

pages/12_🌲_VertXtractor.py

@@ -4,9 +4,8 @@ import folium
 from streamlit_folium import folium_static, st_folium
 from folium.plugins import Draw
 import requests
-from io import BytesIO
-from PIL import Image
 from shapely.geometry import box
+import json
 
 # Liste des couches disponibles
 LAYERS = {
@@ -30,20 +29,12 @@ def draw_box_on_map():
         edit_options={'edit': False}
     )
     draw.add_to(m)
-
-    output = st_folium(m, width=700, height=500)
-
-    if output['last_active_drawing']:
-        coords = output['last_active_drawing']['geometry']['coordinates'][0]
-        return box(min(c[0] for c in coords), min(c[1] for c in coords),
-                   max(c[0] for c in coords), max(c[1] for c in coords))
-    return None
+    return st_folium(m, width=700, height=500)
 
 def create_geojson_from_box(bbox):
     gdf = gpd.GeoDataFrame({'geometry': [bbox]}, crs="EPSG:4326")
     gdf_swiss = gdf.to_crs("EPSG:2056")
-    gdf['geometry_swiss'] = gdf_swiss['geometry']
-    return gdf
+    return json.loads(gdf.to_json()), json.loads(gdf_swiss.to_json())
 
 def extract_swisstopo_data(bbox, layer_id):
     api_url = f"https://data.geo.admin.ch/api/stac/v0.9/collections/{layer_id}/items"
@@ -65,51 +56,60 @@ def extract_esri_data(bbox, layer_id):
 def main():
     st.title("Extracteur de données Swisstopo et ESRI")
 
-    # Étape 1: Dessiner la boîte de sélection
-    st.subheader("Étape 1: Dessiner la zone d'intérêt")
-    bbox = draw_box_on_map()
-
-    if bbox:
-        st.success("Zone sélectionnée avec succès!")
-        gdf = create_geojson_from_box(bbox)
-
-        # Affichage de la carte avec la boîte dessinée
-        m = folium.Map()
-        folium.GeoJson(gdf).add_to(m)
-        m.fit_bounds(m.get_bounds())
-        folium_static(m)
-
-        # Étape 2: Sélection des couches
-        st.subheader("Étape 2: Sélection des couches")
+    col1, col2 = st.columns([2, 1])
+
+    with col1:
+        st.subheader("Étape 1: Définir la zone d'intérêt")
+        method = st.radio("Choisissez une méthode", ["Dessiner sur la carte", "Charger un GeoJSON"])
+
+        if method == "Dessiner sur la carte":
+            map_data = draw_box_on_map()
+            if map_data['last_active_drawing']:
+                coords = map_data['last_active_drawing']['geometry']['coordinates'][0]
+                bbox = box(min(c[0] for c in coords), min(c[1] for c in coords),
+                           max(c[0] for c in coords), max(c[1] for c in coords))
+                geojson_wgs84, geojson_swiss = create_geojson_from_box(bbox)
+                st.session_state['geojson_wgs84'] = geojson_wgs84
+                st.session_state['geojson_swiss'] = geojson_swiss
+                st.success("Zone sélectionnée avec succès!")
+
+        else:
+            uploaded_file = st.file_uploader("Chargez votre GeoJSON", type="geojson")
+            if uploaded_file:
+                gdf = gpd.read_file(uploaded_file)
+                st.session_state['geojson_wgs84'] = json.loads(gdf.to_crs(4326).to_json())
+                st.session_state['geojson_swiss'] = json.loads(gdf.to_crs(2056).to_json())
+                st.success("GeoJSON chargé avec succès!")
+
+    with col2:
+        st.subheader("Étape 2: Sélectionner les couches")
         selected_layers = st.multiselect("Sélectionnez les couches à extraire", list(LAYERS.keys()))
 
         if st.button("Extraire les données"):
-            for layer_name in selected_layers:
-                layer_info = LAYERS[layer_name]
-                
-                # Choix des coordonnées en fonction de la source
-                if layer_info["source"] == "swisstopo":
-                    bbox = gdf['geometry_swiss'].total_bounds
-                else:  # ESRI
-                    bbox = gdf['geometry'].total_bounds
-
-                st.write(f"Extraction de {layer_name}...")
-                
-                if layer_info["source"] == "swisstopo":
-                    data = extract_swisstopo_data(bbox, layer_info["id"])
-                else:
-                    data = extract_esri_data(bbox, layer_info["id"])
-
-                if data is not None:
-                    st.write(data.head())
-                    st.download_button(
-                        label=f"Télécharger {layer_name} (GeoJSON)",
-                        data=data.to_json(),
-                        file_name=f"{layer_name.lower().replace(' ', '_')}.geojson",
-                        mime="application/json"
-                    )
-                else:
-                    st.warning(f"Aucune donnée trouvée pour {layer_name}")
+            if 'geojson_wgs84' not in st.session_state or 'geojson_swiss' not in st.session_state:
+                st.error("Veuillez d'abord définir une zone d'intérêt.")
+            else:
+                for layer_name in selected_layers:
+                    layer_info = LAYERS[layer_name]
+                    
+                    if layer_info["source"] == "swisstopo":
+                        bbox = gpd.GeoDataFrame.from_features(st.session_state['geojson_swiss']).total_bounds
+                        data = extract_swisstopo_data(bbox, layer_info["id"])
+                    else:  # ESRI
+                        bbox = gpd.GeoDataFrame.from_features(st.session_state['geojson_wgs84']).total_bounds
+                        data = extract_esri_data(bbox, layer_info["id"])
+
+                    if data is not None:
+                        st.write(f"Données extraites pour {layer_name}:")
+                        st.write(data.head())
+                        st.download_button(
+                            label=f"Télécharger {layer_name} (GeoJSON)",
+                            data=data.to_json(),
+                            file_name=f"{layer_name.lower().replace(' ', '_')}.geojson",
+                            mime="application/json"
+                        )
+                    else:
+                        st.warning(f"Aucune donnée trouvée pour {layer_name}")
 
     st.sidebar.markdown("## À propos")
     st.sidebar.markdown("Cet outil permet d'extraire des données géospatiales de Swisstopo et ESRI.")