Add showdown slate selections and update lineup initialization to use new database translations for DraftKings and FanDuel, improving data handling for showdown formats.

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',
@@ -208,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
@@ -452,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:
@@ -469,9 +491,6 @@ with app_view_site_column:
         with app_site_column:
             site_var = st.selectbox("What site do you want to view?", ('Draftkings', 'Fanduel'), key='site_selectbox')
 
-slate_names_dk, slate_name_lookup_dk = define_dk_showdown_slates()
-slate_names_fd, slate_name_lookup_fd = define_fd_showdown_slates()
-
 selected_tab = st.segmented_control(
     "Select Tab",
     options=["Player ROO", "Stacks ROO", "Handbuilder", "Optimals"],
@@ -1074,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')