Update app.py

app.py

@@ -138,13 +138,16 @@ def convert_df_to_csv(df):
     return df.to_csv().encode('utf-8')
 
 gamelog_table = init_baselines()
-total_teams = gamelog_table.Team.values.tolist()
+total_teams = gamelog_table.Team.values.unique().tolist()
+total_dates = gamelog_table.Date.values.unique().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()
     
     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')
@@ -153,10 +156,23 @@ with col1:
         team_var1 = st.multiselect('Which teams would you like to include in the tables?', options = total_teams, key='team_var1')
     elif split_var2 == 'All':
         team_var1 = total_teams
+        
+    split_var3 = st.radio("Would you like to view all teams or specific ones?", ('All', 'Specific Dates'), key='split_var3')
+    
+    if split_var3 == 'Specific Dates':
+        low_date = st.date_input('Min Date:', value=None, format="YYYY-MM-DD", key='low_date')
+        high_date = st.date_input('Max Date:', value=None, format="YYYY-MM-DD", key='high_date')
+    elif split_var3 == 'All':
+        low_date = min(total_dates)
+        high_date = max(total_dates)
+    
+    min_var1 = st.slider("Is there a certain minutes range you want to view?", 0, 60, (0, 60), key='min_var1')
 
 with col2:
     if split_var1 == 'Season Logs':
         display = st.container()
+        gamelog_table = gamelog_table[gamelog_table['Min'] >= min_var1[0]]
+        gamelog_table = gamelog_table[gamelog_table['Min'] <= min_var1[1]]
         gamelog_table = gamelog_table[gamelog_table['Team'].isin(team_var1)]
         season_long_table = seasonlong_build(gamelog_table)
         season_long_table = season_long_table.set_index('Player')
@@ -177,6 +193,10 @@ with col2:
             )
         with bottom_menu[0]:
             st.markdown(f"Page **{current_page}** of **{total_pages}** ")
+        gamelog_table = gamelog_table[gamelog_table['Date'] >= low_date]
+        gamelog_table = gamelog_table[gamelog_table['Date'] <= high_date]
+        gamelog_table = gamelog_table[gamelog_table['Min'] >= min_var1[0]]
+        gamelog_table = gamelog_table[gamelog_table['Min'] <= min_var1[1]]
         gamelog_table = gamelog_table[gamelog_table['Team'].isin(team_var1)]
         gamelog_table = gamelog_table.set_index('Player')
         pages = split_frame(gamelog_table, batch_size)