Update app.py

app.py

@@ -90,6 +90,7 @@ def init_baselines():
     data_cols = ['PG', 'SG', 'SF', 'PF', 'C']
     rot_table[data_cols] = rot_table[data_cols].apply(pd.to_numeric, errors='coerce')
     
+    
     return gamelog_table, rot_table
 
 @st.cache_data(show_spinner=False)
@@ -204,6 +205,8 @@ season_data_cols = ['Touches', 'Touch/Min', 'Pts', 'FGM', 'FGA', 'FG%', 'FG3M',
                     'Rebound%', 'Assists/Pass', 'Fantasy/Touch', 'FD Fantasy/Touch']
 indv_teams = gamelog_table.drop_duplicates(subset='Team')
 total_teams = indv_teams.Team.values.tolist()
+indv_rot_teams = rot_table.drop_duplicates(subset='Team')
+total_rot_teams = indv_rot_teams.Team.values.tolist()
 indv_players = gamelog_table.drop_duplicates(subset='Player')
 total_players = indv_players.Player.values.tolist()
 total_dates = gamelog_table.Date.values.tolist()
@@ -228,6 +231,8 @@ with tab1:
                                       'Rebound%', 'Assists/Pass', 'Fantasy/Touch', 'FD Fantasy/Touch']
                   indv_teams = gamelog_table.drop_duplicates(subset='Team')
                   total_teams = indv_teams.Team.values.tolist()
+                  indv_rot_teams = rot_table.drop_duplicates(subset='Team')
+                  total_rot_teams = indv_rot_teams.Team.values.tolist()
                   indv_players = gamelog_table.drop_duplicates(subset='Player')
                   total_players = indv_players.Player.values.tolist()
                   total_dates = gamelog_table.Date.values.tolist()
@@ -344,6 +349,8 @@ with tab2:
                                       'Rebound%', 'Assists/Pass', 'Fantasy/Touch', 'FD Fantasy/Touch']
                   indv_teams = gamelog_table.drop_duplicates(subset='Team')
                   total_teams = indv_teams.Team.values.tolist()
+                  indv_rot_teams = rot_table.drop_duplicates(subset='Team')
+                  total_rot_teams = indv_rot_teams.Team.values.tolist()
                   indv_players = gamelog_table.drop_duplicates(subset='Player')
                   total_players = indv_players.Player.values.tolist()
                   total_dates = gamelog_table.Date.values.tolist()
@@ -426,6 +433,8 @@ with tab3:
                                       'Rebound%', 'Assists/Pass', 'Fantasy/Touch', 'FD Fantasy/Touch']
                   indv_teams = gamelog_table.drop_duplicates(subset='Team')
                   total_teams = indv_teams.Team.values.tolist()
+                  indv_rot_teams = rot_table.drop_duplicates(subset='Team')
+                  total_rot_teams = indv_rot_teams.Team.values.tolist()
                   indv_players = gamelog_table.drop_duplicates(subset='Player')
                   total_players = indv_players.Player.values.tolist()
                   total_dates = gamelog_table.Date.values.tolist()
@@ -502,11 +511,13 @@ with tab4:
                                       'Rebound%', 'Assists/Pass', 'Fantasy/Touch', 'FD Fantasy/Touch']
                   indv_teams = gamelog_table.drop_duplicates(subset='Team')
                   total_teams = indv_teams.Team.values.tolist()
+                  indv_rot_teams = rot_table.drop_duplicates(subset='Team')
+                  total_rot_teams = indv_rot_teams.Team.values.tolist()
                   indv_players = gamelog_table.drop_duplicates(subset='Player')
                   total_players = indv_players.Player.values.tolist()
                   total_dates = gamelog_table.Date.values.tolist()
         
-        team_var4 = st.selectbox('Which team would you like to view?', options = total_teams, key='team_var4')
+        team_var4 = st.selectbox('Which team would you like to view?', options = total_rot_teams, key='team_var4')
     
     with col2:
         working_data = rot_table