Update pages/12_🌲_VertXtractor.py

pages/12_🌲_VertXtractor.py

@@ -7,12 +7,10 @@ import requests
 from shapely.geometry import box
 import json
 
-# Configuration de la page Streamlit
 st.set_page_config(layout="wide", page_title="Extracteur de données géospatiales")
 
-# Définition des couches disponibles
 LAYERS = {
-    # Couches Swisstopo
+    # Swisstopo
     "SWISSIMAGE 10 cm": {"id": "ch.swisstopo.swissimage-dop10", "source": "swisstopo", "type": "image"},
     "Carte nationale 1:25'000": {"id": "ch.swisstopo.pixelkarte-farbe-pk25.noscale", "source": "swisstopo", "type": "image"},
     "MNT": {"id": "ch.swisstopo.swissalti3d", "source": "swisstopo", "type": "raster"},
@@ -20,7 +18,7 @@ LAYERS = {
     "Limites administratives": {"id": "ch.swisstopo.swissboundaries3d-gemeinde-flaeche.fill", "source": "swisstopo", "type": "vector"},
     "Réseau hydrographique": {"id": "ch.bafu.vec25-gewaessernetz", "source": "swisstopo", "type": "vector"},
     "swissBUILDINGS3D 3.0 Beta": {"id": "ch.swisstopo.swissbuildings3d_3_0", "source": "swisstopo", "type": "3d"},
-    # Couches ESRI
+    # ESRI
     "World Imagery": {"id": "World_Imagery", "source": "esri", "type": "image"},
     "World Elevation": {"id": "Elevation/World_Elevation", "source": "esri", "type": "raster"},
     "World Topographic": {"id": "World_Topo_Map", "source": "esri", "type": "image"},
@@ -29,85 +27,46 @@ LAYERS = {
 }
 
 def draw_box_on_map():
-    """Crée une carte Folium avec un outil de dessin pour sélectionner une zone."""
     m = folium.Map(location=[46.8, 8.2], zoom_start=8)
     Draw(draw_options={'polyline': False, 'polygon': False, 'circle': False, 'marker': False, 'circlemarker': False},
          edit_options={'edit': False}).add_to(m)
     return st_folium(m, width=700, height=500)
 
 def create_geojson_from_box(bbox):
-    """Crée des objets GeoJSON à partir d'une boîte englobante."""
     gdf = gpd.GeoDataFrame({'geometry': [bbox]}, crs="EPSG:4326")
     return json.loads(gdf.to_json()), json.loads(gdf.to_crs("EPSG:2056").to_json())
 
 @st.cache_data
 def get_download_url(bbox, layer_info):
-    """Génère l'URL de téléchargement pour une couche spécifique."""
     if layer_info["source"] == "swisstopo":
-        if layer_info["type"] == "3d":
-            # Gestion spéciale pour les données 3D de Swisstopo
-            api_url = "https://data.geo.admin.ch/api/stac/v0.9/collections/ch.swisstopo.swissbuildings3d_3_0/items"
-            params = {
-                "bbox": ",".join(map(str, bbox)),  # Boîte englobante
-                "limit": 1,  # Limite le nombre de résultats
-                "datetime": "2023-01-01/2024-12-31"  # Plage de dates pour les données
-            }
-            response = requests.get(api_url, params=params)
-            data = response.json()
-            if data['features']:
-                feature = data['features'][0]
-                asset_keys = feature['assets'].keys()
-                # Choisit le format de données approprié
-                data_key = 'gltf' if 'gltf' in asset_keys else next(iter(asset_keys))
-                return feature['assets'][data_key]['href']
-        else:
-            # Gestion standard pour les autres données Swisstopo
-            api_url = f"https://data.geo.admin.ch/api/stac/v0.9/collections/{layer_info['id']}/items"
-            params = {"bbox": ",".join(map(str, bbox)), "limit": 1}
-            response = requests.get(api_url, params=params)
-            data = response.json()
-            if data['features']:
-                feature = data['features'][0]
-                asset_keys = feature['assets'].keys()
-                # Choisit le format de données approprié
-                data_key = 'rgb' if 'rgb' in asset_keys else 'data' if 'data' in asset_keys else next(iter(asset_keys))
-                return feature['assets'][data_key]['href']
+        api_url = f"https://data.geo.admin.ch/api/stac/v0.9/collections/{layer_info['id']}/items"
+        params = {"bbox": ",".join(map(str, bbox)), "limit": 1}
+        response = requests.get(api_url, params=params)
+        data = response.json()
+        if data['features']:
+            feature = data['features'][0]
+            asset_keys = feature['assets'].keys()
+            data_key = 'geotiff' if 'geotiff' in asset_keys else 'rgb' if 'rgb' in asset_keys else 'data' if 'data' in asset_keys else next(iter(asset_keys))
+            return feature['assets'][data_key]['href']
     else:  # ESRI
-        # Construction de l'URL pour les services ESRI
-        service_url = f"https://services.arcgisonline.com/arcgis/rest/services/{layer_info['id']}/MapServer/export"
+        service_url = f"https://server.arcgisonline.com/arcgis/rest/services/{layer_info['id']}/ImageServer/exportImage"
         params = {
-            "bbox": f"{bbox[0]},{bbox[1]},{bbox[2]},{bbox[3]}",  # Boîte englobante
-            "bboxSR": 4326,  # Système de référence spatiale de la boîte englobante (WGS84)
-            "size": "1000,1000",  # Taille de l'image en pixels
-            "format": "png",  # Format de l'image
-            "f": "json",  # Format de la réponse (JSON)
-            "imageSR": 4326,  # Système de référence spatiale de l'image
-            "transparent": "true"  # Rend le fond transparent si possible
+            "bbox": f"{bbox[0]},{bbox[1]},{bbox[2]},{bbox[3]}",
+            "bboxSR": 4326,
+            "size": "1000,1000",
+            "format": "tiff",
+            "f": "image",
+            "imageSR": 4326,
+            "pixelType": "U8",
+            "noDataInterpretation": "esriNoDataMatchAny",
+            "interpolation": "+RSP_BilinearInterpolation"
         }
-        try:
-            response = requests.get(service_url, params=params)
-            response.raise_for_status()
-            json_response = response.json()
-            if 'error' in json_response:
-                st.error(f"Erreur ESRI: {json_response['error']['message']}")
-                return None
-            elif 'href' in json_response:
-                image_url = json_response['href']
-                world_file_url = json_response.get('worldFileURL')
-                return {
-                    "image": image_url,
-                    "world_file": world_file_url
-                }
-            else:
-                st.error(f"Erreur ESRI: Réponse inattendue - {json_response}")
-                return None
-        except requests.RequestException as e:
-            st.error(f"Erreur ESRI: {str(e)}")
-            return None
+        url = f"{service_url}?{'&'.join([f'{k}={v}' for k, v in params.items()])}"
+        return url
     return None
 
 def main():
-    st.title("Extracteur de données géospatiales")
+    st.title("Extracteur de rasters géoréférencés")
 
     col1, col2 = st.columns([1, 2])
 
@@ -126,7 +85,7 @@ def main():
                 st.session_state['geojson_swiss'] = json.loads(gdf.to_crs(2056).to_json())
                 st.success("GeoJSON chargé!")
 
-        st.subheader("3. Télécharger les données")
+        st.subheader("3. Télécharger les rasters")
         if st.button("Obtenir les liens de téléchargement"):
             if 'geojson_wgs84' not in st.session_state or 'geojson_swiss' not in st.session_state:
                 st.error("Définissez d'abord une zone d'intérêt.")
@@ -140,12 +99,7 @@ def main():
                     st.write(f"Bbox : {bbox}")
                     download_url = get_download_url(bbox, layer_info)
                     if download_url:
-                        if isinstance(download_url, dict):  # Pour les services ESRI
-                            st.markdown(f"[Télécharger {layer_name} (Image)]({download_url['image']})")
-                            if download_url['world_file']:
-                                st.markdown(f"[Télécharger {layer_name} (Fichier World)]({download_url['world_file']})")
-                        else:  # Pour Swisstopo
-                            st.markdown(f"[Télécharger {layer_name}]({download_url})")
+                        st.markdown(f"[Télécharger {layer_name} (Raster géoréférencé)]({download_url})")
                     else:
                         st.error(f"Impossible d'obtenir le lien pour {layer_name}")
 
@@ -159,7 +113,7 @@ def main():
                 st.session_state['geojson_wgs84'], st.session_state['geojson_swiss'] = create_geojson_from_box(bbox)
                 st.success("Zone sélectionnée!")
 
-    st.sidebar.markdown("## À propos\nExtracteur de données géospatiales Swisstopo et ESRI.\nDéveloppé par Vertdure")
+    st.sidebar.markdown("## À propos\nExtracteur de rasters géoréférencés Swisstopo et ESRI.\nDéveloppé par Vertdure")
 
 if __name__ == "__main__":
     main()