Update app.py to include new column mappings for DraftKings and FanDuel showdown formats, enhancing data display and ensuring accurate representation of team and player metrics in lineups.

app.py

@@ -25,6 +25,8 @@ player_roo_format = {'Top_finish': '{:.2%}','Top_5_finish': '{:.2%}', 'Top_10_fi
 
 dk_columns = ['SP1', 'SP2', 'C', '1B', '2B', '3B', 'SS', 'OF1', 'OF2', 'OF3', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']
 fd_columns = ['P', 'C_1B', '2B', '3B', 'SS', 'OF1', 'OF2', 'OF3', 'UTIL', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']
+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', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']
 
 st.markdown("""
 <style>
@@ -145,19 +147,19 @@ def init_DK_lineups(type_var, slate_var):
             cursor = collection.find().limit(10000)
 
             raw_display = pd.DataFrame(list(cursor))
-            raw_display = raw_display[['CPT', 'FLEX1', 'FLEX2', 'FLEX3', 'FLEX4', 'FLEX5', 'salary', 'proj', 'Own']]
+            raw_display = raw_display[['CPT', 'FLEX1', 'FLEX2', 'FLEX3', 'FLEX4', 'FLEX5', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']]
         elif slate_var == 'Secondary':
             collection = db2['DK_MLB_SD2_seed_frame']
             cursor = collection.find().limit(10000)
 
             raw_display = pd.DataFrame(list(cursor))
-            raw_display = raw_display[['CPT', 'FLEX1', 'FLEX2', 'FLEX3', 'FLEX4', 'FLEX5', 'salary', 'proj', 'Own']]
+            raw_display = raw_display[['CPT', 'FLEX1', 'FLEX2', 'FLEX3', 'FLEX4', 'FLEX5', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']]
         elif slate_var == 'Auxiliary':
             collection = db2['DK_MLB_SD3_seed_frame']
             cursor = collection.find().limit(10000)
 
             raw_display = pd.DataFrame(list(cursor))
-            raw_display = raw_display[['CPT', 'FLEX1', 'FLEX2', 'FLEX3', 'FLEX4', 'FLEX5', 'salary', 'proj', 'Own']]
+            raw_display = raw_display[['CPT', 'FLEX1', 'FLEX2', 'FLEX3', 'FLEX4', 'FLEX5', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']]
 
     DK_seed = raw_display.to_numpy()
 
@@ -216,19 +218,19 @@ def init_FD_lineups(type_var,slate_var):
                 cursor = collection.find().limit(10000)
 
                 raw_display = pd.DataFrame(list(cursor))
-                raw_display = raw_display[['CPT', 'FLEX1', 'FLEX2', 'FLEX3', 'FLEX4', 'salary', 'proj', 'Own']]
+                raw_display = raw_display[['CPT', 'FLEX1', 'FLEX2', 'FLEX3', 'FLEX4', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']]
             elif slate_var == 'Secondary':
                 collection = db2['FD_MLB_SD2_seed_frame']
                 cursor = collection.find().limit(10000)
 
                 raw_display = pd.DataFrame(list(cursor))
-                raw_display = raw_display[['CPT', 'FLEX1', 'FLEX2', 'FLEX3', 'FLEX4', 'salary', 'proj', 'Own']]
+                raw_display = raw_display[['CPT', 'FLEX1', 'FLEX2', 'FLEX3', 'FLEX4', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']]
             elif slate_var == 'Auxiliary':
                 collection = db2['FD_MLB_SD3_seed_frame']
                 cursor = collection.find().limit(10000)
             
                 raw_display = pd.DataFrame(list(cursor))
-                raw_display = raw_display[['CPT', 'FLEX1', 'FLEX2', 'FLEX3', 'FLEX4', 'salary', 'proj', 'Own']]
+                raw_display = raw_display[['CPT', 'FLEX1', 'FLEX2', 'FLEX3', 'FLEX4', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']]
     
         FD_seed = raw_display.to_numpy()
 
@@ -406,7 +408,7 @@ with tab3:
         lineup_num_var = st.number_input("How many lineups do you want to display?", min_value=1, max_value=1000, value=150, step=1)
         
         if slate_type_var3 == 'Regular':
-                raw_baselines = roo_data
+            raw_baselines = roo_data
         elif slate_type_var3 == 'Showdown':
             raw_baselines = sd_roo_data
         
@@ -414,12 +416,14 @@ with tab3:
             if slate_type_var3 == 'Regular':
                 ROO_slice = raw_baselines[raw_baselines['Site'] == 'Draftkings']
                 player_salaries = dict(zip(ROO_slice['Player'], ROO_slice['Salary']))
+                column_names = dk_columns
             elif slate_type_var3 == 'Showdown':
                 player_salaries = dict(zip(raw_baselines['Player'], raw_baselines['Salary']))
+                column_names = dk_sd_columns
             # Get the minimum and maximum ownership values from dk_lineups
             min_own = np.min(dk_lineups[:,8])
             max_own = np.max(dk_lineups[:,8])
-            column_names = dk_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':
@@ -432,11 +436,14 @@ with tab3:
             if slate_type_var3 == 'Regular':
                 ROO_slice = raw_baselines[raw_baselines['Site'] == 'Fanduel']
                 player_salaries = dict(zip(ROO_slice['Player'], ROO_slice['Salary']))
+                column_names = fd_columns
             elif slate_type_var3 == 'Showdown':
                 player_salaries = dict(zip(raw_baselines['Player'], raw_baselines['Salary']))
+                column_names = fd_sd_columns
+            # Get the minimum and maximum ownership values from dk_lineups
             min_own = np.min(fd_lineups[:,8])
             max_own = np.max(fd_lineups[:,8])
-            column_names = fd_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':