Add analysis gui

.gitignore

@@ -2,6 +2,7 @@
 __pycache__/
 *.py[cod]
 *$py.class
+private/
 
 # C extensions
 *.so

pages/00_home.py

@@ -12,7 +12,8 @@ def Page():
         We have used five global datasets viz., ESA, ESRI, JRC, OSM and HydroLAKES, 
         to generate a processed multi-band gridded GSWE dataset at 10 m spatial 
         resolution, with each band corresponding to one of the five datasets used. 
-        The dataset comprises of a total of 5153 grids, each grid having a 2°×2° dimension.
+        The dataset comprises of a total of 5153 grids, each grid having a 2°×2° dimension. 
+        **Click on the GSWIS tab above to visualize the datasets interactively.**
 
         ### Datasets
 

pages/01_gswis.py

@@ -1,17 +1,173 @@
 import ee
 import geemap
-
+import ipyleaflet
 import solara
+import ipywidgets as widgets
+from IPython.display import display
 
 zoom = solara.reactive(3)
 center = solara.reactive([20, 0])
 
 
+def zonal_stats_chart(image, vector, **kwargs):
+    if isinstance(vector, ee.Geometry):
+        fc = ee.FeatureCollection(vector)
+    elif isinstance(vector, ee.FeatureCollection):
+        fc = vector
+    else:
+        raise ValueError(
+            "The vector argument must be an ee.Geometry or ee.FeatureCollection."
+        )
+    result = geemap.zonal_stats(
+        image, fc, statistics_type="SUM", return_fc=True, verbose=False, **kwargs
+    )
+    df = geemap.ee_to_df(result).T
+    df.reset_index(inplace=True)
+    df.columns = ["Type", "Area"]
+    chart = geemap.bar_chart(df, "Type", "Area", x_label='', y_label="Area (m2)")
+    chart.update_layout(
+        margin=dict(l=0, r=0, t=10, b=0),
+        height=280,
+    )
+
+    return chart
+
+
+def add_analysis_gui(m=None, position='topright', opened=True):
+    """Create a toolbar widget.
+
+    Args:
+        m (geemap.Map, optional): The geemap.Map instance. Defaults to None.
+        opened (bool, optional): Whether to open the toolbar. Defaults to True.
+    """
+
+    fc = ee.FeatureCollection('users/giswqs/public/countries')
+    gswe = ee.ImageCollection("users/h2i_lab/gswe/gswe_datasets")
+    image = gswe.mosaic()
+    # esa = gswe.select("esa").mosaic()
+    # esri = gswe.select("esri").mosaic()
+    # jrc = gswe.select("jrc").mosaic()
+    # osm = gswe.select("osm").mosaic()
+    # hydrolakes = gswe.select("hydrolakes").mosaic()
+
+    widget_width = "270px"
+    padding = "0px 0px 0px 5px"  # upper, right, bottom, left
+
+    toolbar_button = widgets.ToggleButton(
+        value=False,
+        tooltip="Toolbar",
+        icon="bar-chart",
+        layout=widgets.Layout(width="28px", height="28px", padding="0px 0px 0px 4px"),
+    )
+
+    close_button = widgets.ToggleButton(
+        value=False,
+        tooltip="Close the tool",
+        icon="times",
+        button_style="primary",
+        layout=widgets.Layout(height="28px", width="28px", padding="0px 0px 0px 4px"),
+    )
+
+    options = ["Draw an area", "Select a country"]
+
+    radio = widgets.RadioButtons(
+        options=options,
+        layout=widgets.Layout(width=widget_width, padding=padding),
+        style={"description_width": "initial"},
+    )
+
+    buttons = widgets.ToggleButtons(
+        value=None,
+        options=["Apply", "Reset", "Close"],
+        tooltips=["Apply", "Reset", "Close"],
+        button_style="primary",
+    )
+    buttons.style.button_width = "88px"
+    label = widgets.Label("Draw an area on the map first.")
+
+    output = widgets.Output(layout=widgets.Layout(width=widget_width, padding=padding))
+
+    toolbar_widget = widgets.VBox()
+    toolbar_widget.children = [toolbar_button]
+    toolbar_header = widgets.HBox()
+    toolbar_header.children = [close_button, toolbar_button]
+    toolbar_footer = widgets.VBox()
+    toolbar_footer.children = [
+        radio,
+        buttons,
+        output,
+    ]
+
+    def toolbar_btn_click(change):
+        if change["new"]:
+            close_button.value = False
+            toolbar_widget.children = [toolbar_header, toolbar_footer]
+        else:
+            if not close_button.value:
+                toolbar_widget.children = [toolbar_button]
+
+    toolbar_button.observe(toolbar_btn_click, "value")
+
+    def close_btn_click(change):
+        if change["new"]:
+            toolbar_button.value = False
+            if m is not None:
+                if m.tool_control is not None and m.tool_control in m.controls:
+                    m.remove_control(m.tool_control)
+                    m.tool_control = None
+            toolbar_widget.close()
+
+    close_button.observe(close_btn_click, "value")
+
+    def button_clicked(change):
+        if change["new"] == "Apply":
+            with output:
+                output.clear_output()
+                # com_label = widgets.Label("PLEASE WAIT ...")
+                # display(com_label)
+                buttons.value = None
+
+                if radio.value == "Draw an area":
+                    if m.user_roi is None:
+                        display(label)
+                        buttons.value = None
+                    else:
+                        chart = zonal_stats_chart(image, m.user_roi, scale=100)
+                        display(chart)
+
+        elif change["new"] == "Reset":
+            output.clear_output()
+        elif change["new"] == "Close":
+            if m is not None:
+                if m.tool_control is not None and m.tool_control in m.controls:
+                    m.remove_control(m.tool_control)
+                    m.tool_control = None
+            toolbar_widget.close()
+
+        buttons.value = None
+
+    buttons.observe(button_clicked, "value")
+
+    toolbar_button.value = opened
+    if m is not None:
+        toolbar_control = ipyleaflet.WidgetControl(
+            widget=toolbar_widget, position=position
+        )
+
+        if toolbar_control not in m.controls:
+            m.add_control(toolbar_control)
+            m.tool_control = toolbar_control
+    else:
+        return toolbar_widget
+
+
 class Map(geemap.Map):
     def __init__(self, **kwargs):
         super().__init__(**kwargs)
+        self.add_basemap('SATELLITE', show=False)
         self.add_ee_data()
         self.add_layer_manager()
+        add_analysis_gui(self)
         # self.add_inspector()
 
     def add_ee_data(self):
@@ -27,9 +183,9 @@ class Map(geemap.Map):
             'ESRI': 'ffff00',
             'JRC': '0000ff',
             'OSM': '00ff00',
-            'Hydrolakes': '800080'
+            'Hydrolakes': '800080',
         }
-        self.add_legend(legend_dict=legend_dict, position='bottomright')
+        self.add_legend(legend_dict=legend_dict, position='bottomleft')
         fc = ee.FeatureCollection('users/giswqs/public/countries')
         style = {'color': '000000ff', 'width': 1, 'fillColor': '00000000'}
         self.addLayer(fc.style(**style), {}, 'Countries', False)
@@ -46,4 +202,5 @@ def Page():
             scroll_wheel_zoom=True,
             add_google_map=True,
             height="800px",
+            data_ctrl=False,
         )