adding a dataframe export function to start creating some ownership databases

app.py

@@ -104,6 +104,39 @@ def export_contest_file(db, sport, type, contest_date, contest_id, contest_data)
 
     return return_message
 
+def export_exposure_frame(db, sport, type, contest_date, contest_id, exposure_frame):
+    if type == 'Classic':
+        db_type = 'reg'
+    elif type == 'Showdown':
+        db_type = 'showdown'
+    collection = db[f'{sport}_{db_type}_exposure_frames']
+    try:
+        cursor = collection.find()
+        exposure_frame_import = pd.DataFrame(list(cursor)).drop('_id', axis=1)
+        if contest_id in exposure_frame_import['Contest ID'].values:
+            return_message = "Data for this contest already exists, no need to upload, but we appreciate the effort!"
+            return return_message
+    except:
+        exposure_frame_import = pd.DataFrame(columns = ['Player', 'Pos', 'Salary', 'Actual', 'Exposure Overall', 'Exposure Top 1%', 'Exposure Top 5%', 'Exposure Top 10%', 'Exposure Top 20%', 'Contest Date', 'Contest ID'])
+    
+    exposure_frame_import['Contest Date'] = contest_date
+    exposure_frame_import['Contest ID'] = contest_id
+    exposure_frame_import = pd.concat([exposure_frame_import, exposure_frame], ignore_index=True)
+
+    chunk_size = 100000
+    collection.drop()
+    for i in range(0, len(exposure_frame_import), chunk_size):
+        for _ in range(5):
+            try:
+                df_chunk = exposure_frame_import.iloc[i:i + chunk_size]
+                collection.insert_many(df_chunk.to_dict('records'), ordered=False)
+                break
+            except Exception as e:
+                print(f"Retry due to error: {e}")
+    return_message = "Contest data uploaded successfully! We appreciate the data!"
+
+    return return_message
+
 def get_payout_for_position(finish_pos, payout_df, dupes_count):
     """
     Calculate payout for a position, handling ties by splitting the combined payout.
@@ -812,7 +845,13 @@ if selected_tab == 'Contest Analysis':
                             format(formatter='{:.2f}', subset=st.session_state['player_frame'].iloc[:, 2:4].select_dtypes(include=['number']).columns).
                             format(formatter='{:.2%}', subset=st.session_state['player_frame'].iloc[:, 4:].select_dtypes(include=['number']).columns),
                             hide_index=True)
-                        st.download_button(label="Download Player Info", data=st.session_state['player_frame'].to_csv(index=False), file_name="player_info.csv", mime="text/csv", key='player_exp_comp_download')
+                        player_exposure_download_col, player_exposure_export_col, player_exposure_blank_col = st.columns([2, 2, 8])
+                        with player_exposure_download_col:
+                            st.download_button(label="Download Player Info", data=st.session_state['player_frame'].to_csv(index=False), file_name="player_info.csv", mime="text/csv", key='player_exp_comp_download')
+                        with player_exposure_export_col:
+                            if st.button("Export Player Info"):
+                                return_message = export_exposure_frame(db, st.session_state['sport_select'], st.session_state['type_init'], st.session_state['contest_date'], st.session_state['contest_id'], st.session_state['player_frame'])
+                                st.write(return_message)
                     else:
                         st.session_state['player_frame'] = create_player_exposures(st.session_state['display_contest_info'], st.session_state['player_columns'], st.session_state['entry_names'])
                         hold_frame = st.session_state['player_frame'].copy()