Sending all changes and seeing what sticks

app.py

@@ -1,21 +1,32 @@
+from boettiger_app import read_polygon
 import leafmap.foliumap as leafmap
 import rioxarray
 import geopandas as gpd
 import streamlit as st
-
-# import altair as alt
-# import ibis
-# from ibis import _
-# import ibis.selectors as s
+import altair as alt
+import ibis
+from ibis import _
+import ibis.selectors as s
 from streamlit_folium import st_folium
 import json
 from leafmap import pmtiles_style
 
+
 # Common Functions
+def read_polygon(polygon):
+    """
+    Take in a polygon from the app and convert it to GeoJSON
+    """
+    geojson_str = json.dumps(polygon)
+    gdf = gpd.read_file(geojson_str, driver="GeoJSON")
+    gdf.set_crs("epsg:4326")
+    return gdf
 
 
 # Actual Page content below
-st.set_page_config(layout="wide", page_title="Huggingface Toy", page_icon="❦")
+started = None
+if not started:
+    st.set_page_config(layout="wide", page_title="Huggingface Toy", page_icon="❦")
 st.title("Huggingface Streamlit Toy")
 
 """
@@ -70,9 +81,10 @@ m.add_pmtiles(
 # TODO: Dynamically check the center of the map and change what's displayed depending on that.
 
 polygon_ex = '{"type":"Feature","properties":{},"geometry":{"type":"Polygon","coordinates":[[[-6.108398,40.780541],[-6.108398,54.342149],[15.908203,54.342149],[15.908203,40.780541],[-6.108398,40.780541]]]}}'
-code_ex = """
-m.add_gdf(geo, layer_name="selection")
-"""
+# code_ex = """
+# m.add_gdf(polygon, layer_name="selection")
+# """
+code_ex = ""
 
 
 ## Map controls sidebar
@@ -96,28 +108,35 @@ with st.sidebar:
 # if bounds is not None:
 #     bounds.set_crs("epsg:4326")
 
-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)
-    .mean()
-)
-value = x
+# 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)
+#     .mean()
+# )
+# value = x
 
 
 "### Average of tons of carbon lost:"
-st.write(f"{value:,}")
+# st.write(f"{value:,}")
 st.divider()
 
 # run whatever python code is in the python box, just for fun
 eval(compile(code, "<string>", "exec"))
+
 st_data = m.to_streamlit(height=700, bidirectional=True)
 polygon = st_data["last_active_drawing"]
+"### Could there be a polygon here?"
+st.write(f"{polygon is not None}")
+
 if polygon is not None:
     "### The Polygon is: "
+    gdf = read_polygon(polygon)
     st.write(f"{polygon}")
 
+started = True
+
 "## Explore Further"
 st.write("""
 To modify the map, add code to the `code` section in the sidebar to the left. Some examples of code to add are below.