Update app.py

app.py

@@ -18,8 +18,10 @@ from matplotlib.gridspec import GridSpec
 import matplotlib.pyplot as plt
 from scipy.stats import gaussian_kde
 
+season = 2024
+
 ### Import Datasets
-dataset = load_dataset('nesticot/mlb_data', data_files=['mlb_pitch_data_2024.csv'])
+dataset = load_dataset('nesticot/mlb_data', data_files=[f'mlb_pitch_data_{season}.csv'])
 dataset_train = dataset['train']
 df_2023 = dataset_train.to_pandas().set_index(list(dataset_train.features.keys())[0]).reset_index(drop=True)
 # Paths to data
@@ -68,7 +70,7 @@ df_2023_bip['h_la'] = df_2023_bip['h_la'].round(0)
 
 df_2023_bip['season'] = df_2023_bip['game_date'].str[0:4].astype(int)
 
-df_2023_bip = df_2023_bip[df_2023_bip['season'] == 2024]
+df_2023_bip = df_2023_bip[df_2023_bip['season'] == {season}]
 df_2022_bip = df_2023_bip[df_2023_bip['season'] == 2022]
 
 batter_dict = df_2023_bip.sort_values('batter_name').set_index('batter_id')['batter_name'].to_dict()
@@ -84,10 +86,10 @@ def server(input,output,session):
     def plot():
 
         batter_id_select = int(input.batter_id())
-        df_batter_2023 = df_2023_bip.loc[(df_2023_bip['batter_id'] == batter_id_select)&(df_2023_bip['season']==2023)]
+        df_batter_2023 = df_2023_bip.loc[(df_2023_bip['batter_id'] == batter_id_select)&(df_2023_bip['season']=={season})]
         df_batter_2022 = df_2023_bip.loc[(df_2023_bip['batter_id'] == batter_id_select)&(df_2023_bip['season']==2022)]
 
-        df_non_batter_2023 = df_2023_bip.loc[(df_2023_bip['batter_id'] != batter_id_select)&(df_2023_bip['season']==2023)]
+        df_non_batter_2023 = df_2023_bip.loc[(df_2023_bip['batter_id'] != batter_id_select)&(df_2023_bip['season']=={season})]
         df_non_batter_2022 = df_2023_bip.loc[(df_2023_bip['batter_id'] != batter_id_select)&(df_2023_bip['season']==2022)]
 
         traj_df = df_batter_2023.groupby(['traj'])['launch_speed'].count() / len(df_batter_2023)
@@ -317,7 +319,7 @@ def server(input,output,session):
         # # ax12.text(s='Less\nOften',x=0.5,y=0.26,
         # #         va='bottom',ha='center',fontsize=12)
 
-        ax01.text(s=f"{df_batter_2023['batter_name'].values[0]}'s 2024 Batted Ball Tendencies",
+        ax01.text(s=f"{df_batter_2023['batter_name'].values[0]}'s {season} Batted Ball Tendencies",
                 x=0.5,
                 y=0.8,va='top',ha='center',fontsize=20)