Update app.py

app.py

@@ -61,13 +61,30 @@ def df_final(df:pl.dataframe,year_input:int,sport_id:int):
 
 
     print('schedule')
-
+    # Calculate mound_to_release as 60.5 - extension
+    df = df.with_columns([
+        (60.5 - df["extension"]).alias("release_pos_y")
+    ])
+    
+    # Calculate delta time (Δt)
+    delta_t = (df["release_pos_y"] - df["y0"]) / df["vy0"]
+
+    # Corrected back-calculation of release_pos_x and release_pos_z
+    df = df.with_columns([
+        (df["x0"] + df["vx0"] * delta_t + 0.5 * df["ax"] * delta_t ** 2).alias("release_pos_x"),
+        (df["z0"] + df["vz0"] * delta_t + 0.5 * df["az"] * delta_t ** 2).alias("release_pos_z")
+    ])
+    
     df_stuff = stuff_apply.stuff_apply(fe.feature_engineering(df))
     print('stuff')
     df_up = update.update(df)
     print('update')
     df_total = df_up.join(df_stuff[['play_id','tj_stuff_plus']], on='play_id', how='left')
     print('total')
+
+
+
+    
     return df_total