Update pages/12_🌲_VertXtractor.py

pages/12_🌲_VertXtractor.py

@@ -13,6 +13,7 @@ import numpy as np
 # Liste des couches de données Swisstopo
 LAYERS = {
     "Bâtiments": "ch.swisstopo.vec25-gebaeude",
+    "Bâtiments 3D": "ch.swisstopo.swissbuildings3d_2",
     "Réseau hydrographique": "ch.swisstopo.vec25-gewaessernetz",
     "Réseau routier": "ch.swisstopo.vec25-strassen",
     "Couverture du sol": "ch.swisstopo.vec25-landesbedeckung",
@@ -25,7 +26,6 @@ LAYERS = {
     "Noms géographiques": "ch.swisstopo.swissnames3d",
 }
 
-# Fonction pour extraire les données Swisstopo
 def extract_swisstopo_data(min_x, min_y, max_x, max_y, layer):
     bbox = box(min_x, min_y, max_x, max_y)
     api_url = "https://api3.geo.admin.ch/rest/services/api/MapServer/identify"
@@ -50,9 +50,30 @@ def extract_swisstopo_data(min_x, min_y, max_x, max_y, layer):
     gdf = gpd.GeoDataFrame.from_features(features, crs="EPSG:2056")
     return gdf
 
-# Fonction pour télécharger l'image SWISSIMAGE (inchangée)
 def download_swissimage(min_x, min_y, max_x, max_y):
-    # ... (le code reste le même)
+    api_url = "https://data.geo.admin.ch/api/stac/v0.9/collections/ch.swisstopo.swissimage-dop10/items"
+    
+    params = {
+        "bbox": f"{min_x},{min_y},{max_x},{max_y}",
+        "datetime": "2022-01-01T00:00:00Z/..",
+        "limit": 1
+    }
+    
+    response = requests.get(api_url, params=params)
+    response.raise_for_status()
+    
+    data = response.json()
+    if data['features']:
+        asset_url = data['features'][0]['assets']['rgb']['href']
+        image_response = requests.get(asset_url)
+        image_response.raise_for_status()
+        
+        with zipfile.ZipFile(BytesIO(image_response.content)) as zf:
+            image_file = [f for f in zf.namelist() if f.endswith('.tif')][0]
+            with zf.open(image_file) as file:
+                image = Image.open(file)
+                return np.array(image)
+    return None
 
 # Interface utilisateur Streamlit
 st.title("Extracteur de données Swisstopo")