Update pages/12_🌲_VertXtractor.py

pages/12_🌲_VertXtractor.py

@@ -364,10 +364,16 @@ with col3:
 with col4:
     ymax = st.number_input("Max Latitude", value=46.2500, step=0.0001, format="%.4f")
 
+if 'bbox' not in st.session_state:
+    st.session_state.bbox = None
+if 'urls' not in st.session_state:
+    st.session_state.urls = None
+
 if st.button("Set Bounding Box"):
     st.session_state.bbox = [xmin, ymin, xmax, ymax]
+    st.session_state.urls = None  # Reset urls when bbox changes
 
-if 'bbox' in st.session_state:
+if st.session_state.bbox:
     st.write(f"Selected bounding box (WGS84): {st.session_state.bbox}")
     
     # Convert bbox to LV95
@@ -379,33 +385,33 @@ if 'bbox' in st.session_state:
         
         if st.button("Get Download Links"):
             with st.spinner("Fetching download links..."):
-                urls = get_urls(bbox_wgs84, mnt, mns, bati3D_v2, bati3D_v3, ortho, mnt_resol, ortho_resol)
+                st.session_state.urls = get_urls(bbox_wgs84, mnt, mns, bati3D_v2, bati3D_v3, ortho, mnt_resol, ortho_resol)
                 
-                if urls:
-                    st.success(f"Found {len(urls)} files to download:")
-                    for url in urls:
-                        st.write(url)
-                    
-                    # Option to download individual files
-                    if st.button("Download Individual Files"):
-                        with st.spinner("Downloading files..."):
-                            download_path = download_files(urls, "downloads")
-                            st.success(f"Files downloaded to: {download_path}")
-                    
-                    # Option to download and merge ortho images
-                    ortho_urls = [url for url in urls if 'swissimage-dop10' in url]
-                    if ortho_urls and st.button("Download and Merge Ortho Images"):
-                        with st.spinner("Downloading and merging ortho images..."):
-                            output_path = "merged_ortho.tif"
-                            merged_path = download_and_merge_ortho(ortho_urls, output_path)
-                            st.success(f"Merged ortho image saved to: {merged_path}")
-                            
-                            # Option to display the merged image
-                            if st.button("Display Merged Image"):
-                                image = Image.open(merged_path)
-                                st.image(image, caption="Merged Ortho Image", use_column_width=True)
-                else:
-                    st.warning("No files found for the selected area and options.")
+        if st.session_state.urls:
+            st.success(f"Found {len(st.session_state.urls)} files to download:")
+            for url in st.session_state.urls:
+                st.write(url)
+            
+            # Option to download individual files
+            if st.button("Download Individual Files"):
+                with st.spinner("Downloading files..."):
+                    download_path = download_files(st.session_state.urls, "downloads")
+                    st.success(f"Files downloaded to: {download_path}")
+            
+            # Option to download and merge ortho images
+            ortho_urls = [url for url in st.session_state.urls if 'swissimage-dop10' in url]
+            if ortho_urls:
+                if st.button("Download and Merge Ortho Images"):
+                    with st.spinner("Downloading and merging ortho images..."):
+                        output_path = "merged_ortho.tif"
+                        merged_path = download_and_merge_ortho(ortho_urls, output_path)
+                        st.success(f"Merged ortho image saved to: {merged_path}")
+                        
+                        # Display the merged image
+                        image = Image.open(merged_path)
+                        st.image(image, caption="Merged Ortho Image", use_column_width=True)
+        else:
+            st.warning("No files found for the selected area and options.")
         
         # Option to download forest data
         if st.button("Download Forest Data"):