Add plotted leaderboard

daily_weekly_leaderboard.py

@@ -1,5 +1,9 @@
 import gradio as gr
 from gradio_calendar import Calendar
+import polars as pl
+import pandas as pd
+import matplotlib.pyplot as plt
+from plottable import Table, ColumnDefinition
 
 from data import df, game_df
 from gradio_function import *
@@ -23,17 +27,24 @@ def filter_pitcher_leaderboard_by_date(date, *args, **kwargs):
     monday = date - datetime.timedelta(days=date.weekday())
     sunday = date + datetime.timedelta(days=6-date.weekday())
     week_df = df.filter((pl.col('game_date') >= monday) & (pl.col('game_date') <= sunday))
+    daily_whiffs, daily_velos = compute_pitcher_leaderboards(day_df, *args, **kwargs)
+    weekly_whiffs, weekly_velos = compute_pitcher_leaderboards(week_df, *args, **kwargs)
     return (
         f'<center><h1>Daily Leaderboard<h1><h2>{date.strftime("%B %d, %Y")}</h2><h3>{date.strftime("%A")}</h3></center>',
         f'<center><h1>Weekly Leaderboard<h1><h2>{monday.strftime("%B %d, %Y")} to {sunday.strftime("%B %d, %Y")}</h2><h3>{monday.strftime("%A")} to {sunday.strftime("%A")} </h3></center>',
-        *compute_pitcher_leaderboards(day_df, *args, **kwargs),
-        *compute_pitcher_leaderboards(week_df, *args, **kwargs),
+        daily_whiffs,
+        daily_velos,
+        weekly_whiffs,
+        weekly_velos,
+        plot_tables(daily_whiffs, daily_velos, 'Daily'),
+        plot_tables(weekly_whiffs, weekly_velos, 'Weekly'),
         gr.update(interactive=True),
         gr.update(interactive=True),
         gr.update(interactive=True),
         gr.update(interactive=True)
     )
-    
+
+
 def compute_pitcher_leaderboards(df, top_players, strict, ignore_zero_whiffs, show_rank, debug):
     # _df = df.filter(pl.col('game_date') == date)
     _df = df
@@ -96,18 +107,60 @@ def compute_pitcher_leaderboards(df, top_players, strict, ignore_zero_whiffs, sh
     # )
 
 
+def plot_tables(whiffs, velos, time_type):
+    fig, (whiff_ax, velo_ax) = plt.subplots(ncols=2, constrained_layout=True, sharey=True)
+
+    whiffs = whiffs.to_pandas()
+    if 'Rank' in whiffs.columns:
+        whiffs = whiffs.set_index('Rank')
+    else:
+        whiffs.index = pd.Series(range(1, len(whiffs)+1), name='Rank')
+    Table(
+      whiffs,
+      column_definitions=[
+          ColumnDefinition(name="Rank", title="Rank", width=0.25),
+          ColumnDefinition(name="Name", title="Player", textprops={'ha': 'left'}),
+          ColumnDefinition(name="Whiffs", title="#", width=0.25)
+      ],
+      ax=whiff_ax
+    )
+    whiff_ax.set_title('Whiffs')
+
+    velos = velos.to_pandas()
+    if 'Rank' in velos.columns:
+        velos = velos.set_index('Rank')
+    else:
+        velos.index = pd.Series(range(1, len(velos)+1), name='Rank')
+    Table(
+      velos,
+      column_definitions=[
+          ColumnDefinition(name="Rank", title="Rank", width=0.25),
+          ColumnDefinition(name="Name", title="Player", textprops={'ha': 'left'}),
+          ColumnDefinition(name="Velocity", title="MPH")
+      ],
+      ax=velo_ax
+    )
+    velo_ax.set_title('Velocity')
+
+    fig.suptitle(f'{time_type} Leaderboard')
+    return fig
+
 def go_back_day(date):
     return date - datetime.timedelta(days=1)
 
+
 def go_forward_day(date):
     return date + datetime.timedelta(days=1)
 
+
 def go_back_week(date):
     return date - datetime.timedelta(days=7)
 
+
 def go_forward_week(date):
     return date + datetime.timedelta(days=7)
-    
+
+
 def create_daily_pitcher_leaderboard():
     with gr.Blocks(
         css=css
@@ -151,10 +204,18 @@ def create_daily_pitcher_leaderboard():
             weekly_whiffs = gr.Dataframe(pl.DataFrame({'Name': [], 'Whiffs': []}), label='Whiffs', interactive=False, height=1000)
             weekly_velos = gr.Dataframe(pl.DataFrame({'Name': [], 'Velocity': []}), label='Velocity', interactive=False, height=1000)
 
+        gr.Markdown('''# Plotted leaderboards
+
+        For easier sharing
+        ''')
+        with gr.Row():
+            daily_plot = gr.Plot(label='Daily plot')
+            weekly_plot = gr.Plot(label='Weekly plot')
+            
         search_kwargs = dict(
             fn=filter_pitcher_leaderboard_by_date,
             inputs=[date_picker, top_players, strict, ignore_zero_whiffs, show_rank, debug],
-            outputs=[daily_header, weekly_header, daily_whiffs, daily_velos, weekly_whiffs, weekly_velos, prev_day_btn, next_day_btn, prev_week_btn, next_week_btn]
+            outputs=[daily_header, weekly_header, daily_whiffs, daily_velos, weekly_whiffs, weekly_velos, daily_plot, weekly_plot, prev_day_btn, next_day_btn, prev_week_btn, next_week_btn]
         )
         search_btn.click(**search_kwargs)
         for btn, fn in (