# !pip install geemap

import os
import ee
import geemap
import ipywidgets as widgets

Map = geemap.Map()
Map.add_basemap('HYBRID')
Map

style = {'description_width': 'initial'}
title = widgets.Text(
    description='Title:', value='Landsat Timelapse', width=200, style=style
)

bands = widgets.Dropdown(
    description='Select RGB Combo:',
    options=[
        'Red/Green/Blue',
        'NIR/Red/Green',
        'SWIR2/SWIR1/NIR',
        'NIR/SWIR1/Red',
        'SWIR2/NIR/Red',
        'SWIR2/SWIR1/Red',
        'SWIR1/NIR/Blue',
        'NIR/SWIR1/Blue',
        'SWIR2/NIR/Green',
        'SWIR1/NIR/Red',
    ],
    value='NIR/Red/Green',
    style=style,
)

hbox1 = widgets.HBox([title, bands])
hbox1

speed = widgets.IntSlider(
    description='  Frames per second:',
    tooltip='Frames per second:',
    value=10,
    min=1,
    max=30,
    style=style,
)

cloud = widgets.Checkbox(
    value=True, description='Apply fmask (remove clouds, shadows, snow)', style=style
)

hbox2 = widgets.HBox([speed, cloud])
hbox2

start_year = widgets.IntSlider(
    description='Start Year:', value=1984, min=1984, max=2020, style=style
)
end_year = widgets.IntSlider(
    description='End Year:', value=2020, min=1984, max=2020, style=style
)
start_month = widgets.IntSlider(
    description='Start Month:', value=5, min=1, max=12, style=style
)
end_month = widgets.IntSlider(
    description='End Month:', value=10, min=1, max=12, style=style
)
hbox3 = widgets.HBox([start_year, end_year, start_month, end_month])
hbox3

font_size = widgets.IntSlider(
    description='Font size:', value=30, min=10, max=50, style=style
)

font_color = widgets.ColorPicker(
    concise=False, description='Font color:', value='white', style=style
)

progress_bar_color = widgets.ColorPicker(
    concise=False, description='Progress bar color:', value='blue', style=style
)

hbox4 = widgets.HBox([font_size, font_color, progress_bar_color])
hbox4

create_gif = widgets.Button(
    description='Create timelapse',
    button_style='primary',
    tooltip='Click to create timelapse',
    style=style,
)

download_gif = widgets.Button(
    description='Download GIF',
    button_style='primary',
    tooltip='Click to download timelapse',
    disabled=False,
    style=style,
)

output = widgets.Output()

hbox5 = widgets.HBox([create_gif])
hbox5

def submit_clicked(b):
    with output:
        output.clear_output()
        if start_year.value > end_year.value:
            print('The end year must be great than the start year.')
            return
        if start_month.value > end_month.value:
            print('The end month must be great than the start month.')
            return
        if start_year.value == end_year.value:
            add_progress_bar = False
        else:
            add_progress_bar = True

        start_date = str(start_month.value).zfill(2) + '-01'
        end_date = str(end_month.value).zfill(2) + '-30'

        print('Computing...')

        Map.add_landsat_ts_gif(
            roi=Map.user_roi,
            label=title.value,
            start_year=start_year.value,
            end_year=end_year.value,
            start_date=start_date,
            end_date=end_date,
            bands=bands.value.split('/'),
            font_color=font_color.value,
            frames_per_second=speed.value,
            font_size=font_size.value,
            add_progress_bar=add_progress_bar,
            progress_bar_color=progress_bar_color.value,
            download=True,
            apply_fmask=cloud.value,
        )


create_gif.on_click(submit_clicked)

output