Refactor app.py: Enhance data initialization and timestamp handling. Updated init_baselines to return both processed data and timestamp, ensuring better data management. Adjusted references to the updated function throughout the code, including site selection logic and display updates.

app.py

@@ -32,14 +32,18 @@ def init_baselines():
     collection = db["PGA_Range_of_Outcomes"] 
     cursor = collection.find()
     player_frame = pd.DataFrame(cursor)
-    roo_data = player_frame
     
-    return roo_data
+    timestamp = player_frame['Timestamp'][0]
+
+    roo_data = player_frame.drop(columns=['_id', 'index', 'Timestamp'])
+    roo_data['Salary'] = roo_data['Salary'].astype(int)
+    
+    return roo_data, timestamp
 
 def convert_df_to_csv(df):
     return df.to_csv().encode('utf-8')
 
-roo_data = init_baselines()
+roo_data, timestamp = init_baselines()
 hold_display = roo_data
 lineup_display = []
 check_list = []
@@ -54,18 +58,19 @@ with tab1:
               # Clear values from *all* all in-memory and on-disk data caches:
               # i.e. clear values from both square and cube
               st.cache_data.clear()
-              roo_data = init_baselines()
+              roo_data, timestamp = init_baselines()
               hold_display = roo_data
               lineup_display = []
               check_list = []
               rand_player = 0
               boost_player = 0
               salaryCut = 0
+    st.write(timestamp)
     options_container = st.empty()
     hold_container = st.empty()
 
     with options_container:
-        site_var = st.selectbox("Select a Site", ["DraftKings", "FanDuel"])
+        site_var = st.selectbox("Select a Site", ["Draftkings", "FanDuel"])
         
     with hold_container:
         display = hold_display[hold_display['Site'] == site_var]