Update app.py

app.py

@@ -60,11 +60,56 @@ def init_baselines():
     
     raw_display = pd.DataFrame(list(cursor))
     FD_seed = raw_display[['P', 'C_1B', '2B', '3B', 'SS', 'OF1', 'OF2', 'OF3', 'UTIL', 'Team_count', 'salary', 'proj']]
+    
+    worksheet = sh.worksheet('DK_Projections')
+    load_display = pd.DataFrame(worksheet.get_all_records())
+    load_display.replace('', np.nan, inplace=True)
+    
+    dk_raw = load_display.dropna(subset=['Median'])
+    
+    worksheet = sh.worksheet('FD_Projections')
+    load_display = pd.DataFrame(worksheet.get_all_records())
+    load_display.replace('', np.nan, inplace=True)
+    
+    fd_raw = load_display.dropna(subset=['Median'])
+
+    return DK_seed, FD_seed, dk_raw, fd_raw
 
-    return DK_seed, FD_seed
+DK_seed, FD_seed, dk_raw, fd_raw = init_baselines()
 
-DK_seed, FD_seed = init_baselines()
 
-st.dataframe(DK_seed.head(1000).style.format(precision=2), height=500, use_container_width=True)
+col1, col2 = st.columns([1, 7])
+with col1:
+    if st.button("Load/Reset Data", key='reset1'):
+          st.cache_data.clear()
+          for key in st.session_state.keys():
+              del st.session_state[key]
+          DK_seed, FD_seed, dk_raw, fd_raw = init_baselines()
+          
+    slate_var1 = st.radio("Which data are you loading?", ('Main Slate'))
+    site_var1 = st.radio("What site are you working with?", ('Draftkings', 'Fanduel'))
+    if site_var1 == 'Draftkings':
+        raw_baselines = dk_raw
+    elif site_var1 == 'Fanduel':
+        raw_baselines = fd_raw
+        
+    contest_var1 = st.selectbox("What contest size are you simulating?", ('Small', 'Medium', 'Large'))
+    if contest_var1 == 'Small':
+        Contest_Size = 1000
+    elif contest_var1 == 'Medium':
+        Contest_Size = 5000
+    elif contest_var1 == 'Large':
+        Contest_Size = 10000
+    strength_var1 = st.selectbox("How sharp is the field in the contest?", ('Not Very', 'Average', 'Very'))
+    if strength_var1 == 'Not Very':
+        sharp_split = [400000,100000] 
+    elif strength_var1 == 'Average':
+        sharp_split = [500000,200000]
+    elif strength_var1 == 'Very':
+        sharp_split = [500000,300000]
 
-st.dataframe(FD_seed.head(1000).style.format(precision=2), height=500, use_container_width=True)
+with col2:
+    if site_var1 == 'Draftkings':
+        st.dataframe(DK_seed.head(1000).style.format(precision=2), height=500, use_container_width=True)
+    elif site_var1 == 'Fanduel':
+        st.dataframe(FD_seed.head(1000).style.format(precision=2), height=500, use_container_width=True)