Update app.py

app.py

@@ -118,6 +118,7 @@ def get_dem_slope_maps(buffer_ee_geometry):
     # Create a colormap
     colormap = cm.LinearColormap(colors=vis_params['palette'], vmin=vis_params['min'], vmax=vis_params['max'])
     dem_map.add_child(colormap)
+    dem_map.addLayer(buffer_ee_geometry, )
     
     # Create the map for Slope
     slope_map = gee_folium.Map()
@@ -282,13 +283,14 @@ def format_fn(x):
 input_geometry_idx = st.selectbox("Select the geometry", input_gdf.index, format_func=format_fn)
 geometry_gdf = input_gdf[input_gdf.index == input_geometry_idx]
 buffer_geometry_gdf = geometry_gdf.copy()
-buffer_geometry_gdf["geometry"] = buffer_geometry_gdf["geometry"].buffer(buffer).difference(geometry_gdf.geometry.unary_union)
+buffer_geometry_gdf["geometry"] = buffer_geometry_gdf["geometry"].buffer(buffer)
 check_valid_geometry(geometry_gdf)
 
 # Derived Inputs
 ee_geometry = ee.Geometry(geometry_gdf.to_crs(4326).geometry.item().__geo_interface__)
 ee_feature_collection = ee.FeatureCollection(ee_geometry)
 buffer_ee_geometry = ee.Geometry(buffer_geometry_gdf.to_crs(4326).geometry.item().__geo_interface__)
+buffer_ee_geometry = buffer_ee_geometry.difference(ee_geometry)
 buffer_ee_feature_collection = ee.FeatureCollection(buffer_ee_geometry)
 
 # visualize the geometry
@@ -466,12 +468,12 @@ if "result" in st.session_state:
 
     st.write("<h3><div style='text-align: center;'>DEM and Slope from SRTM at 30m resolution</div></h3>", unsafe_allow_html=True)
     dem_col, slope_col = st.columns(2)
-    ee_geometry = ee.Geometry(geometry_gdf.to_crs(4326).geometry.item().__geo_interface__)
-    dem_map, slope_map = get_dem_slope_maps(ee_geometry.buffer(buffer))
+    dem_map, slope_map = get_dem_slope_maps(ee.Geometry(buffer_geometry_gdf.to_crs(4326).geometry.item().__geo_interface__))
     # DEM Map
     dem_col.write("DEM Map")
     add_geometry_to_maps([dem_map])
     dem_map.to_streamlit()
+    
     # Slope Map
     slope_col.write("Slope Map")
     slope_map.to_streamlit()