Update app.py

app.py

@@ -308,22 +308,23 @@ def server(input, output, session):
                                                 end_date = end_date,
                                                 game_type = [input.type_input()])
 
-        data_list = scrape.get_data(game_list_input = game_list[:])
-        df = (stuff_apply.stuff_apply(fe.feature_engineering(update.update(scrape.get_data_df(data_list = data_list).filter(
-                            (pl.col("pitcher_id") == player_input)&
-                            (pl.col("is_pitch") == True)&
-                            (pl.col('batter_hand').is_in(split_dict_hand[input.split_id()]))
-
-                            )))).with_columns(
-                pl.col('pitch_type').count().over('pitch_type').alias('pitch_count')
-            ))
-        
-        df = df.with_columns(
-            prop_percent=(pl.col('is_pitch') / pl.col('is_pitch').sum()).over("pitch_type"),
-            prop=pl.col('is_pitch').sum().over("pitch_type")
-        )
+        try:
+            df = (stuff_apply.stuff_apply(fe.feature_engineering(update.update(scrape.get_data_df(data_list = data_list).filter(
+                                (pl.col("pitcher_id") == player_input)&
+                                (pl.col("is_pitch") == True)&
+                                (pl.col("start_speed") >= 50)&
+                                (pl.col('batter_hand').is_in(split_dict_hand[input.split_id()]))
+    
+                                )))).with_columns(
+                    pl.col('pitch_type').count().over('pitch_type').alias('pitch_count')
+                ))
+            return df
 
-        return df
+            
+        except TypeError:
+            print("NONE")
+            return None
+        
 
     @render.ui
     @reactive.event(input.player_button, input.year_input, input.level_input, input.type_input,ignore_none=False)
@@ -385,8 +386,16 @@ def server(input, output, session):
     def in_brush():
         # if input.plot_brush() is None:  # Note: changed to match the brush ID
         #     return None
+
+
+        df = cached_data()
+        if df is None:
+                fig = plt.figure(figsize=(10,10))
+                fig.text(x=0.1,y=0.9,s='No Statcast Data For This Pitcher',fontsize=24,ha='left')
+                return fig
+        
         brushed_df = pl.DataFrame(brushed_points(
-            cached_data().to_pandas(),
+            df.to_pandas(),
             input.plot_brush(),
             xvar="hb",
             yvar="ivb",