# !pip install geemap

# Installs geemap package
import subprocess

try:
    import geemap
except ImportError:
    print('geemap package not installed. Installing ...')
    subprocess.check_call(["python", '-m', 'pip', 'install', 'geemap'])

import ee

import geemap
from ipyleaflet import *
from ipywidgets import *
from geemap.utils import *

Map = geemap.Map(center=(-100, 40), zoom=4)
Map.default_style = {'cursor': 'pointer'}
Map

Map.setOptions('ROADMAP')
# Load National Hydrography Dataset (NHD)
HUC10 = ee.FeatureCollection('USGS/WBD/2017/HUC10')
# 18,487 HUC10 watersheds in the U.S.

# Add HUC layer to the map
Map.setCenter(-99.00, 47.01, 8)
Map.addLayer(
    ee.Image().paint(HUC10, 0, 1), {}, 'HUC-10 Watershed'
)  # HUC10 for the entire U.S.

label = Label("Click on the map to select a watershed")
widget_control = WidgetControl(widget=label, position='bottomright')
Map.add_control(widget_control)

layer = None


def handle_interaction(**kwargs):
    latlon = kwargs.get('coordinates')
    if kwargs.get('type') == 'click':
        Map.default_style = {'cursor': 'wait'}
        xy = ee.Geometry.Point(latlon[::-1])
        watershed = HUC10.filterBounds(xy)
        huc10_id = watershed.first().get('huc10').getInfo()
        Map.layers = Map.layers[:3]
        Map.addLayer(
            ee.Image().paint(watershed, 0, 2), {'palette': 'red'}, 'HUC ID: ' + huc10_id
        )

        NAIP_images = find_NAIP(watershed)
        first_image = ee.Image(NAIP_images.toList(5).get(0))
        Map.addLayer(first_image, {'bands': ['N', 'R', 'G']}, 'first image')
        count = NAIP_images.size().getInfo()
        for i in range(0, count):
            image = ee.Image(NAIP_images.toList(count).get(i))
            Map.addLayer(image, {'bands': ['N', 'R', 'G']}, str(i))

        Map.default_style = {'cursor': 'pointer'}


Map.on_interaction(handle_interaction)

Map

Map.layers