import ee
import geemap

import solara

zoom = solara.reactive(4)
center = solara.reactive([40, -100])


class Map(geemap.Map):
    def __init__(self, **kwargs):
        super().__init__(**kwargs)
        self.add_ee_data()
        self.add_layer_manager()
        # self.add_inspector()

    def add_ee_data(self):
        gswe = ee.ImageCollection("users/h2i_lab/gswe/gswe_datasets")
        self.addLayer(gswe.select("esa"), {'palette': ['red']}, "ESA")
        self.addLayer(gswe.select("osm"), {'palette': ['green']}, "OSM")
        self.addLayer(gswe.select("jrc"), {'palette': ['blue']}, "JRC")
        self.addLayer(gswe.select("esri"), {'palette': ['yellow']}, "ESRI")
        self.addLayer(gswe.select("hydrolakes"), {'palette': ['purple']}, "Hydrolakes")


@solara.component
def Page():
    with solara.Column(style={"min-width": "500px"}):
        Map.element(  # type: ignore
            zoom=zoom.value,
            on_zoom=zoom.set,
            center=center.value,
            on_center=center.set,
            scroll_wheel_zoom=True,
            add_google_map=True,
            height="800px",
        )