Add fallback for TAs with no counties_served

.github/workflows/provider-map-jobs.yml

@@ -54,10 +54,27 @@ jobs:
           import json
           import pandas as pd
 
-          df = pd.read_csv('src/metadata/providers/providers.csv')
+          providers_file = 'src/metadata/providers/providers.csv'
 
-          # Drop the null values for TAs that don't have counties served
-          df = df[df['counties_served'].notnull()]
+          df = pd.read_csv(providers_file')
+          city_lookup = pd.read_csv('src/metadata/cities_to_county.csv')
+          city_to_county = dict(zip(city_lookup['City'], city_lookup['County']))
+
+          lookup_records = df[df['counties_served'].isna()]['ntd_id']
+
+          # Fill in the null values for counties served with the HQ county
+          for record in lookup_records:
+            city = df[df['ntd_id'] == record]['hq_city'].values[0]
+
+            try:
+              county = city_to_county[city] or city_to_county[f'City of {city}']
+              df.loc[df['ntd_id'] == record, 'hq_county'] = county
+              df.loc[df['ntd_id'] == record, 'counties_served'] = county
+
+            except KeyError:
+              print("No county found for city: ", city)
+
+          df.to_csv(providers_file)
 
           # Do a group by for the counties served
           county_counts = df['counties_served'].str.split(';') \