import ee
import geemap

import solara


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

    def add_ee_data(self):
        landsat7 = ee.Image('LANDSAT/LE7_TOA_5YEAR/1999_2003').select(
            ['B1', 'B2', 'B3', 'B4', 'B5', 'B7']
        )

        landsat_vis = {'bands': ['B4', 'B3', 'B2'], 'gamma': 1.4}
        self.addLayer(landsat7, landsat_vis, "Landsat")

        hyperion = ee.ImageCollection('EO1/HYPERION').filter(
            ee.Filter.date('2016-01-01', '2017-03-01')
        )

        hyperion_vis = {
            'min': 1000.0,
            'max': 14000.0,
            'gamma': 2.5,
        }
        self.addLayer(hyperion, hyperion_vis, 'Hyperion')


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