adding parties

app.py

@@ -63,8 +63,9 @@ year = st.slider("Select a year", min_value=1988, max_value=2024, value=2022, st
 import leafmap.maplibregl as leafmap
 m = leafmap.Map(style="positron",   center=(-100, 40), zoom=3)
 
-url = "https://huggingface.co/datasets/boettiger-lab/landvote/resolve/main/landvote_polygons.pmtiles"
-# url = "https://huggingface.co/datasets/boettiger-lab/landvote/resolve/main/votes.pmtiles"
+# url = "https://huggingface.co/datasets/boettiger-lab/landvote/resolve/main/landvote_polygons.pmtiles"
+url = "https://huggingface.co/datasets/boettiger-lab/landvote/resolve/main/votes.pmtiles"
+parties = "https://huggingface.co/datasets/boettiger-lab/landvote/resolve/main/votes.parquet"
 
 dark_orange = 'rgba(171, 86, 1, 1)'   # dark orange - min value 
 light_orange = 'rgba(243, 211, 177, 1)' # light orange 
@@ -72,6 +73,40 @@ grey = 'rgba(211, 211, 211, 1)'        # grey
 light_green = 'rgba(195, 219, 195, 1)' # light green 
 dark_green = 'rgba(65, 125, 65, 1)'    # dark green  - max value 
 
+con = ibis.duckdb.connect(extensions=["spatial"])
+
+party = (con
+            .read_parquet(parties)
+            .cast({"geometry": "geometry"})
+            )
+
+    
+def get_summary(party, year):
+    total_measures = party.filter(_.year == year).count().execute()
+    
+    df = (party
+        .filter(_.year == year)
+        .mutate(
+            # Convert 'amount' from string with '$' and ',' to numeric
+            amount_numeric=_.amount.replace('$', '').replace(',', '').cast('float64')
+        )
+        .group_by("party")
+        .aggregate(
+            percent_passed= (_.Status.isin(["Pass", "Pass*"]).sum() / total_measures).round(2),
+            approved_funds= ibis.case()
+                            .when(_.Status.isin(["Pass", "Pass*"]), _.amount_numeric)
+                            .else_(ibis.literal(0))
+                            .end()
+                            .sum()
+        )
+        .mutate(color=ibis.case()
+            .when(_.party == "DEMOCRAT", ibis.literal("#083A90"))
+            .else_(ibis.literal("#E81B23"))
+            .end())
+    )
+    
+    df = df.to_pandas()
+    return df
 
 style_municipals = {
     "layers": [
@@ -232,6 +267,16 @@ m.add_pmtiles(
     fit_bounds=False
 )
 
+
 m.add_layer_control()
 m.to_streamlit()
+# st.dataframe(df)
+
+df = get_summary(party, year)
+st.bar_chart(df, x= "party",y = "percent_passed",color="color")
+
+st.bar_chart(df, x= "party",y = "approved_funds",color="color")
+
+# st.divider()
+# footer = st.container()