Update app

demo.py

@@ -18,12 +18,12 @@ css = '''
 .pitch-loc {height: 256px}
 .pitch-loc .js-plotly-plot {height: 100%}
 
-.pitch-velo-summary {height: 256px}
+.pitch-velo-summary {height: 384px}
 .pitch-velo-summary .js-plotly-plot {height: 100%}
 '''
-# .pitch-col {width: 256px !important}
-# .pitch-col .gr-group {width: 256px}
-# .pitch-col .js-plotly-plot {width: 100%}
+  # display: flex;
+  # align-items: center;
+  # justify-content: center;
 
 with gr.Blocks(css=css) as demo:
   gr.Markdown('''
@@ -72,11 +72,11 @@ with gr.Blocks(css=css) as demo:
           with pitch_group:
             pitch_names.append(gr.Markdown(f'### Pitch {col+1}', visible=visible))
             pitch_infos.append(gr.DataFrame(pd.DataFrame([{'Whiff%': None, 'CSW%': None}]), interactive=False, visible=visible))
-            pitch_velos.append(gr.Plot(label='Velocity distribution', elem_classes='pitch-velo', visible=visible))
+            pitch_velos.append(gr.Plot(show_label=False, 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')
+  pitch_velo_summary = gr.Plot(show_label=False, elem_classes='pitch-velo-summary')
 
   gr.Markdown('## Bugs and other notes')
   with gr.Accordion('Click to open', open=False):
@@ -84,7 +84,6 @@ with gr.Blocks(css=css) as demo:
     - 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
     '''
     )
 

gradio_function.py

@@ -229,7 +229,7 @@ def plot_empty_pitch_velo():
       x=(170+125)/2,
       y=0.3/2,
       text='No visualization<br>as less than 10 pitches thrown',
-      showarrow=False
+      showarrow=False,
   )
   fig.update_layout(
     xaxis=dict(
@@ -242,8 +242,10 @@ def plot_empty_pitch_velo():
         range=[0, 0.3],
         scaleanchor='x',
         scaleratio=1,
-        tickvals=np.linspace(0, 0.3, 3),
-        ticktext=np.linspace(0, 0.3, 3),
+        # tickvals=np.linspace(0, 0.3, 3),
+        # ticktext=np.linspace(0, 0.3, 3),
+        tickvals=[0.15],
+        ticktext=[0.15]
     ),
     autosize=True,
     # width=512,
@@ -255,7 +257,7 @@ def plot_empty_pitch_velo():
 
 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')
@@ -264,7 +266,7 @@ def plot_all_pitch_velo(player=None, player_df=None, pitch_counts=None, min_pitc
   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']
@@ -292,7 +294,7 @@ def plot_all_pitch_velo(player=None, player_df=None, pitch_counts=None, min_pitc
           meanline_visible=True,
           points=False,
           legendgroup=pitch_name,
-          legendrank=2+i,
+          legendrank=2+(len(pitch_counts) - i),
           name=pitch_name
       ))
     else:
@@ -304,7 +306,7 @@ def plot_all_pitch_velo(player=None, player_df=None, pitch_counts=None, min_pitc
           hovertext=False,
           mode="lines+text",
           legendgroup=pitch_name,
-          legendrank=2+i,
+          legendrank=2+(len(pitch_counts) - i),
           name=pitch_name,
       ))
 
@@ -333,7 +335,7 @@ def plot_all_pitch_velo(player=None, player_df=None, pitch_counts=None, min_pitc
   ))
 
   fig.update_xaxes(title='Velocity')
-  
+
   return fig
 
 
@@ -383,6 +385,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)
+  pitch_velo_summary = plot_all_pitch_velo(player=player, player_df=_df_by_pitch_name, pitch_counts=pitch_counts.sort_values(ascending=True))
 
   return player_name, usage_fig, *pitch_groups, *pitch_names, *pitch_infos, *pitch_velos, *pitch_maps, pitch_velo_summary