Add download options for ROO data in Streamlit app, allowing users to export both Regular and Portfolio Manager formats. Streamlined data preparation for exports based on selected slate type, enhancing user experience and data accessibility.

src/streamlit_app.py

@@ -215,15 +215,38 @@ with tab1:
             site_var = st.radio("Select a Site", ["Draftkings", "FanDuel"])
         with col3:
             type_var = st.radio("Select a Type", ["Full Slate", "Showdown"])
+            if type_var == "Full Slate":
+                display = hold_display[hold_display['Site'] == site_var]
+                display = display.drop_duplicates(subset=['Player'])
+            elif type_var == "Showdown":
+                display = sd_roo_data
+                display = display.drop_duplicates(subset=['Player'])
     
+    export_data = display.copy()
+    export_data_pm = display[['Player', 'Salary', 'Median', 'Own']]
+    export_data_pm['Position'] = 'G'
+    export_data_pm['Team'] = 'Golf'
+    export_data_pm['captain ownership'] = ''
+    export_data_pm = export_data_pm.rename(columns={'Own': 'ownership', 'Median': 'median', 'Player': 'player_names', 'Position': 'position', 'Team': 'team', 'Salary': 'salary'})
+    
+    reg_dl_col, pm_dl_col, blank_col = st.columns([2, 2, 6])
+    with reg_dl_col:
+        st.download_button(
+                    label="Export ROO (Regular)",
+                    data=convert_df_to_csv(export_data),
+                    file_name='NBA_ROO_export.csv',
+                    mime='text/csv',
+        )
+    with pm_dl_col:
+        st.download_button(
+                    label="Export ROO (Portfolio Manager)",
+                    data=convert_df_to_csv(export_data_pm),
+                    file_name='NBA_ROO_export.csv',
+                    mime='text/csv',
+        )
+
     with st.container():
-        if type_var == "Full Slate":
-            display = hold_display[hold_display['Site'] == site_var]
-            display = display.drop_duplicates(subset=['Player'])
-        elif type_var == "Showdown":
-            display = sd_roo_data
-            display = display.drop_duplicates(subset=['Player'])
-            
+        
         if view_var == "Simple":
             if type_var == "Full Slate":
                 display = display[['Player', 'Cut_Odds', 'Salary', 'Median', '10x%', 'Own']]
@@ -236,13 +259,6 @@ with tab1:
             display = display
             display = display.set_index('Player')
             st.dataframe(display.style.background_gradient(axis=0).background_gradient(cmap='RdYlGn').format(player_roo_format, precision=2), height=750, use_container_width = True)
-    
-    st.download_button(
-        label="Export Projections",
-        data=convert_df_to_csv(display),
-        file_name='PGA_DFS_export.csv',
-        mime='text/csv',
-    )
 
 with tab2:
     with st.expander("Info and Filters"):