Update stuff_model/calculate_arm_angles.py

stuff_model/calculate_arm_angles.py

@@ -12,7 +12,8 @@ def calculate_arm_angles(df: pl.DataFrame,pitcher_id:int) -> pl.DataFrame:
     }
     
     response = requests.get(url, headers=headers)
-    
+
+    daily_check = df['game_date'][0] == df['game_date'][-1]
     
     # Assuming response.text contains the CSV formatted string
     csv_data = response.text
@@ -24,6 +25,7 @@ def calculate_arm_angles(df: pl.DataFrame,pitcher_id:int) -> pl.DataFrame:
     df_arm_angle = pl.read_csv(data)
 
     if pitcher_id not in df_arm_angle["pitcher"]:
+        old_data = True
         df_arm_angle = pl.read_csv('stuff_model/pitcher_arm_angles_2024.csv')
 
         
@@ -84,10 +86,21 @@ def calculate_arm_angles(df: pl.DataFrame,pitcher_id:int) -> pl.DataFrame:
             )
             #.drop(["Opp", "arm_angle_rad"])
             )
-    
-        df_filter = df_filter.with_columns(
-                ((pl.col("arm_angle") * 0.0) + (ball_angle * 1)).alias("arm_angle")
-            )
+        
+        if old_data:
+            df_filter = df_filter.with_columns(
+                    ((pl.col("arm_angle") * 0.5) + (ball_angle * 0.5)).alias("arm_angle")
+                )
+            
+        elif daily_check:
+            df_filter = df_filter.with_columns(
+                    ((pl.col("arm_angle") * 0.25) + (ball_angle * 0.75)).alias("arm_angle")
+                )
+            
+        else:
+            df_filter = df_filter.with_columns(
+                    ((pl.col("arm_angle") * 0.0) + (ball_angle * 1)).alias("arm_angle")
+                )