## Europe
m = Map(center=(50.0, 6.5), zoom=4)

# Create draw control for rectangles only
draw_control = DrawControl()
draw_control.rectangle = {"shapeOptions": {"fillOpacity": 0.3}}
draw_control.polygon = {}  # Disable polygon
draw_control.polyline = {}  # Disable polyline
draw_control.circle = {}  # Disable circle
draw_control.marker = {}  # Disable marker

# Global variable to store the bbox
rectangle_bbox = None

# Output widget
output = widgets.Output()

# Handle rectangle drawing
def handle_draw(target, action, geo_json):
    global rectangle_bbox
    
    if action == 'created':
        with output:
            output.clear_output()
            # Get current map bounds using built-in method
            bounds = m.bounds
            print(f"Map bounds: {bounds}")
            
            # Extract bbox from drawn rectangle
            coords = geo_json['geometry']['coordinates'][0]
            lons = [coord[0] for coord in coords]
            lats = [coord[1] for coord in coords]
            rectangle_bbox = [min(lons), min(lats), max(lons), max(lats)]
            print(f"Rectangle bbox: {rectangle_bbox}")
            print(f"Bbox stored in variable 'rectangle_bbox'")

# Function to get the current bbox
def get_bbox():
    """Return the current rectangle bbox"""
    return rectangle_bbox

draw_control.on_draw(handle_draw)
m.add_control(draw_control)

# Display
display(m)
display(output)

# Global variable to store selected date
selected_date = None

# Create date picker widget (simplest approach)
date_picker = widgets.DatePicker(
    description='Select Date:',
    value=datetime.date(2024,1,1),  # Default to today
    disabled=False
)

# Output widget to display selected date
output = widgets.Output()

# Function to handle date selection
def on_date_change(change):
    global selected_date
    selected_date = change['new']
    
    with output:
        output.clear_output()
        print(f"Selected date: {selected_date}")
        print(f"Formatted: {selected_date.strftime('%Y-%m-%d')}")

# Attach the event handler
date_picker.observe(on_date_change, names='value')

# Function to get the selected date
def get_selected_date():
    """Return the currently selected date"""
    return selected_date

# Display the calendar
display(date_picker)
display(output)

selected_date.strftime('%Y-%m-%d')

# Time options: 00:00 to 23:00
time_options = [f"{hour:02d}:00" for hour in range(24)]

# Create dropdown widget
time_selector = widgets.Dropdown(
    options=time_options,
    description='Select Time:',
    value="00:00",
    style={'description_width': 'initial'}
)

# Initialize variable
selected_time = time_selector.value

# Define observer function
def on_time_change(change):
    global selected_time
    selected_time = change['new']
    print(f"Time selected: {selected_time}")

# Attach observer
time_selector.observe(on_time_change, names='value')

# Display the widget
display(time_selector)

selected_time

int(selected_time[:2])

Animation = wrapper_vector_field_animation(
    API_KEY = my_api_key,
    date = selected_date.strftime('%Y-%m-%d'),
    roi = rectangle_bbox,
    time = selected_time
)

## Run Manim Scene using inline
%manim -ql -v WARNING Animation

Animation = wrapper_vector_field_animation(
    API_KEY = my_api_key,
    date = selected_date.strftime('%Y-%m-%d'),
    roi = rectangle_bbox,
    time = selected_time,
    include_features = True,
    title = 'ERA5 Wind Animation',
    colorized_animation = True,
    colorscheme = 'viridis'
)

## Run Manim Scene using inline
%manim -ql -v WARNING Animation