Merge pull request #11 from boettiger-lab/feat/update_app

fixing state filter in charts

Author: cassiebuhler

app/app.py

@@ -65,7 +65,7 @@
 legend, position, bg_color, fontsize, shape_type, controls = get_legend(paint, leafmap_choice)
 # get all the ids that correspond to the filter
 gdf = filter_data(tpl_table, state_choice, county_choice, year_range)
-gdf_landvote = filter_data(landvote_table, state_choice, county_choice, year_range)
+gdf_landvote = filter_data(landvote_table, state_codes[state_choice], county_choice, year_range)
 unique_ids = gdf.select('fid').distinct().execute()['fid'].to_list()
 
 ##### Chatbot stuff 
@@ -291,6 +291,8 @@ def run_sql(query, llm_choice):
 
 st.markdown('#')
 
+
+
 col1, col2 = st.columns(2)
 with col1:
     gdf_tpl = group_data(gdf, 'Acquisition Cost')

app/variables.py

@@ -56,6 +56,61 @@
     "West Virginia", "Wisconsin", "Wyoming"
 )
 
+state_codes = {
+    "Alabama": "AL",
+    "Alaska": "AK",
+    "Arizona": "AZ",
+    "Arkansas": "AR",
+    "California": "CA",
+    "Colorado": "CO",
+    "Connecticut": "CT",
+    "Delaware": "DE",
+    "Florida": "FL",
+    "Georgia": "GA",
+    "Hawaii": "HI",
+    "Idaho": "ID",
+    "Illinois": "IL",
+    "Indiana": "IN",
+    "Iowa": "IA",
+    "Kansas": "KS",
+    "Kentucky": "KY",
+    "Louisiana": "LA",
+    "Maine": "ME",
+    "Maryland": "MD",
+    "Massachusetts": "MA",
+    "Michigan": "MI",
+    "Minnesota": "MN",
+    "Mississippi": "MS",
+    "Missouri": "MO",
+    "Montana": "MT",
+    "Nebraska": "NE",
+    "Nevada": "NV",
+    "New Hampshire": "NH",
+    "New Jersey": "NJ",
+    "New Mexico": "NM",
+    "New York": "NY",
+    "North Carolina": "NC",
+    "North Dakota": "ND",
+    "Ohio": "OH",
+    "Oklahoma": "OK",
+    "Oregon": "OR",
+    "Pennsylvania": "PA",
+    "Puerto Rico": "PR",
+    "Rhode Island": "RI",
+    "South Carolina": "SC",
+    "South Dakota": "SD",
+    "Tennessee": "TN",
+    "Texas": "TX",
+    "Utah": "UT",
+    "Vermont": "VT",
+    "Virginia": "VA",
+    "Washington": "WA",
+    "West Virginia": "WV",
+    "Wisconsin": "WI",
+    "Wyoming": "WY",
+}
+
+
 # Define color hex codes
 darkblue = "#00008B"
 blue = "#0096FF"