Added framework for play-by-play data to be used in rotations.

src/streamlit_app.py

@@ -25,8 +25,10 @@ season_data_cols = ['Touches', 'Touch/Min', 'Pts', 'FGM', 'FGA', 'FG%', 'FG3M',
                     'FG3%', 'FTM', 'FTA', 'FT%', 'OREB Chance', 'OREB', 'DREB Chance', 'DREB', 'REB Chance', 'REB',
                     'Passes', 'Alt Assists', 'FT Assists', 'Assists', 'Stl', 'Blk', 'Tov', 'PF', 'DD', 'TD', 'Fantasy', 'FD_Fantasy',
                     'FPPM', 'Rebound%', 'Assists/Pass', 'Fantasy/Touch', 'FD Fantasy/Touch']
-game_rot_cols = ['PLAYER_NAME', 'backlog_lookup', 'spread', 'MIN', 'PTS', 'FGM', 'FGA', 'FG3M', 'FG3A', 'FTM', 'FTA', 'REB', 'AST', 'STL', 'BLK', 'TOV', 'PF',
+game_rot_cols_base = ['PLAYER_NAME', 'backlog_lookup', 'spread', 'MIN', 'PTS', 'FGM', 'FGA', 'FG3M', 'FG3A', 'FTM', 'FTA', 'REB', 'AST', 'STL', 'BLK', 'TOV', 'PF',
                 'Fantasy', 'FD_Fantasy']
+game_rot_cols_stint = ['stint_PTS', 'stint_FGM', 'stint_FGA', 'stint_FG3M', 'stint_REB', 'stint_AST', 'stint_STL', 'stint_BLK', 'stint_TOV', 'stint_PF', 'stint_Fantasy', 'stint_FD_Fantasy']
+game_rot_cols = game_rot_cols_base + game_rot_cols_stint
 
 percentages_format = {'PG': '{:.2%}', 'SG': '{:.2%}', 'SF': '{:.2%}', 'PF': '{:.2%}', 'C': '{:.2%}'}
 
@@ -172,6 +174,42 @@ def init_baselines(data_req: str):
         game_rot[data_cols] = game_rot[data_cols].apply(pd.to_numeric, errors='coerce')
         game_rot['spread'] = game_rot['GAME_ID'].map(spread_dict)
         game_rot['GAME_DATE'] = pd.to_datetime(game_rot['GAME_DATE']).dt.date
+        
+        # Query play-by-play stint stats
+        try:
+            collection = db["playbyplay_stints"]
+            cursor = collection.find()
+            stint_stats = pd.DataFrame(list(cursor))
+            
+            if len(stint_stats) > 0:
+                # Create stint_id in game_rot for merging
+                game_rot['stint_id'] = (game_rot['GAME_ID'].astype(str) + '_' + 
+                                        game_rot['PERSON_ID'].astype(str) + '_' + 
+                                        game_rot['Start'].astype(str))
+                
+                # Select columns to merge from stint_stats
+                stint_cols = ['stint_id', 'stint_PTS', 'stint_FGM', 'stint_FGA', 'stint_FG3M', 'stint_FG3A',
+                             'stint_FTM', 'stint_FTA', 'stint_REB', 'stint_OREB', 'stint_DREB',
+                             'stint_AST', 'stint_STL', 'stint_BLK', 'stint_TOV', 'stint_PF',
+                             'stint_Fantasy', 'stint_FD_Fantasy']
+                
+                # Keep only columns that exist
+                available_cols = [col for col in stint_cols if col in stint_stats.columns]
+                stint_stats_subset = stint_stats[available_cols]
+                
+                # Convert stint stats to numeric
+                numeric_cols = [col for col in available_cols if col != 'stint_id']
+                stint_stats_subset[numeric_cols] = stint_stats_subset[numeric_cols].apply(pd.to_numeric, errors='coerce')
+                
+                # Merge stint stats with rotations
+                game_rot = game_rot.merge(stint_stats_subset, on='stint_id', how='left')
+                
+                # Fill NaN stint stats with 0
+                for col in numeric_cols:
+                    if col in game_rot.columns:
+                        game_rot[col] = game_rot[col].fillna(0)
+        except Exception as e:
+            print(f"Note: Could not load stint stats: {e}")
     
     return gamelog_table, game_rot, timestamp
 
@@ -744,7 +782,10 @@ if selected_tab == 'Game Rotations':
             fig.add_vline(x=24, line_width=3, line_dash="dash", line_color="green")
             fig.add_vline(x=36, line_width=3, line_dash="dash", line_color="green")
 
-            game_rot_stats = check_rotation.reindex(game_rot_cols,axis="columns")
+            # Dynamically build columns list based on what exists in data
+            available_cols = [col for col in game_rot_cols_base if col in check_rotation.columns]
+            available_cols += [col for col in game_rot_cols_stint if col in check_rotation.columns]
+            game_rot_stats = check_rotation.reindex(available_cols, axis="columns")
             game_rot_stats = game_rot_stats.drop_duplicates(subset='backlog_lookup')
 
             # pages = pages.set_index('Player')
@@ -765,7 +806,10 @@ if selected_tab == 'Game Rotations':
             stats_disp = st.container()
             check_rotation = working_data[working_data['backlog_lookup'] == game_id_var]
             check_rotation = check_rotation.sort_values(by='Start', ascending=True)
-            game_rot_stats = check_rotation.reindex(game_rot_cols,axis="columns")
+            # Dynamically build columns list based on what exists in data
+            available_cols = [col for col in game_rot_cols_base if col in check_rotation.columns]
+            available_cols += [col for col in game_rot_cols_stint if col in check_rotation.columns]
+            game_rot_stats = check_rotation.reindex(available_cols, axis="columns")
             game_rot_stats = game_rot_stats.drop_duplicates(subset='PLAYER_NAME')
 
             # Create figure