# !pip install leafmap

import folium
import rioxarray
import xarray as xr
import leafmap.foliumap as leafmap

m = leafmap.Map(height=500)
m.split_map(left_layer="TERRAIN", right_layer="OpenTopoMap")
m

# leafmap.basemaps.keys()

m = leafmap.Map(height=600, center=[39.4948, -108.5492], zoom=12)
url = "https://github.com/opengeos/data/releases/download/raster/Libya-2023-07-01.tif"
url2 = "https://github.com/opengeos/data/releases/download/raster/Libya-2023-09-13.tif"
m.split_map(url, url2)
m

m = leafmap.Map(center=[40, -100], zoom=4)

url1 = "https://www.mrlc.gov/geoserver/mrlc_display/NLCD_2001_Land_Cover_L48/wms?"
url2 = "https://www.mrlc.gov/geoserver/mrlc_display/NLCD_2019_Land_Cover_L48/wms?"

left_layer = folium.WmsTileLayer(
    url=url1,
    layers="NLCD_2001_Land_Cover_L48",
    name="NLCD 2001",
    attr="MRLC",
    fmt="image/png",
    transparent=True,
)
right_layer = folium.WmsTileLayer(
    url=url2,
    layers="NLCD_2019_Land_Cover_L48",
    name="NLCD 2019",
    attr="MRLC",
    fmt="image/png",
    transparent=True,
)

m.split_map(left_layer, right_layer)
m

url = "https://opengeos.org/data/raster/srtm90.tif"
dem = leafmap.download_file(url, "srtm90.tif")

dem_ds = rioxarray.open_rasterio(dem)
dem_class = xr.where(dem_ds < 2000, 0, 1)

m = leafmap.Map(center=[37.6, -119], zoom=9)
m.split_map(
    dem_ds,
    dem_class,
    left_args={"layer_name": "DEM", "colormap": "terrain"},
    right_args={"layer_name": "Classified DEM", "colormap": "coolwarm"},
)
m