more streamlined

app.py

@@ -19,48 +19,73 @@ import altair as alt
 import ibis
 from ibis import _
 import ibis.selectors as s
+from streamlit_folium import st_folium
+import json
+
+def extract_geom(gdf, cog):
+    x = (rioxarray.
+         open_rasterio('/vsicurl/'+cog, masked=True).
+         rio.clip(gdf.geometry.values, gdf.crs, from_disk=True)
+    )
+    return x
+
+def read_polygon(polygon):
+    geojson_str = json.dumps(polygon)
+    gdf = gpd.read_file(geojson_str, driver='GeoJSON')
+    gdf.set_crs('epsg:4326')
+    return gdf
+
+def area_hectares(gdf):
+    area = gdf.to_crs("EPSG:3857").area / 10000.
+    return area
 
 
 # +
 st.set_page_config(layout="wide", page_title="Leafmap Explorer", page_icon="⚡")
 
-'''
-# Demo Carbon Calculator
+st.title("Demo Carbon Calculator")
 
-Use the map tools to draw a polygon (pentagon tool), bounding box (square tool) or other shape anywhere on the map.
-Click inside the box when you are done to show the coordinates.
-Please copy-paste those coordinates (or any geojson of your choosing) into the polygon selector.
-Map will zoom in on selected area and display the tons of carbon lost between 2002 - 2022 below.
+DESCRIPTION='''
+Pan and zoom to the desired location on the map. Then, use the map tools to draw a polygon (pentagon tool), bounding box (square tool) or other shape anywhere on the map.
+(use esc key to exit drawing mode).  Map will display the tons of carbon lost between 2002 - 2022 below.
 Data comes from Vizzuality repo on [source.coop](https://beta.source.coop/repositories/vizzuality/lg-land-carbon-data/description/).
 '''
 
+code_ex = ""
+
 deforest = "https://data.source.coop/vizzuality/lg-land-carbon-data/deforest_carbon_100m_cog.tif"
+# measured in Mg / Hct (tons)
 irrecoverable = "https://data.source.coop/cboettig/carbon/cogs/irrecoverable_c_total_2018.tif"
 vulnerable = "https://data.source.coop/cboettig/carbon/cogs/vulnerable_c_total_2018.tif"
+manageable = "https://data.source.coop/cboettig/carbon/cogs/manageable_c_total_2018.tif"
 
-m = leafmap.Map(center=[35, -100], zoom=5)
-m.add_cog_layer("https://data.source.coop/vizzuality/lg-land-carbon-data/deforest_carbon_100m_cog.tif", palette="reds", name="deforested", transparent_bg=True, opacity = 0.8, zoom_to_layer=False)
-# -
+#rsr = "https://data.source.coop/cboettig/mobi/range-size-rarity-all/RSR_All.tif"
+#richness = "https://data.source.coop/cboettig/mobi/species-richness-all/SpeciesRichness_All.tif"
+
+# Use signed data layers
+#rsr = "https://data.source.coop/cboettig/mobi/range-size-rarity-all/RSR_All.tif"
+#richness = "https://data.source.coop/cboettig/mobi/species-richness-all/SpeciesRichness_All.tif"
 
 
-polygon_ex ='{"type":"Feature","properties":{},"geometry":{"type":"Polygon","coordinates":[[[-124.628906,34.741612],[-124.628906,42.423457],[-117.246094,42.423457],[-117.246094,34.741612],[-124.628906,34.741612]]]}}'
-code_ex = '''
-m.add_gdf(geo, layer_name="selection")
-'''
 
+m = leafmap.Map(center=[35, -100], zoom=2)
 
 ## Map controls sidebar
 with st.sidebar:
-    '''
-    # Map controls
-    
-    ###  Polygon selector
-    '''
-    
-    polygon = st.text_area(
-        label = "geometry",
-        value = polygon_ex, 
-        height = 400)
+    st.markdown(DESCRIPTION)
+
+    cog_layers = {
+            "Carbon Lost, 2002-2022": deforest,
+            "Vulnerable Carbon (2018)": vulnerable,
+            "Manageable Carbon (2018)": manageable,
+            "Irrecoverable Carbon (2018)": irrecoverable
+            }
+
+    selection = st.radio("Data", cog_layers)
+    cog = cog_layers[selection]
+    m.add_cog_layer(cog, palette="reds", name=selection,
+                    transparent_bg=True, opacity = 0.8, 
+                    zoom_to_layer=False)
 
     "### python code for map layer:"
     "adjust options or add additional layers using leafmap"
@@ -69,66 +94,71 @@ with st.sidebar:
         value = code_ex, 
         height = 400)
 
+# run whatever python code is in the python box, just for fun
+eval(compile(code, "<string>", "exec"))
+st_data = m.to_streamlit(height=400, bidirectional=True)
+
+units="Tonnes"
 
+polygon = st_data["last_active_drawing"]
 # Here we actually compute the total carbon in the requested polygon
-geo = gpd.read_file(polygon, driver='GeoJSON')
-geo.set_crs('epsg:4326')
-x = (rioxarray.
-     open_rasterio('/vsicurl/'+deforest, masked=True).
-     rio.clip(geo.geometry.values, geo.crs, from_disk=True).
-     sum()
-    )
-value = float(x)
+if polygon is not None:
+    gdf = read_polygon(polygon)
+    x = extract_geom(gdf, cog)
+    value = float(x.sum())
+    if(selection in ["Vulnerable Carbon (2018)",
+                     "Manageable Carbon (2018)",
+                     "Irrecoverable Carbon (2018)"]):
+        value = value * 9 # 300m pixels, each pixel is 9 hectres
 
+    st.metric(label=f"{selection}", value=f"{value:,}  {units}")
 
-"### Tons of carbon lost:"
-st.write(f'{value:,}')
-st.divider()
 
-# run whatever python code is in the python box, just for fun
-eval(compile(code, "<string>", "exec"))
-m.to_streamlit(height=700)
+st.divider()
 
 "## Explore further"
-st.write('''
-Try adding additional options to the map. Some suggested examples are shown below. 
-(Of course any self-respecting streamlit app would make these into toggle buttons,
-but this is a demo and it's fun & flexible to be able to execute arbitrary code).
 
-To explore further, simply modify the Streamlit app.py file from the Files menu up top!
-'''
-)
-
-st.code('''
-# irrecoverable carbon (Conservation International):
-carbon = "https://data.source.coop/cboettig/carbon/cogs/irrecoverable_c_total_2018.tif"
-m.add_cog_layer(carbon, palette="reds", name="irrecoverable carbon", transparent_bg=True, opacity = 0.8, zoom_to_layer=False)')
-
-# Human Impacts, Vizzuality
-hi="https://data.source.coop/vizzuality/hfp-100/hfp_2021_100m_v1-2_cog.tif"
-m.add_cog_layer(hi, palette="purples", name="human impact", transparent_bg=True, opacity = 0.8, zoom_to_layer=False)
-
-# Fire Polygons, USGS
-usgs = "https://data.source.coop/cboettig/fire/usgs-mtbs.pmtiles"
-combined_style = {
-    "version": 8,
-    "sources": {
-        "source1": {
-            "type": "vector",
-            "url": "pmtiles://" + usgs,
-            "attribution": "USGS"}},
-    "layers": [{
-            "id": "usgs",
-            "source": "source1",
-            "source-layer": "mtbs_perims_DD",
-            "type": "fill",
-            "paint": {"fill-color": "#FFA500", "fill-opacity": 0.2}}]}
-
-m.add_pmtiles(usgs, name="Fire", style=combined_style, overlay=True, show=True, zoom_to_layer=False)
-''')       
+with st.expander("code examples"):
+    st.write('''
+    Try adding additional options to the map. Some suggested examples are shown below. 
+    (Of course any self-respecting streamlit app would make these into toggle buttons,
+    but this is a demo and it's fun & flexible to be able to execute arbitrary code).
+
+    To explore further, simply modify the Streamlit app.py file from the Files menu up top!
+    '''
+    )
+
+    st.code('''
+    # irrecoverable carbon (Conservation International):
+    carbon = "https://data.source.coop/cboettig/carbon/cogs/irrecoverable_c_total_2018.tif"
+    m.add_cog_layer(carbon, palette="reds", name="irrecoverable carbon",
+                    transparent_bg=True, opacity = 0.8, zoom_to_layer=False)
+
+    # Human Impacts, Vizzuality
+    hi="https://data.source.coop/vizzuality/hfp-100/hfp_2021_100m_v1-2_cog.tif"
+    m.add_cog_layer(hi, palette="purples", name="human impact", 
+                    transparent_bg=True, opacity = 0.8, zoom_to_layer=False)
+
+    # Fire Polygons, USGS
+    usgs = "https://data.source.coop/cboettig/fire/usgs-mtbs.pmtiles"
+    combined_style = {
+        "version": 8,
+        "sources": {
+            "source1": {
+                "type": "vector",
+                "url": "pmtiles://" + usgs,
+                "attribution": "USGS"}},
+        "layers": [{
+                "id": "usgs",
+                "source": "source1",
+                "source-layer": "mtbs_perims_DD",
+                "type": "fill",
+                "paint": {"fill-color": "#FFA500", "fill-opacity": 0.2}}]}
+
+    m.add_pmtiles(usgs, name="Fire", style=combined_style, overlay=True, show=True, zoom_to_layer=False)
+    ''')       
 
 
-# +
 st.divider()
 
 '''
@@ -137,9 +167,13 @@ st.divider()
 ### Data sources
 
 - Carbon-loss by Vizzuality, on https://beta.source.coop/repositories/vizzuality/lg-land-carbon-data. citation: https://doi.org/10.1101/2023.11.01.565036, License: CC-BY
-- Human Footprint by Vizzuality, on https://beta.source.coop/repositories/vizzuality/hfp-100,  citation: https://doi.org/10.3389/frsen.2023.1130896, License: Public Domain
-- Fire polygons by USGS, reprocessed to PMTiles on https://beta.source.coop/cboettig/fire/, License: Public Domain.
 - Irrecoverable Carbon from Conservation International, reprocessed to COG on https://beta.source.coop/cboettig/carbon, citation: https://doi.org/10.1038/s41893-021-00803-6, License: CC-BY-NC
+- Fire polygons by USGS, reprocessed to PMTiles on https://beta.source.coop/cboettig/fire/, License: Public Domain.
+
+### Software stack
+ 
+- Streamlit (python) app hosted on free-tier HuggingFace spaces ([source code](https://huggingface.co/spaces/boettiger-lab/leafmap/blob/main/app.py)).
+- Cloud-optimized geotifs hosted on [Source.Coop](https://source.coop)
+- Mapping with Leafmap, calculations with rasterio
 
-Software stack: Streamlit (python) app hosted on free-tier HuggingFace spaces.  Mapping with Leafmap.
 '''

requirements.txt

@@ -7,3 +7,4 @@ ibis-framework[duckdb]
 altair
 rioxarray
 geopandas
+streamlit-folium