Add new convert_pm_df function for Portfolio Manager exports in app.py. Update download buttons to utilize this new function, enhancing data export functionality and ensuring consistent CSV formatting for both IDs and Names.

app.py

@@ -323,6 +323,11 @@ def convert_df(array):
     array = pd.DataFrame(array, columns=column_names)
     return array.to_csv().encode('utf-8')
 
+@st.cache_data
+def convert_pm_df(array):
+    array = pd.DataFrame(array)
+    return array.to_csv().encode('utf-8')
+
 col1, col2 = st.columns([1, 9])
 with col1:
     if st.button("Load/Reset Data", key='reset'):
@@ -665,7 +670,7 @@ with tab3:
                         data_export = data_export.set_index('CPT').drop(columns=['salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own'], axis=1)
                 st.download_button(
                     label="Portfolio Manager Export (IDs)",
-                    data=convert_df(data_export),
+                    data=convert_pm_df(data_export),
                     file_name='MLB_optimals_export.csv',
                     mime='text/csv',
                 )
@@ -682,7 +687,7 @@ with tab3:
                         name_export = name_export.set_index('CPT').drop(columns=['salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own'], axis=1)
                 st.download_button(
                     label="Portfolio Manager Export (Names)",
-                    data=convert_df(name_export),
+                    data=convert_pm_df(name_export),
                     file_name='MLB_optimals_export.csv',
                     mime='text/csv',
                 )