Refactor data export logic in app.py to streamline column dropping and index setting for DraftKings and FanDuel. This update enhances the clarity of exported data by ensuring consistent handling across different slate types, improving overall user experience.

app.py

@@ -655,31 +655,31 @@ 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').drop(columns=['SP1'])
+                        data_export = data_export.drop(columns=['salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']).set_index('SP1')
                     elif slate_type_var3 == 'Showdown':
-                        data_export = data_export.drop(columns=['salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']).set_index('CPT').drop(columns=['CPT'])
+                        data_export = data_export.drop(columns=['salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']).set_index('CPT')
                 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').drop(columns=['P'])
+                        data_export = data_export.drop(columns=['salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']).set_index('P')
                     elif slate_type_var3 == 'Showdown':
-                        data_export = data_export.drop(columns=['salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']).set_index('CPT').drop(columns=['CPT'])
+                        data_export = data_export.drop(columns=['salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']).set_index('CPT')
                 st.download_button(
                     label="Portfolio Manager Export (IDs)",
-                    data=convert_df(data_export.drop(columns=['salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']).set_index('SP1').drop(columns=['SP1'])),
+                    data=convert_df(data_export),
                     file_name='MLB_optimals_export.csv',
                     mime='text/csv',
                 )
                 
                 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').drop(columns=['SP1'])
+                        name_export = name_export.drop(columns=['salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']).set_index('SP1')
                     elif slate_type_var3 == 'Showdown':
-                        name_export = name_export.drop(columns=['salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']).set_index('CPT').drop(columns=['CPT'])
+                        name_export = name_export.drop(columns=['salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']).set_index('CPT')
                 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').drop(columns=['P'])
+                        name_export = name_export.drop(columns=['salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']).set_index('P')
                     elif slate_type_var3 == 'Showdown':
-                        name_export = name_export.drop(columns=['salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']).set_index('CPT').drop(columns=['CPT'])
+                        name_export = name_export.drop(columns=['salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']).set_index('CPT')
                 st.download_button(
                     label="Portfolio Manager Export (Names)",
                     data=convert_df(name_export),