Update app.py

app.py

@@ -157,12 +157,20 @@ 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"])
+    colormap = cm.LinearColormap(colors=vis_params["palette"], vmin=vis_params["min"], vmax=vis_params["max"])
     #dem_map.add_child(colormap)
-    dem_map.add_colormap(width=4,vmin=min_value, vmax=max_value, palette=vis_params['palette'],
-                         vis_params=vis_params, label='Elevation(m)',
-                         discrete=False, tick_size=10, bg_color='white',
-                         transparent=True, orientation='vertical')
+    #dem_map.add_colormap(width=4,vmin=min_value, vmax=max_value, palette=vis_params['palette'],
+    #                     vis_params=vis_params, label='Elevation(m)',
+    #                     discrete=False, tick_size=10, bg_color='white',
+    #                     transparent=True, orientation='horizontal')
+    tick_size = int((max_value-min_value)/4)
+    dem_map.add_legend(title="Elevation (m)",
+                                legend_dict={'{}-{} m'.format(min_value, min_value+tick_size): cmap(min_value),
+                                             '{}-{} m'.format(min_value+tick_size, min_value+2*tick_size): cmap(min_value+tick_size),
+                                             '{}-{} m'.format(min_value+2*tick_size, min_value+3*tick_size): cmap(min_value+2*tick_size),
+                                             '{}-{} m'.format(min_value+3*tick_size, max_value): cmap(min_value)},
+                                position='bottomright',
+                                font_size=12)
 
     # Create the map for Slope
     slope_map = gee_folium.Map()