Add showdown slate definitions and update lineup initialization in Streamlit app

src/streamlit_app.py

@@ -22,6 +22,17 @@ fd_hb_columns = ['QB', 'RB1', 'RB2', 'WR1', 'WR2', 'WR3', 'TE', 'FLEX', 'DST']
 dk_sd_columns = ['CPT', 'FLEX1', 'FLEX2', 'FLEX3', 'FLEX4', 'FLEX5', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']
 fd_sd_columns = ['CPT', 'FLEX1', 'FLEX2', 'FLEX3', 'FLEX4', 'FLEX5', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']
 
+showdown_selections = ['Showdown #1', 'Showdown #2', 'Showdown #3', 'Showdown #4', 'Showdown #5', 'Showdown #6', 'Showdown #7', 'Showdown #8', 'Showdown #9', 'Showdown #10', 'Showdown #11', 'Showdown #12', 'Showdown #13', 'Showdown #14', 'Showdown #15']
+dk_db_showdown_selections = ['DK_NFL_SD_seed_frame_Showdown #1', 'DK_NFL_SD_seed_frame_Showdown #2', 'DK_NFL_SD_seed_frame_Showdown #3', 'DK_NFL_SD_seed_frame_Showdown #4', 'DK_NFL_SD_seed_frame_Showdown #5', 'DK_NFL_SD_seed_frame_Showdown #6',
+'DK_NFL_SD_seed_frame_Showdown #7', 'DK_NFL_SD_seed_frame_Showdown #8', 'DK_NFL_SD_seed_frame_Showdown #9', 'DK_NFL_SD_seed_frame_Showdown #10', 'DK_NFL_SD_seed_frame_Showdown #11', 'DK_NFL_SD_seed_frame_Showdown #12', 'DK_NFL_SD_seed_frame_Showdown #13',
+'DK_NFL_SD_seed_frame_Showdown #14', 'DK_NFL_SD_seed_frame_Showdown #15']
+fd_db_showdown_selections = ['FD_NFL_SD_seed_frame_Showdown #1', 'FD_NFL_SD_seed_frame_Showdown #2', 'FD_NFL_SD_seed_frame_Showdown #3', 'FD_NFL_SD_seed_frame_Showdown #4', 'FD_NFL_SD_seed_frame_Showdown #5', 'FD_NFL_SD_seed_frame_Showdown #6',
+'FD_NFL_SD_seed_frame_Showdown #7', 'FD_NFL_SD_seed_frame_Showdown #8', 'FD_NFL_SD_seed_frame_Showdown #9', 'FD_NFL_SD_seed_frame_Showdown #10', 'FD_NFL_SD_seed_frame_Showdown #11', 'FD_NFL_SD_seed_frame_Showdown #12', 'FD_NFL_SD_seed_frame_Showdown #13',
+'FD_NFL_SD_seed_frame_Showdown #14', 'FD_NFL_SD_seed_frame_Showdown #15']
+
+dk_showdown_db_translation = dict(zip(showdown_selections, dk_db_showdown_selections))
+fd_showdown_db_translation = dict(zip(showdown_selections, fd_db_showdown_selections))
+
 # Probably should have done this in a dictionary to start with
 wrong_team_names = ['Denver Broncos', 'Washington Commanders', 'Cincinnati Bengals', 'Arizona Cardinals', 'Los Angeles Rams', 'Pittsburgh Steelers',
 'Jacksonville Jaguars', 'New England Patriots', 'Tampa Bay Buccaneers', 'San Francisco 49ers', 'Green Bay Packers', 'New York Jets',
@@ -69,6 +80,40 @@ st.markdown("""
 
 </style>""", unsafe_allow_html=True)
 
+@st.cache_resource(ttl=60)
+def define_dk_showdown_slates():
+    collection = db["DK_SD_NFL_ROO"]
+    cursor = collection.find()
+    raw_display = pd.DataFrame(list(cursor))
+    unique_slates = raw_display['slate'].unique()
+
+    slate_names = []
+
+    for slate in unique_slates:
+        slate_data = raw_display[raw_display['slate'] == slate]
+        slate_name = slate_data.iloc[0]['Team'] + ' vs. ' + slate_data.iloc[0]['Opp']
+        slate_names.append(slate_name)
+    
+    slate_name_lookup = dict(zip(slate_names, unique_slates))
+    return slate_names, slate_name_lookup
+
+@st.cache_resource(ttl=60)
+def define_fd_showdown_slates():
+    collection = db["FD_SD_NFL_ROO"]
+    cursor = collection.find()
+    raw_display = pd.DataFrame(list(cursor))
+    unique_slates = raw_display['slate'].unique()
+
+    slate_names = []
+
+    for slate in unique_slates:
+        slate_data = raw_display[raw_display['slate'] == slate]
+        slate_name = slate_data.iloc[0]['Team'] + ' vs. ' + slate_data.iloc[0]['Opp']
+        slate_names.append(slate_name)
+    
+    slate_name_lookup = dict(zip(slate_names, unique_slates))
+    return slate_names, slate_name_lookup
+
 @st.cache_resource(ttl=60)
 def init_handbuilder_data(site_var):
     """Load only the data needed for handbuilder"""
@@ -174,7 +219,15 @@ def init_baselines():
     return player_stats, dk_stacks_raw, fd_stacks_raw, dk_roo_raw, fd_roo_raw, dk_sd_roo_raw, fd_sd_roo_raw, dk_id_map, fd_id_map, dk_sd_id_map, fd_sd_id_map
 
 @st.cache_resource(ttl = 60)
-def init_DK_lineups(type_var, slate_var, prio_var, prio_mix):  
+def init_DK_lineups(type_var, slate_var, prio_var, prio_mix, db_translation):  
+
+    if type_var == 'Showdown':
+        collection = db[db_translation[slate_var]]
+        cursor = collection.find()
+        raw_display = pd.DataFrame(list(cursor))
+        raw_display = raw_display[['CPT', 'FLEX1', 'FLEX2', 'FLEX3', 'FLEX4', 'FLEX5', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']]
+        for column in ['CPT', 'FLEX1', 'FLEX2', 'FLEX3', 'FLEX4', 'FLEX5']:
+            raw_display[column] = raw_display[column].map(dict(zip(wrong_team_names, right_name_teams)), na_action='ignore').fillna(raw_display[column])
 
     if prio_var == 'Mix':
         prio_var = None
@@ -418,13 +471,16 @@ def convert_hb_df(array, column_names):
     array = pd.DataFrame(array, columns=column_names)
     return array.to_csv().encode('utf-8')
 
+slate_names_dk, slate_name_lookup_dk = define_dk_showdown_slates()
+slate_names_fd, slate_name_lookup_fd = define_fd_showdown_slates()
+
 app_load_reset_column, app_view_site_column = st.columns([1, 9])
 with app_load_reset_column:
     if st.button("Load/Reset Data", key='reset_data_button'):
         st.cache_data.clear()
         player_stats, dk_stacks_raw, fd_stacks_raw, dk_roo_raw, fd_roo_raw, dk_sd_roo_raw, fd_sd_roo_raw, dk_id_map, fd_id_map, dk_sd_id_map, fd_sd_id_map = init_baselines()
-        dk_lineups = init_DK_lineups('Regular', 'Main', 'proj', 50)
-        fd_lineups = init_FD_lineups('Regular', 'Main', 'proj', 50)
+        dk_lineups = init_DK_lineups('Regular', slate_names_dk[0], 'proj', 50, dk_showdown_db_translation)
+        fd_lineups = init_FD_lineups('Regular', slate_names_fd[0], 'proj', 50, fd_showdown_db_translation)
         for key in st.session_state.keys():
             del st.session_state[key]
 with app_view_site_column:
@@ -921,19 +977,19 @@ if selected_tab == 'Player ROO':
         if slate_type_var2 == 'Regular':
             slate_var2 = st.radio("Which data are you loading?", ('Main Slate', 'Secondary Slate', 'Late Slate', 'Thurs-Mon Slate'), key='slate_var2_radio')
         elif slate_type_var2 == 'Showdown':
-            slate_var2 = st.radio("Which slate are you loading?", ("Showdown #1", "Showdown #2", "Showdown #3"), key='slate_var2_radio')
+            slate_var2 = st.radio("Which slate are you loading?", (slate_names_dk if site_var == 'Draftkings' else slate_names_fd), key='slate_var2_radio')
         if site_var == 'Draftkings':
             if slate_type_var2 == 'Regular':
                 raw_baselines = dk_roo_raw[dk_roo_raw['slate'] == str(slate_var2)]
             elif slate_type_var2 == 'Showdown':
-                raw_baselines = dk_sd_roo_raw[dk_sd_roo_raw['slate'] == str(slate_var2)]
+                raw_baselines = dk_sd_roo_raw[dk_sd_roo_raw['slate'] == str(slate_name_lookup_dk[slate_var2])]
             
             raw_baselines = raw_baselines.iloc[:,:-2]
         elif site_var == 'Fanduel':
             if slate_type_var2 == 'Regular':
                 raw_baselines = fd_roo_raw[fd_roo_raw['slate'] == str(slate_var2)]
             elif slate_type_var2 == 'Showdown':
-                raw_baselines = fd_sd_roo_raw[fd_sd_roo_raw['slate'] == str(slate_var2)]
+                raw_baselines = fd_sd_roo_raw[fd_sd_roo_raw['slate'] == str(slate_name_lookup_fd[slate_var2])]
             raw_baselines = raw_baselines.iloc[:,:-2]
 
         split_var2 = st.radio("Would you like to view the whole slate or just specific games?", ('Full Slate Run', 'Specific Games'), key='split_var2_radio')
@@ -1037,17 +1093,17 @@ if selected_tab == 'Optimals':
                     raw_baselines = dk_sd_roo_raw
                 elif site_var == 'Fanduel':
                     raw_baselines = fd_sd_roo_raw
-            slate_var3 = st.radio("Which slate data are you loading?", ('Main', 'Secondary', 'Auxiliary'), key='slate_var3_radio')
+            slate_var3 = st.radio("Which slate data are you loading?", (slate_names_dk if slate_type_var3 == 'Showdown' else ['Main', 'Secondary', 'Auxiliary']), key='slate_var3_radio')
             if slate_type_var3 == 'Regular':
                 if site_var == 'Draftkings':
-                    dk_lineups = init_DK_lineups(slate_type_var3, slate_var3, prio_var, prio_mix)
+                    dk_lineups = init_DK_lineups(slate_type_var3, slate_var3, prio_var, prio_mix, dk_showdown_db_translation)
                 elif site_var == 'Fanduel':
-                    fd_lineups = init_FD_lineups(slate_type_var3, slate_var3, prio_var, prio_mix)
+                    fd_lineups = init_FD_lineups(slate_type_var3, slate_var3, prio_var, prio_mix, fd_showdown_db_translation)
             elif slate_type_var3 == 'Showdown':
                 if site_var == 'Draftkings':
-                    dk_lineups = init_DK_lineups(slate_type_var3, slate_var3, prio_var, prio_mix)
+                    dk_lineups = init_DK_lineups(slate_type_var3, slate_name_lookup_dk[slate_var3], prio_var, prio_mix, dk_showdown_db_translation)
                 elif site_var == 'Fanduel':
-                    fd_lineups = init_FD_lineups(slate_type_var3, slate_var3, prio_var, prio_mix)
+                    fd_lineups = init_FD_lineups(slate_type_var3, slate_name_lookup_fd[slate_var3], prio_var, prio_mix, fd_showdown_db_translation)
         with optimals_macro_col:
             lineup_num_var = st.number_input("How many lineups do you want to display?", min_value=1, max_value=1000, value=150, step=1, key='lineup_num_var_input')
             player_var1 = st.radio("Do you want a frame with specific Players?", ('Full Slate', 'Specific Players'), key='player_var1_radio')