Update app.py

app.py

@@ -157,8 +157,9 @@ def get_dem_slope_maps(buffer_ee_geometry):
     vis_params = {"min": min_value, "max": max_value, "palette": ["blue", "green", "yellow", "red"]}
     dem_map.addLayer(contours, vis_params, "Contours")
     # Create a colormap
-    colormap = cm.LinearColormap(colors=vis_params["palette"], vmin=vis_params["min"], vmax=vis_params["max"])
-    dem_map.add_child(colormap)
+    #colormap = cm.LinearColormap(colors=vis_params["palette"], vmin=vis_params["min"], vmax=vis_params["max"])
+    #dem_map.add_child(colormap)
+    dem_map.add_colormap(vis_params=vis_params, discrete=True, tick_size=12, bg_color='white')
 
     # Create the map for Slope
     slope_map = gee_folium.Map()
@@ -569,5 +570,5 @@ if "result" in st.session_state:
                 style_function=lambda x: {"color": "blue", "fillOpacity": 0.0, "fillColor": "blue"},
             )
             write_info(f"""<div style="text-align: center;">{title}</div>""")
-            param_map.add_scale()
+            param_map.add_layer_control()
             param_map.to_streamlit()