Update app

data.py

@@ -41,6 +41,21 @@ pitch_df = pl.concat(pitch_df)
 player_df = pl.read_csv('player.csv')
 
 # translate pa data
+
+def identify_bb_type(hit_type):
+  if hit_type in list(range(1, 10)) + list(range(40, 49)):
+    return 'ground_ball'
+  elif hit_type in list(range(58, 67))+list(range(201, 209)):
+    return 'line_drive'
+  elif hit_type in list(range(28, 31)) + list(range(55, 58)) + list(range(107, 110)) + list(range(247, 251)):
+    return 'fly_ball'
+  elif hit_type in list(range(49, 55)) + list(range(103, 107)) + list(range(242, 248)):
+    return 'pop_up'
+  elif hit_type in [31, 32]:
+    return None
+  else:
+    raise Exception(f'Unexpect hit_type {hit_type}')
+
 # pa_df['_des'] = pa_df['des'].str.strip()
 # pa_df['des'] = pa_df['des'].str.strip()
 # pa_df['des_more'] = pa_df['des_more'].str.strip()
@@ -84,6 +99,12 @@ pa_df = (
     .with_columns(
         pl.col('des').map_elements(translate_pa_outcome, return_dtype=str)
     )
+    .with_columns(
+        pl.col('bb_type').alias('hit_type').str.strip_prefix('dakyu').cast(int).alias('hit_type')
+    )
+    .with_columns(
+        pl.col('hit_type').map_elements(lambda hit_type: identify_bb_type(hit_type), return_dtype=str).alias('bb_type')
+    )
 )
 
 # translate pitch data

gradio_function.py

@@ -203,7 +203,8 @@ def plot_velo_summary(df, league_df, player):
         legendgroup='NPB',
         legendrank=1,
         # visible='legendonly',
-        showlegend=False,
+        # showlegend=False,
+        showlegend=i==0,
         name='NPB',
     ))
     if count >= min_pitches:
@@ -215,7 +216,7 @@ def plot_velo_summary(df, league_df, player):
           meanline_visible=True,
           points=False,
           legendgroup=pitch_name,
-          legendrank=2+(len(pitch_counts) - i),
+          legendrank=len(pitch_counts) - i, #2+(len(pitch_counts) - i),
           name=pitch_name
       ))
     else:
@@ -227,37 +228,38 @@ def plot_velo_summary(df, league_df, player):
           hovertext=False,
           mode="lines+text",
           legendgroup=pitch_name,
-          legendrank=2+(len(pitch_counts) - i),
+          legendrank=len(pitch_counts) - i, #2+(len(pitch_counts) - i),
           name=pitch_name,
       ))
 
-  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.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.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.update_xaxes(title='Velocity', range=[player_df['release_speed'].dropna().min() - 2, player_df['release_speed'].dropna().max() + 2])
   fig.update_xaxes(title='Velocity', range=[player_df['release_speed'].min() - 2, player_df['release_speed'].max() + 2])
-  fig.update_yaxes(range=[0, len(pitch_counts)+1-0.25], visible=False)
+  # fig.update_yaxes(range=[0, len(pitch_counts)+1-0.25], visible=False)
+  fig.update_yaxes(range=[0, len(pitch_counts)-0.25], visible=False)
   fig.update_layout(
       violingap=0,
       violingroupgap=0,