Refactor data export logic in app.py to ensure consistent column dropping and index setting for DraftKings and FanDuel across slate types. This update improves data clarity and maintains a streamlined export process for both Portfolio Manager options.

app.py

@@ -655,14 +655,14 @@ with tab3:
             with pm_opt_col:
                 if site_var == 'Draftkings':
                     if slate_type_var3 == 'Regular':
-                        data_export = data_export.drop(columns=['salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']).set_index('SP1')
+                        data_export = data_export.set_index('SP1').drop(columns=['SP1', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own'], axis=1)
                     elif slate_type_var3 == 'Showdown':
-                        data_export = data_export.drop(columns=['salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']).set_index('CPT')
+                        data_export = data_export.set_index('CPT').drop(columns=['CPT', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own'], axis=1)
                 elif site_var == 'Fanduel':
                     if slate_type_var3 == 'Regular':
-                        data_export = data_export.drop(columns=['salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']).set_index('P')
+                        data_export = data_export.set_index('P').drop(columns=['P', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own'], axis=1)
                     elif slate_type_var3 == 'Showdown':
-                        data_export = data_export.drop(columns=['salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']).set_index('CPT')
+                        data_export = data_export.set_index('CPT').drop(columns=['CPT', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own'], axis=1)
                 st.download_button(
                     label="Portfolio Manager Export (IDs)",
                     data=convert_df(data_export),
@@ -672,14 +672,14 @@ with tab3:
                 
                 if site_var == 'Draftkings':
                     if slate_type_var3 == 'Regular':
-                        name_export = name_export.drop(columns=['salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']).set_index('SP1')
+                        name_export = name_export.set_index('SP1').drop(columns=['SP1', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own'], axis=1)
                     elif slate_type_var3 == 'Showdown':
-                        name_export = name_export.drop(columns=['salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']).set_index('CPT')
+                        name_export = name_export.set_index('CPT').drop(columns=['CPT', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own'], axis=1)
                 elif site_var == 'Fanduel':
                     if slate_type_var3 == 'Regular':
-                        name_export = name_export.drop(columns=['salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']).set_index('P')
+                        name_export = name_export.set_index('P').drop(columns=['P', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own'], axis=1)
                     elif slate_type_var3 == 'Showdown':
-                        name_export = name_export.drop(columns=['salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']).set_index('CPT')
+                        name_export = name_export.set_index('CPT').drop(columns=['CPT', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own'], axis=1)
                 st.download_button(
                     label="Portfolio Manager Export (Names)",
                     data=convert_df(name_export),