Refactor lineup initialization in app.py: remove try-except blocks for lineup creation, ensuring consistent DataFrame initialization for both Draftkings and Fanduel across Regular and Showdown slate types, enhancing code clarity and reliability.

app.py

@@ -253,13 +253,10 @@ salary_dict = dict(zip(roo_raw.Player, roo_raw.Salary))
 id_dict = dict(zip(roo_raw.Player, roo_raw.player_ID))
 salary_dict_sd = dict(zip(sd_raw.Player, sd_raw.Salary))
 id_dict_sd = dict(zip(sd_raw.Player, sd_raw.player_ID))
-
-try:
-    dk_lineups = init_DK_lineups('Main Slate')
-    fd_lineups = init_FD_lineups('Main Slate')
-except:
-    dk_lineups = pd.DataFrame(columns=dk_columns)
-    fd_lineups = pd.DataFrame(columns=fd_columns)
+dk_lineups = pd.DataFrame(columns=dk_columns)
+dk_sd_lineups = pd.DataFrame(columns=dk_sd_columns)
+fd_lineups = pd.DataFrame(columns=fd_columns)
+fd_sd_lineups = pd.DataFrame(columns=fd_sd_columns)
 t_stamp = f"Last Update: " + str(timestamp) + f" CST"
 
 tab1, tab2 = st.tabs(['Range of Outcomes', 'Optimals'])
@@ -282,13 +279,10 @@ with tab1:
                     id_dict = dict(zip(roo_raw.Player, roo_raw.player_ID))
                     salary_dict_sd = dict(zip(sd_raw.Player, sd_raw.Salary))
                     id_dict_sd = dict(zip(sd_raw.Player, sd_raw.player_ID))
-
-                    try:
-                        dk_lineups = init_DK_lineups('Main Slate')
-                        fd_lineups = init_FD_lineups('Main Slate')
-                    except:
-                        dk_lineups = pd.DataFrame(columns=dk_columns)
-                        fd_lineups = pd.DataFrame(columns=fd_columns)
+                    dk_lineups = pd.DataFrame(columns=dk_columns)
+                    dk_sd_lineups = pd.DataFrame(columns=dk_sd_columns)
+                    fd_lineups = pd.DataFrame(columns=fd_columns)
+                    fd_sd_lineups = pd.DataFrame(columns=fd_sd_columns)
                     t_stamp = f"Last Update: " + str(timestamp) + f" CST"
                     for key in st.session_state.keys():
                         del st.session_state[key]
@@ -382,13 +376,10 @@ with tab2:
             id_dict = dict(zip(roo_raw.Player, roo_raw.player_ID))
             salary_dict_sd = dict(zip(sd_raw.Player, sd_raw.Salary))
             id_dict_sd = dict(zip(sd_raw.Player, sd_raw.player_ID))
-
-            try:
-                dk_lineups = init_DK_lineups('Main Slate')
-                fd_lineups = init_FD_lineups('Main Slate')
-            except:
-                dk_lineups = pd.DataFrame(columns=dk_columns)
-                fd_lineups = pd.DataFrame(columns=fd_columns)
+            dk_lineups = pd.DataFrame(columns=dk_columns)
+            dk_sd_lineups = pd.DataFrame(columns=dk_sd_columns)
+            fd_lineups = pd.DataFrame(columns=fd_columns)
+            fd_sd_lineups = pd.DataFrame(columns=fd_sd_columns)
             t_stamp = f"Last Update: " + str(timestamp) + f" CST"
             for key in st.session_state.keys():
                 del st.session_state[key]
@@ -404,7 +395,10 @@ with tab2:
             lineup_num_var = st.number_input("How many lineups do you want to display?", min_value=1, max_value=1000, value=150, step=1)
         with col5:
             if site_var1 == 'Draftkings':
-                column_names = dk_columns
+                if slate_type_var1 == 'Regular':
+                    column_names = dk_columns
+                elif slate_type_var1 == 'Showdown':
+                    column_names = dk_sd_columns
                 
                 player_var1 = st.radio("Do you want a frame with specific Players?", ('Full Slate', 'Specific Players'), key='player_var1')
                 if player_var1 == 'Specific Players':
@@ -413,7 +407,10 @@ with tab2:
                         player_var2 = dk_raw.Player.values.tolist()
                         
             elif site_var1 == 'Fanduel':
-                column_names = fd_columns
+                if slate_type_var1 == 'Regular':
+                    column_names = fd_columns
+                elif slate_type_var1 == 'Showdown':
+                    column_names = fd_sd_columns
                 
                 player_var1 = st.radio("Do you want a frame with specific Players?", ('Full Slate', 'Specific Players'), key='player_var1')
                 if player_var1 == 'Specific Players':
