add county level us map

app.py

@@ -10,6 +10,7 @@ from google.analytics.data_v1beta.types import (
 import gradio as gr
 import os
 import json
+import time
 import pandas as pd
 import plotly.express as px
 
@@ -33,6 +34,18 @@ iso['Alpha-3 code'] = iso['Alpha-3 code'].str.strip()
 iso.set_index('Alpha-2 code', inplace=True)
 alpha_2_map = iso['Alpha-3 code'].to_dict()
 
+# read counties json file
+with open('counties.json') as f:
+    counties = json.load(f)
+
+cities = pd.read_csv('uscities.csv')
+cities.set_index('city', inplace=True)
+city_county_map = cities['county_fips'].to_dict()
+city_county_name_map = cities['county_name'].to_dict()
+
+cached_report = None
+cache_time = 0
+
 def full_report():
     client = BetaAnalyticsDataClient()
 
@@ -42,17 +55,15 @@ def full_report():
                     Dimension(name='eventName'),
                     Dimension(name="continent"),
                     Dimension(name="country"),
-                    Dimension(name="countryId")],
+                    Dimension(name="countryId"),
+                    Dimension(name="city")],
         metrics=[Metric(name="eventValue")],
         #return_property_quota=True,
         date_ranges=[DateRange(start_date="2023-06-30", end_date="today")],
     )
     response = client.run_report(request)
 
-    
-
-    total_jumps = 0
-    res = {'day': [], 'jumps': [], 'continent': [], 'country': [], 'iso': []}
+    res = {'day': [], 'jumps': [], 'continent': [], 'country': [], 'iso': [], 'city': []}
 
     for row in response.rows:
         event_name = row.dimension_values[1].value
@@ -61,18 +72,22 @@ def full_report():
             continent = row.dimension_values[2].value
             country = row.dimension_values[3].value
             country_iso = row.dimension_values[4].value
+            city = row.dimension_values[5].value
             event_value = float(row.metric_values[0].value)
-            total_jumps += int(event_value)
             res['day'].append(day)
             res['jumps'].append(event_value)
             res['continent'].append(continent)
             res['country'].append(country)
             res['iso'].append(country_iso)
+            res['city'].append(city)
 
-    print(f"Total jumps: {total_jumps}")
+    
 
     df = pd.DataFrame.from_dict(res)
+    total_jumps = df['jumps'].sum()
+    print(f"Total jumps: {total_jumps}")
     df['iso'] = df['iso'].map(alpha_2_map)
+    
     country_df = df.groupby(['country', 'iso']).sum().reset_index()
     country_df = country_df.sort_values(by=['jumps'], ascending=False)
 
@@ -96,8 +111,23 @@ def full_report():
                                 template="plotly_dark")
     # remove the legend
     total_map.update_layout(showlegend=False)
+    total_map.update_layout(margin={"r":0,"t":0,"l":0,"b":0})
     total_map.update(layout_coloraxis_showscale=False)
 
+    county_df = df.copy()
+    county_df['county'] = county_df['city'].map(city_county_map)
+    county_df['count_name'] = county_df['city'].map(city_county_name_map)
+    county_df = county_df.groupby(['county', 'count_name']).sum().reset_index()
+    print(county_df)
+
+    county_map = px.choropleth(county_df, geojson=counties, locations='county', color='jumps',
+                           color_continuous_scale="plasma",
+                           scope="usa",
+                           hover_name="count_name",
+                           template="plotly_dark"
+                          )
+    county_map.update_layout(margin={"r":0,"t":0,"l":0,"b":0})
+
     df = df.groupby(['day', 'continent']).sum().reset_index()
     df = df.sort_values(by=['day'])
     df['total_jumps'] = df.groupby('continent')['jumps'].cumsum()
@@ -106,7 +136,7 @@ def full_report():
                               color='continent', 
                               template="plotly_dark")
 
-    return f"## Total Jumps: {total_jumps:,}", total, avg, total_map, jumps_over_time
+    return f"## Total Jumps: {total_jumps:,}", total, avg, total_map, jumps_over_time, county_map
 
 
 with gr.Blocks(theme='WeixuanYuan/Soft_dark') as demo:
@@ -114,6 +144,8 @@ with gr.Blocks(theme='WeixuanYuan/Soft_dark') as demo:
         total_jumps_label = gr.Markdown("Total Jumps: 0")
     with gr.Row():
         map_fig = gr.Plot(label="Map")
+    with gr.Row():
+        county_map = gr.Plot(label="US Map")
     with gr.Row():
         jumps_over_time = gr.Plot(label="Jumps Over Time")
     with gr.Row():
@@ -121,7 +153,7 @@ with gr.Blocks(theme='WeixuanYuan/Soft_dark') as demo:
     with gr.Row():
         avg_plot = gr.Plot(label="Average Jumps per Day")
 
-    outputs = [total_jumps_label, total_plot, avg_plot, map_fig, jumps_over_time]
+    outputs = [total_jumps_label, total_plot, avg_plot, map_fig, jumps_over_time, county_map]
     dep = demo.load(full_report, None, outputs)
 
 if __name__ == "__main__":