Update app

demo.py

@@ -17,6 +17,9 @@ css = '''
 
 .pitch-loc {height: 256px}
 .pitch-loc .js-plotly-plot {height: 100%}
+
+.pitch-velo-summary {height: 256px}
+.pitch-velo-summary .js-plotly-plot {height: 100%}
 '''
 # .pitch-col {width: 256px !important}
 # .pitch-col .gr-group {width: 256px}
@@ -72,16 +75,20 @@ with gr.Blocks(css=css) as demo:
             pitch_velos.append(gr.Plot(label='Velocity distribution', elem_classes='pitch-velo', visible=visible))
             pitch_maps.append(gr.Plot(label='Pitch location', elem_classes='pitch-loc', visible=visible))
 
+  gr.Markdown('## Pitch Velocity')
+  pitch_velo_summary = gr.Plot(label='Velocity distribution', elem_classes='pitch-velo-summary')
+
   gr.Markdown('## Bugs and other notes')
   with gr.Accordion('Click to open', open=False):
     gr.Markdown('''
     - Whiff% and CSW% has not been verified
     - No padding in pie charts leads to hovertext getting cut off near the bottom for some players
     - Y axis ticks messy when no velocity distribution is plotted
+    - No padding in pitch velo chart leads to top row being hard to read
     '''
     )
 
-  player.input(get_data, inputs=player, outputs=[player_info, usage, *pitch_groups, *pitch_names, *pitch_infos, *pitch_velos, *pitch_maps])
+  player.input(get_data, inputs=player, outputs=[player_info, usage, *pitch_groups, *pitch_names, *pitch_infos, *pitch_velos, *pitch_maps, pitch_velo_summary])
 
 demo.launch(
     share=True

gradio_function.py

@@ -90,6 +90,7 @@ pitch_stats = pd.merge(
     on=['name', 'pitch_name']
 ).set_index(['name', 'pitch_name'])
 
+
 # GRADIO FUNCTIONS
 
 # location maps
@@ -199,7 +200,7 @@ def plot_pitch_velo(player=None, velos=None, pitch_type=None, pitch_name=None):
     pitch_col = 'pitch_type' if pitch_type else 'pitch_name'
     velos = df.set_index(['name', pitch_col]).loc[(player, pitch_val), 'release_speed']
 
-  fig = go.Figure(data=go.Violin(x=velos, side='positive', hoveron='points', name='Velocity Distribution'))
+  fig = go.Figure(data=go.Violin(x=velos, side='positive', hoveron='points', points=False, meanline_visible=True, name='Velocity Distribution'))
   fig.update_layout(
     xaxis=dict(
         title='Velocity',
@@ -221,6 +222,7 @@ def plot_pitch_velo(player=None, velos=None, pitch_type=None, pitch_name=None):
   )
   return fig
 
+
 def plot_empty_pitch_velo():
   fig = go.Figure()
   fig.add_annotation(
@@ -251,6 +253,90 @@ def plot_empty_pitch_velo():
   return fig
 
 
+def plot_all_pitch_velo(player=None, player_df=None, pitch_counts=None, min_pitches=10):
+  # assert not ((player is None and player_df is None) or (player is not None and player_df is not None)), 'exactly one of `player` or `player_df` must be specified'
+  
+  if player_df is None and player is not None:
+    assert pitch_counts is None, '`pitch_counts` must be `None` if `player_df` is None'
+    player_df = df.sort_values('name').set_index('name').loc[player].sort_values('pitch_name').set_index('pitch_name')
+    pitch_counts = player_df.index.value_counts(ascending=True)
+
+  league_df = df.set_index('pitch_name')
+
+  fig = go.Figure()
+  
+  velo_center = (player_df['release_speed'].min() + player_df['release_speed'].max()) / 2
+  for i, (pitch_name, count) in enumerate(pitch_counts.items()):
+    velos = player_df.loc[pitch_name, 'release_speed']
+    league_velos = league_df.loc[pitch_name, 'release_speed']
+    fig.add_trace(go.Violin(
+        x=league_velos,
+        y=[pitch_name]*len(league_velos),
+        line_color='gray',
+        side='positive',
+        orientation='h',
+        meanline_visible=True,
+        points=False,
+        legendgroup='NPB',
+        legendrank=1,
+        # visible='legendonly',
+        showlegend=False,
+        name='NPB',
+    ))
+    if count >= min_pitches:
+      fig.add_trace(go.Violin(
+          x=velos,
+          y=[pitch_name]*len(velos),
+          side='positive',
+          orientation='h',
+          meanline_visible=True,
+          points=False,
+          legendgroup=pitch_name,
+          legendrank=2+i,
+          name=pitch_name
+      ))
+    else:
+      fig.add_trace(go.Scatter(
+          x=[velo_center],
+          y=[pitch_name],
+          text=['No visualization as less than 10 pitches thrown'],
+          textposition='top center',
+          hovertext=False,
+          mode="lines+text",
+          legendgroup=pitch_name,
+          legendrank=2+i,
+          name=pitch_name,
+      ))
+
+  fig.add_trace(go.Violin(
+      x=player_df['release_speed'],
+      y=[player]*len(player_df),
+      side='positive',
+      orientation='h',
+      meanline_visible=True,
+      points=False,
+      legendrank=0,
+      name=player
+  ))
+  fig.add_trace(go.Violin(
+      x=league_df['release_speed'],
+      y=[player]*len(league_df),
+      line_color='gray',
+      side='positive',
+      orientation='h',
+      meanline_visible=True,
+      points=False,
+      legendgroup='NPB',
+      legendrank=1,
+      # visible='legendonly',
+      name='NPB',
+  ))
+
+  fig.update_xaxes(title='Velocity')
+  
+  return fig
+
+
 def get_data(player):
   player_name = f'# {player}'
 
@@ -297,5 +383,6 @@ def get_data(player):
     pitch_velos.append(gr.update(value=None, visible=False))
     pitch_maps.append(gr.update(value=None, visible=False))
 
+  pitch_velo_summary = plot_all_pitch_velo(player=player, player_df=_df_by_pitch_name, pitch_counts=pitch_counts)
 
-  return player_name, usage_fig, *pitch_groups, *pitch_names, *pitch_infos, *pitch_velos, *pitch_maps
+  return player_name, usage_fig, *pitch_groups, *pitch_names, *pitch_infos, *pitch_velos, *pitch_maps, pitch_velo_summary