@@ -443,9 +440,12 @@ with tab2:
                 st.session_state.working_seed = st.session_state.working_seed[np.equal.outer(st.session_state.working_seed, player_var2).any(axis=1).all(axis=1)]
             elif player_var1 == 'Full Slate':
                 st.session_state.working_seed = dk_lineups.copy()
-            st.session_state.data_export_display = pd.DataFrame(st.session_state.working_seed[0:lineup_num_var], columns=column_names)
+                st.session_state.data_export_display = pd.DataFrame(st.session_state.working_seed[0:lineup_num_var], columns=column_names)
         elif 'working_seed' not in st.session_state:
-            st.session_state.working_seed = dk_lineups.copy()
+            if slate_type_var1 == 'Regular':
+                st.session_state.working_seed = init_DK_lineups(slate_var1)
+            elif slate_type_var1 == 'Showdown':
+                st.session_state.working_seed = init_DK_SD_lineups(slate_var1)
             st.session_state.working_seed = st.session_state.working_seed
             if player_var1 == 'Specific Players':
                 st.session_state.working_seed = st.session_state.working_seed[np.equal.outer(st.session_state.working_seed, player_var2).any(axis=1).all(axis=1)]
@@ -462,7 +462,10 @@ with tab2:
                 st.session_state.working_seed = fd_lineups.copy()
             st.session_state.data_export_display = pd.DataFrame(st.session_state.working_seed[0:lineup_num_var], columns=column_names)
         elif 'working_seed' not in st.session_state:
-            st.session_state.working_seed = fd_lineups.copy()
+            if slate_type_var1 == 'Regular':
+                st.session_state.working_seed = init_FD_lineups(slate_var1)
+            elif slate_type_var1 == 'Showdown':
+                st.session_state.working_seed = init_FD_SD_lineups(slate_var1)
             st.session_state.working_seed = st.session_state.working_seed
             if player_var1 == 'Specific Players':
                 st.session_state.working_seed = st.session_state.working_seed[np.equal.outer(st.session_state.working_seed, player_var2).any(axis=1).all(axis=1)]
@@ -472,20 +475,34 @@ with tab2:
 
     export_file = st.session_state.data_export_display.copy()
     if site_var1 == 'Draftkings':
-        for col_idx in range(8):
-            export_file.iloc[:, col_idx] = export_file.iloc[:, col_idx].map(id_dict)
+        if slate_type_var1 == 'Regular':
+            for col_idx in range(8):
+                export_file.iloc[:, col_idx] = export_file.iloc[:, col_idx].map(id_dict)
+        elif slate_type_var1 == 'Showdown':
+            for col_idx in range(5):
+                export_file.iloc[:, col_idx] = export_file.iloc[:, col_idx].map(id_dict_sd)
     elif site_var1 == 'Fanduel':
-        for col_idx in range(9):
-            export_file.iloc[:, col_idx] = export_file.iloc[:, col_idx].map(id_dict)
+        if slate_type_var1 == 'Regular':
+            for col_idx in range(9):
+                export_file.iloc[:, col_idx] = export_file.iloc[:, col_idx].map(id_dict)
+        elif slate_type_var1 == 'Showdown':
+            for col_idx in range(5):
+                export_file.iloc[:, col_idx] = export_file.iloc[:, col_idx].map(id_dict_sd)
             
     with st.container():
         if st.button("Reset Optimals", key='reset3'):
             for key in st.session_state.keys():
                 del st.session_state[key]
             if site_var1 == 'Draftkings':
-                st.session_state.working_seed = dk_lineups.copy()
+                if slate_type_var1 == 'Regular':
+                    st.session_state.working_seed = dk_lineups.copy()
+                elif slate_type_var1 == 'Showdown':
+                    st.session_state.working_seed = dk_sd_lineups.copy()
             elif site_var1 == 'Fanduel':
-                st.session_state.working_seed = fd_lineups.copy()
+                if slate_type_var1 == 'Regular':
+                    st.session_state.working_seed = fd_lineups.copy()
+                elif slate_type_var1 == 'Showdown':
+                    st.session_state.working_seed = fd_sd_lineups.copy()
         if 'data_export_display' in st.session_state:
             st.dataframe(st.session_state.data_export_display.style.background_gradient(axis=0).background_gradient(cmap='RdYlGn').format(precision=2), height=500, use_container_width = True)
         st.download_button(