Add features to daily pitcher leaderboard

daily_pitcher_leaderboard.py

@@ -1,9 +1,12 @@
 import gradio as gr
+from gradio_calendar import Calendar
 
 from data import df, game_df
 from gradio_function import *
 from css import css
 
+import datetime
+
 df = (
     df
     .join(game_df, on='game_pk').with_columns(pl.col('game_date').str.to_datetime())
@@ -14,42 +17,126 @@ df = (
 )
 
 
-def get_pitcher_leaderboards(datetime):
-    _df = df.filter(pl.col('game_date') == datetime)
+def get_pitcher_leaderboards(date, top_players, strict, ignore_zero_whiffs, show_rank, debug):
+    _df = df.filter(pl.col('game_date') == date)
+
+    other_cols = ['Name']
+    
+    if debug:
+        other_cols = ['game_date'] + other_cols
+        
     whiffs = (
         _df
         .group_by(['pitcher'])
         .agg(
             pl.col('whiff').sum().alias('Whiffs'),
-            pl.col('Name').first()
+            *[pl.col(col).first() for col in other_cols]
         )
-        .select('Name', 'Whiffs')
+        .select(*other_cols, 'Whiffs')
         .sort('Whiffs', descending=True)
-        [:10]
     )
+    if ignore_zero_whiffs:
+        whiffs = whiffs.filter(pl.col('Whiffs') > 0)
+    if len(whiffs) >top_players:
+        whiffs = (
+            whiffs
+            .filter(pl.col('Whiffs') >= whiffs['Whiffs'][top_players])
+        )
+    if strict:
+        whiffs = whiffs[:top_players]
+    if show_rank:
+        whiffs = (
+            whiffs
+            .with_row_index(offset=1)
+            .rename({'index': 'Rank'})
+        )
+    
     velos = (
         _df
-        .select('Name', 'Velocity')
+        .select(*other_cols, 'Velocity')
         .drop_nulls()
         .sort(['Velocity', 'Name'], descending=[True, False])
-        [:10]
     )
-    
-    return f'<center><h1>Daily Leaderboard<h1><h2>{datetime.strftime("%B %d, %Y")}</h2></center>', whiffs, velos
+    if len(velos) > top_players:
+        velos = velos.filter(pl.col('Velocity') >= velos['Velocity'][top_players])
+    if strict:
+        velos = velos[:top_players]
+    if show_rank:
+        velos = (
+            velos
+            .with_row_index(offset=1)
+            .rename({'index': 'Rank'})
+        )
+        
+    return (
+        f'<center><h1>Daily Leaderboard<h1><h2>{date.strftime("%B %d, %Y")}</h2><h3>{date.strftime("%A")}</h3></center>',
+        whiffs,
+        velos,
+        gr.update(interactive=True),
+        gr.update(interactive=True)
+    )
+
+
+def go_back_day(date):
+    return date - datetime.timedelta(days=1)
+
+def go_forward_day(date):
+    return date + datetime.timedelta(days=1)
     
 def create_daily_pitcher_leaderboard():
     with gr.Blocks(
         css=css
     ) as demo:
-        date_picker = gr.DateTime(value=df['game_date'].max().strftime('%Y-%m-%d'), include_time=False, type='datetime', label='Date')
-        search_btn = gr.Button('Search')
+        with gr.Row():
+            # date_picker = gr.DateTime(
+                # value=df['game_date'].max().strftime('%Y-%m-%d'),
+                # include_time=False,
+                # type='datetime',
+                # label='Date',
+                # scale=4
+            # )
+            date_picker = Calendar(
+                value=df['game_date'].max().strftime('%Y-%m-%d'),
+                type='datetime',
+                label='Date',
+                scale=2,
+                min_width=50
+            )
+            top_players = gr.Number(10, label='# Top players', scale=1, min_width=100)
+            strict = gr.Checkbox(False, label='Strict', info='Ignore ties and restrict to # top players', scale=1.5, min_width=100)
+            ignore_zero_whiffs = gr.Checkbox(False, label='Ignore zero whiffs', info='Ignore zero whiff players if in top ranked', scale=1.5, min_width=100)
+            show_rank = gr.Checkbox(False, label='Show rank', scale=1, min_width=100)
+            debug = gr.Checkbox(False, label='Debug', info='Show dates', scale=1, min_width=100)
+            search_btn = gr.Button('Search', scale=1, min_width=100)
+
+        with gr.Row():
+            prev_btn = gr.Button('Previous', interactive=False)
+            next_btn = gr.Button('Next', interactive=False)
+
 
-        header = gr.HTML('<center><h1>Daily Leaderboard<h2><br></center>')
+        header = gr.HTML('<center><h1>Daily Leaderboard<h1><h2 style="display: none;"></h2><h3 style="display: none;"></h3></center>')
         with gr.Row():
-            whiffs = gr.Dataframe(pl.DataFrame({'Name': [], 'Whiffs': []}), label='Whiffs')
-            velos = gr.Dataframe(pl.DataFrame({'Name': [], 'Velocity': []}), label='Velocity')
+            whiffs = gr.Dataframe(pl.DataFrame({'Name': [], 'Whiffs': []}), label='Whiffs', interactive=False)
+            velos = gr.Dataframe(pl.DataFrame({'Name': [], 'Velocity': []}), label='Velocity', interactive=False)
 
-        search_btn.click(get_pitcher_leaderboards, date_picker, [header, whiffs, velos])
+        search_kwargs = dict(
+            fn=get_pitcher_leaderboards,
+            inputs=[date_picker, top_players, strict, ignore_zero_whiffs, show_rank, debug],
+            outputs=[header, whiffs, velos, prev_btn, next_btn]
+        )
+        search_btn.click(**search_kwargs)
+        (
+            prev_btn
+            .click(go_back_day, date_picker, date_picker)
+            .then(**search_kwargs)
+        )
+        (
+            next_btn
+            .click(go_forward_day, date_picker, date_picker)
+            .then(**search_kwargs)
+        )
+        
+        
         
     return demo