import ee
import geemap

import solara


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

    def add_ee_data(self):
        # Select the eight NLCD epochs after 2000.
        years = ["2001", "2004", "2006", "2008", "2011", "2013", "2016", "2019"]

        # Get an NLCD image by year.
        def getNLCD(year):
            # Import the NLCD collection.
            dataset = ee.ImageCollection("USGS/NLCD_RELEASES/2019_REL/NLCD")

            # Filter the collection by year.
            nlcd = dataset.filter(ee.Filter.eq("system:index", year)).first()

            # Select the land cover band.
            landcover = nlcd.select("landcover")
            return landcover

        ## Create an NLCD image collection for the selected years.
        collection = ee.ImageCollection(ee.List(years).map(lambda year: getNLCD(year)))

        # Create a list of labels to populate the dropdown list.
        labels = [f"NLCD {year}" for year in years]

        # Add a split-panel map for visualizing NLCD land cover change.
        self.ts_inspector(
            left_ts=collection,
            right_ts=collection,
            left_names=labels,
            right_names=labels,
        )

        # Add the NLCD legend to the map.
        self.add_legend(
            title="NLCD Land Cover Type",
            builtin_legend="NLCD",
            height="460px",
            add_header=False,
        )


@solara.component
def Page():
    with solara.Column(style={"min-width": "500px"}):
        Map.element(
            center=[40, -100],
            zoom=4,
            height="600px",
        )