add treemap

app.py

@@ -11,6 +11,7 @@ import gradio as gr
 import os
 import json
 import time
+import math
 import pandas as pd
 import plotly.express as px
 
@@ -126,6 +127,9 @@ def full_report():
     country_df_to_plot = df.groupby(['country', 'iso']).sum().reset_index()
     country_df_to_plot = country_df_to_plot[country_df_to_plot['country'].isin(top_10_countries)].reset_index(drop=True)
     country_df_to_plot = country_df_to_plot.sort_values(by=['jumps'], ascending=True)
+    df['rank'] = df['jumps'].rank(ascending=False)
+    df['world'] = 'Earth'
+
     total = px.bar(country_df_to_plot, 
                    y='country', x='jumps', 
                    color='country',
@@ -139,6 +143,27 @@ def full_report():
     city_df = city_df.sort_values(by=['jumps'], ascending=False)
     city_df['city'] = city_df.apply(lambda row: row['city'] + ', ' + row['iso'], axis=1)
     top_10_cities = city_df.iloc[:10]['city'].tolist()
+    
+    icicle_df = df.groupby(['world', 'continent', 'country', 'city']).sum().reset_index()
+    #icicle_df['log10_jumps'] = icicle_df['jumps'].apply(lambda x: math.log10(x) if x > 0 else 0)
+
+    # icicle = px.icicle(icicle_df, path=['world', 'continent', 'country', 'city'], 
+    #                    values='jumps', 
+    #                    title='Jumps by Continent/Country', 
+    #                    template="plotly_dark", 
+    #                    color_continuous_scale='OrRd',
+    #                    maxdepth=7,
+    #                    branchvalues='remainder',
+    #                    color='jumps')
+
+    icicle = px.treemap(icicle_df, path=['world', 'continent', 'country', 'city'], 
+                       values='jumps', 
+                       title='Jumps by Continent/Country/City', 
+                       template="plotly_dark", 
+                       color_continuous_scale='OrRd',
+                       branchvalues='total',
+                       height=800,
+                       color='jumps')
 
     city_df = df.groupby(['city', 'iso']).sum().reset_index()
     city_df = city_df[city_df['city'] != '(not set)']
@@ -209,7 +234,7 @@ def full_report():
                               trendline_scope='overall',
                               template="plotly_dark")
 
-    return f"# {total_jumps:,} total jumps in {unique_cities:,} cities across {unique_countries:,} countries", total, avg, total_map, jumps_over_time, county_map, per_day_plot
+    return f"# {total_jumps:,} total jumps in {unique_cities:,} cities across {unique_countries:,} countries", total, avg, total_map, icicle, jumps_over_time, county_map, per_day_plot
 
 
 with gr.Blocks(theme='WeixuanYuan/Soft_dark') as demo:
@@ -223,12 +248,14 @@ with gr.Blocks(theme='WeixuanYuan/Soft_dark') as demo:
         total_plot = gr.Plot(label="Top 10 Countries")
     with gr.Row():
         avg_plot = gr.Plot(label="Top 10 Cities")
+    with gr.Row():
+        icicle_fig = gr.Plot(label="Treemap")
     with gr.Row():
         per_day_plot = gr.Plot(label="Jumps per Day")
     with gr.Row():
         county_map = gr.Plot(label="US Map")
 
-    outputs = [total_jumps_label, total_plot, avg_plot, map_fig, jumps_over_time, county_map, per_day_plot]
+    outputs = [total_jumps_label, total_plot, avg_plot, map_fig, icicle_fig, jumps_over_time, county_map, per_day_plot]
     dep = demo.load(full_report, None, outputs)
 
 if __name__ == "__main__":