# !pip install geemap

import ee
import geemap
from geemap.algorithms import river

Map = geemap.Map()
Map

point = ee.Geometry.Point([-88.08, 37.47])

image = (
    ee.ImageCollection("LANDSAT/LC08/C01/T1_SR")
    .filterBounds(point)
    .sort("CLOUD_COVER")
    .first()
)

Map.addLayer(image, {'min': 0, 'max': 3000, 'bands': ['B5', 'B4', 'B3']}, "Landsat")
Map.centerObject(image)

river.rwc(image, folder="export", water_method='Jones2019')

fc = ee.FeatureCollection("users/giswqs/public/river_width")
Map.addLayer(fc, {}, "River width")

water_mask = ee.ImageCollection(
    "projects/sat-io/open-datasets/GRWL/water_mask_v01_01"
).median()
Map.addLayer(water_mask, {'min': 11, 'max': 125, 'palette': 'blue'}, 'GRWL Water Mask')

grwl_water_vector = ee.FeatureCollection(
    "projects/sat-io/open-datasets/GRWL/water_vector_v01_01"
)
Map.addLayer(
    grwl_water_vector.style(**{'fillColor': '00000000', 'color': 'FF5500'}),
    {},
    'GRWL Vector',
)

images = (
    ee.ImageCollection("LANDSAT/LC08/C01/T1_SR")
    .filterBounds(point)
    .sort("CLOUD_COVER")
    .limit(3)
)

ids = images.aggregate_array("LANDSAT_ID").getInfo()
ids

river.rwc_batch(ids, folder="export", water_method='Jones2019')