Update app.py to modify the init_DK_lineups and init_FD_lineups functions by adding a slate_var parameter, enhancing flexibility in lineup initialization. Refactor the "Load/Reset Data" button implementation to improve code organization and maintainability within the main container.

app.py

@@ -82,7 +82,7 @@ def init_baselines():
     return roo_data, sd_roo_data, scoring_percentages
 
 @st.cache_data(ttl = 60)
-def init_DK_lineups():  
+def init_DK_lineups(slate_var):  
         
         collection = db2['DK_MLB_SD1_seed_frame']
         cursor = collection.find().limit(10000)
@@ -94,7 +94,7 @@ def init_DK_lineups():
         return DK_seed
 
 @st.cache_data(ttl = 60)
-def init_FD_lineups():  
+def init_FD_lineups(slate_var):  
         
         collection = db2['FD_MLB_SD1_seed_frame']
         cursor = collection.find().limit(10000)
@@ -121,14 +121,18 @@ with st.container():
     with col2:
         site_var = st.selectbox("What site do you want to view?", ('Draftkings', 'Fanduel'), key='site_var')
     with col3:
-        if st.button("Load/Reset Data", key='reset'):
-            st.cache_data.clear()
-            roo_data, sd_roo_data, scoring_percentages = init_baselines()
-            hold_display = roo_data
-            dk_lineups = init_DK_lineups('Main')
-            fd_lineups = init_FD_lineups('Main')
-            for key in st.session_state.keys():
-                del st.session_state[key]
+        container = st.container()
+        with container:
+            col = st.columns(3)[1]  # Create 3 columns and use middle one
+            with col:
+                if st.button("Load/Reset Data", key='reset'):
+                    st.cache_data.clear()
+                    roo_data, sd_roo_data, scoring_percentages = init_baselines()
+                    hold_display = roo_data
+                    dk_lineups = init_DK_lineups('Main')
+                    fd_lineups = init_FD_lineups('Main')
+                    for key in st.session_state.keys():
+                        del st.session_state[key]
 
 tab1, tab2, tab3 = st.tabs(["Scoring Percentages", "Player ROO", "Optimals"])