# !pip install leafmap

# !pip install rio-cogeo

import leafmap

url = "https://github.com/opengeos/leafmap/raw/master/examples/data/cog.tif"
in_cog = "cog.tif"
out_cog = "ndvi.tif"

leafmap.download_from_url(url, in_cog)

arr = leafmap.image_to_numpy(in_cog)

arr.shape

ndvi = (arr[3] - arr[0]) / (arr[3] + arr[0])

ndvi.shape

leafmap.numpy_to_cog(ndvi, out_cog, profile=in_cog)

m = leafmap.Map()
m.add_raster(in_cog, band=[4, 1, 2], layer_name="Color infrared")
m.add_raster(out_cog, palette="Greens", layer_name="NDVI")
m