Update src/streamlit_app.py

src/streamlit_app.py

@@ -126,11 +126,24 @@ def get_player_career_stats_brscraper(player_name, seasons_to_check=10):
 def get_team_season_stats_brscraper(year):
     if not BRSCRAPER_AVAILABLE:
         return pd.DataFrame()
+
     try:
         df = nba.get_stats(year, info='per_game', rename=False)
-        teams = df[df['Player'].isna() & (df['Rk'].astype(str).str.lower()!='rk')].copy()
+
+        # Ensure 'Rk' and 'Player' columns exist
+        if 'Rk' not in df.columns or 'Player' not in df.columns:
+            st.warning(f"'Rk' or 'Player' columns missing for {year}.")
+            return pd.DataFrame()
+
+        # Drop rows that are header repeats (e.g. 'Rk' == 'Rk' or 'rk')
+        df = df[df['Rk'].apply(lambda x: str(x).lower() != 'rk')]
+
+        # Keep only rows where Player is NaN (which correspond to team totals)
+        teams = df[df['Player'].isna()].copy()
         if teams.empty:
             return pd.DataFrame()
+
+        # Rename & clean columns
         mapping = {
             'G':'GP','MP':'MIN','FG%':'FG_PCT','3P%':'FG3_PCT','FT%':'FT_PCT',
             'TRB':'REB','AST':'AST','STL':'STL','BLK':'BLK','TOV':'TO',
@@ -139,13 +152,18 @@ def get_team_season_stats_brscraper(year):
             '2P':'FGM2','2PA':'FGA2','2P%':'FG2_PCT','eFG%':'EFG_PCT',
             'FT':'FTM','FTA':'FTA','ORB':'OREB','DRB':'DREB','Tm':'Team'
         }
-        teams = teams.rename(columns={o:n for o,n in mapping.items() if o in teams.columns})
-        teams['Team'] = teams['Team'].astype(str).str.replace('*','',regex=False).str.strip()
-        non_num = {'Team','RANK'}
+        teams = teams.rename(columns={o: n for o, n in mapping.items() if o in teams.columns})
+
+        if 'Team' in teams.columns:
+            teams['Team'] = teams['Team'].astype(str).str.replace('*', '', regex=False).str.strip()
+
+        non_num = {'Team', 'RANK'}
         for col in teams.columns:
             if col not in non_num:
                 teams[col] = pd.to_numeric(teams[col], errors='coerce')
+
         return teams
+
     except Exception as e:
         st.error(f"Error fetching team stats for {year}: {e}")
         return pd.DataFrame()