# !pip install -U geemap gradio

import ee
import gradio as gr
import geemap.foliumap as geemap

Map = geemap.Map(center=[21.79, 70.87], zoom=3)
image = ee.Image('USGS/SRTMGL1_003')
vis_params = {
    'min': 0,
    'max': 6000,
    'palette': 'terrain',
}
Map.addLayer(image, vis_params, 'SRTM')
Map

def viz_dem(vmin, vmax, palette):
    Map = geemap.Map()
    image = ee.Image('USGS/SRTMGL1_003')
    vis_params = {
        'min': vmin,
        'max': vmax,
        'palette': palette,
    }
    Map.addLayer(image, vis_params, 'SRTM')
    return Map.to_gradio()


vmin = gr.Number(value=0, label="Min value")
vmax = gr.Number(value=6000, label="Max value")
palette = gr.Textbox(value="terrain", label="Palette")
title = 'Visualize Earth Engine Data'
demo = gr.Interface(viz_dem, [vmin, vmax, palette], "html", title=title)
demo.launch()

def split(left, right):
    Map = geemap.Map(center=(40, -100), zoom=4, height=600)

    nlcd_left = ee.Image(f'USGS/NLCD_RELEASES/2019_REL/NLCD/{left}').select('landcover')
    nlcd_right = ee.Image(f'USGS/NLCD_RELEASES/2019_REL/NLCD/{right}').select(
        'landcover'
    )

    left_layer = geemap.ee_tile_layer(nlcd_left, {}, f'NLCD {left}')
    right_layer = geemap.ee_tile_layer(nlcd_right, {}, f'NLCD {right}')

    Map.split_map(
        left_layer,
        right_layer,
    )
    return Map.to_gradio()


left_input = gr.Textbox(value='2001', label="Left Layer URL")
right_input = gr.Textbox(value='2019', label="Right Layer URL")

title = 'Visualizing National Land Cover Database (NLCD)'
demo = gr.Interface(split, [left_input, right_input], "html", title=title)
demo.launch()

def split(left, right):
    Map = geemap.Map(center=[21.79, 70.87], zoom=3)
    Map.split_map(left, right)
    print(Map.options["layersControl"])
    return Map.to_gradio()


left_url = 'https://opendata.digitalglobe.com/events/california-fire-2020/pre-event/2018-02-16/pine-gulch-fire20/1030010076004E00.tif'
right_url = 'https://opendata.digitalglobe.com/events/california-fire-2020/post-event/2020-08-14/pine-gulch-fire20/10300100AAC8DD00.tif'
left_input = gr.Textbox(value=left_url, label="Left Layer URL")
right_input = gr.Textbox(value=right_url, label="Right Layer URL")
title = 'Visualze Cloud Optimized GeoTIFF (COG)'
demo = gr.Interface(split, [left_input, right_input], "html", title=title)
demo.launch()