Update app.py

app.py

@@ -138,16 +138,20 @@ def convert_df_to_csv(df):
     return df.to_csv().encode('utf-8')
 
 gamelog_table = init_baselines()
-total_teams = gamelog_table.Team.values.unique().tolist()
-total_dates = gamelog_table.Date.values.unique().tolist()
+indv_teams = gamelog_table.drop_duplicates(subset='Team')
+total_teams = indv_teams.Team.values.tolist()
+indv_dates = gamelog_table.drop_duplicates(subset='Dates')
+total_dates = indv_dates.Date.values.tolist()
 
 col1, col2 = st.columns([1, 9])
 with col1:
     if st.button("Reset Data", key='reset1'):
               st.cache_data.clear()
               gamelog_table = init_baselines()
-              total_teams = gamelog_table.Team.values.unique().tolist()
-              total_dates = gamelog_table.Date.values.unique().tolist()
+              indv_teams = gamelog_table.drop_duplicates(subset='Team')
+              total_teams = indv_teams.Team.values.tolist()
+              indv_dates = gamelog_table.drop_duplicates(subset='Dates')
+              total_dates = indv_dates.Date.values.tolist()
     
     split_var1 = st.radio("What table would you like to view?", ('Season Logs', 'Gamelogs'), key='split_var1')
     split_var2 = st.radio("Would you like to view all teams or specific ones?", ('All', 'Specific Teams'), key='split_var2')