Fix NDVI Legend

app.py

@@ -159,17 +159,12 @@ def get_dem_slope_maps(buffer_ee_geometry):
     dem_map.addLayer(contours, vis_params, "Contours")
     # Create a colormap
     cmap = 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='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): '#0000FF',
                                              '{}-{} m'.format(min_value+tick_size, min_value+2*tick_size): '#00FF00',
                                              '{}-{} m'.format(min_value+2*tick_size, min_value+3*tick_size): '#FFFF00',
-                                             '{}-{} m'.format(min_value+3*tick_size, max_value): 'FF0000'},
+                                             '{}-{} m'.format(min_value+3*tick_size, max_value): '#FF0000'},
                                 position='bottomright')
 
     # Create the map for Slope
@@ -194,13 +189,12 @@ def get_dem_slope_maps(buffer_ee_geometry):
     slope_map.addLayer(slope_layer, vis_params, "Slope Layer")
     # Create a colormap
     colormap = cm.LinearColormap(colors=vis_params["palette"], vmin=vis_params["min"], vmax=vis_params["max"])
-    #slope_map.add_child(colormap)
     tick_size=int((max_value-min_value)/4)
     slope_map.add_legend(title="Slope (degrees)",
                                 legend_dict={'{}-{} deg'.format(min_value, min_value+tick_size): '#0000FF',
                                              '{}-{} deg'.format(min_value+tick_size, min_value+2*tick_size): '#00FF00',
                                              '{}-{} deg'.format(min_value+2*tick_size, min_value+3*tick_size): '#FFFF00',
-                                             '{}-{} deg'.format(min_value+3*tick_size, max_value): 'FF0000'},
+                                             '{}-{} deg'.format(min_value+3*tick_size, max_value): '#FF0000'},
                                 position='bottomright')
     return dem_map, slope_map
 
@@ -529,6 +523,18 @@ if "result" in st.session_state:
                 else:
                     raise ValueError(f"Unknown satellite: {satellite}")
 
+                if veg_index=='NDVI':
+                    m.add_legend(title="NDVI Class/Value",
+                                legend_dict={'<0:Waterbody': '#0000FF',
+                                             '0-0.1: Open': '#FF0000',
+                                             '0.1-0.2: Highly Degraded':'#FFFF00',
+                                             '0.2-0.3: Degraded': '#FFA500',
+                                             '0.3-0.4: Moderately Degraded': '#00FE00',
+                                             '0.4-0.5: Dense': '#00A400',
+                                             '>0.5: Very Dense': '#006D00',
+                                            },
+                                position='bottomright')
+                
                 m.add_layer(mosaic.select(veg_index), vis_params)
                 # add colorbar
                 # m.add_colorbar(colors=["#000000", "#00FF00"], vmin=0.0, vmax=1.0)