Enhance data export functionality in app.py by implementing conditional column dropping and index setting based on selected site and slate type for both DraftKings and FanDuel. This update ensures accurate data formatting for Portfolio Manager exports, improving user experience and data clarity.

app.py

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