Update pages/12_🌲_VertXtractor.py

pages/12_🌲_VertXtractor.py

@@ -6,12 +6,22 @@ import requests
 from io import BytesIO
 from PIL import Image
 import tempfile
-import os
 
 st.set_page_config(layout="wide", page_title="GeoJSON Orthophoto Downloader")
 
 st.title("GeoJSON Orthophoto Downloader")
 
+@st.cache_data
+def get_swisstopo_layers():
+    url = "https://api3.geo.admin.ch/rest/services/api/MapServer"
+    response = requests.get(url)
+    if response.status_code == 200:
+        layers = response.json()
+        return {layer['layerBodId']: layer['layerName'] for layer in layers['layers']}
+    return {}
+
+swisstopo_layers = get_swisstopo_layers()
+
 # File uploader for GeoJSON
 uploaded_file = st.file_uploader("Choose a GeoJSON file", type="geojson")
 
@@ -37,6 +47,9 @@ if uploaded_file is not None:
     # Choose the map provider
     provider = st.selectbox("Choose map provider", ["ESRI World Imagery", "Swisstopo"])
     
+    if provider == "Swisstopo":
+        selected_layer = st.selectbox("Choose Swisstopo layer", list(swisstopo_layers.keys()), format_func=lambda x: swisstopo_layers[x])
+    
     if st.button("Download Orthophoto"):
         if provider == "ESRI World Imagery":
             # ESRI World Imagery WMS URL
@@ -55,17 +68,31 @@ if uploaded_file is not None:
             # Make the request
             response = requests.get(wms_url, params=params)
             
-            if response.status_code == 200:
-                # Save the image
-                image = Image.open(BytesIO(response.content))
-                with tempfile.NamedTemporaryFile(delete=False, suffix=".png") as tmp_file:
-                    image.save(tmp_file.name)
-                    st.success(f"Image saved as {tmp_file.name}")
-            else:
-                st.error("Failed to download the image")
-        
         elif provider == "Swisstopo":
-            st.warning("Swisstopo implementation is not available in this demo. You would need to use Swisstopo's specific API or services.")
+            # Swisstopo MapServer URL
+            mapserver_url = f"https://api3.geo.admin.ch/rest/services/api/MapServer/{selected_layer}/image"
+            
+            # Parameters for the MapServer request
+            params = {
+                'bbox': f"{minx},{miny},{maxx},{maxy}",
+                'imageSize': '1000,1000',
+                'format': 'png'
+            }
+            
+            # Make the request
+            response = requests.get(mapserver_url, params=params)
+        
+        if response.status_code == 200:
+            # Save the image
+            image = Image.open(BytesIO(response.content))
+            with tempfile.NamedTemporaryFile(delete=False, suffix=".png") as tmp_file:
+                image.save(tmp_file.name)
+                st.success(f"Image saved as {tmp_file.name}")
+            
+            # Display the downloaded image
+            st.image(image, caption=f"Downloaded {provider} Orthophoto", use_column_width=True)
+        else:
+            st.error(f"Failed to download the image. Status code: {response.status_code}")
 
     st.write("Note: This is a demo application. For production use, please ensure you comply with the usage terms of the chosen map provider.")