Update app.py

app.py

@@ -71,14 +71,17 @@ def grab_baseline_stuff():
     
     worksheet = sh.worksheet('Timestamp')
     timestamp = worksheet.acell('A1').value
+    
+    worksheet = sh.worksheet('Player_Lines_ROO')
+    line_frame = pd.DataFrame(worksheet.get_all_records())
 
-    return dk_raw_proj, fd_raw_proj, dk_ids, fd_ids, timestamp
+    return dk_raw_proj, fd_raw_proj, dk_ids, fd_ids, timestamp, line_frame
 
 @st.cache_data
 def convert_df_to_csv(df):
     return df.to_csv().encode('utf-8')
 
-dk_raw_proj, fd_raw_proj, dkid_dict, fdid_dict, timestamp = grab_baseline_stuff()
+dk_raw_proj, fd_raw_proj, dkid_dict, fdid_dict, timestamp, line_frame = grab_baseline_stuff()
 t_stamp = f"Last Update: " + str(timestamp) + f" CST"
 opp_dict = dict(zip(dk_raw_proj.Team, dk_raw_proj.Opp))
 
@@ -106,7 +109,7 @@ with tab2:
         st.info(t_stamp)
         if st.button("Load/Reset Data", key='reset1'):
               st.cache_data.clear()
-              dk_raw_proj, fd_raw_proj, dk_ids, fd_ids, timestamp = grab_baseline_stuff()
+              dk_raw_proj, fd_raw_proj, dk_ids, fd_ids, timestamp, line_frame = grab_baseline_stuff()
               t_stamp = f"Last Update: " + str(timestamp) + f" CST"
               for key in st.session_state.keys():
                   del st.session_state[key]
@@ -170,6 +173,7 @@ with tab2:
         raw_baselines['lock'] = np.where(raw_baselines['Player'].isin(lock_var1), 1, 0)
         st.session_state.export_baselines = raw_baselines.copy()
         st.session_state.display_baselines = raw_baselines.copy()
+        st.session_state.display_lines = line_frame.copy()
         
         display_container = st.empty()
         display_dl_container = st.empty()
@@ -583,7 +587,11 @@ with tab2:
         with display_container:
             display_container = st.empty()
             if 'display_baselines' in st.session_state:
-                st.dataframe(st.session_state.display_baselines.style.background_gradient(axis=0).background_gradient(cmap='RdYlGn').format(precision=2), use_container_width = True)
+                tab1, tab2 = st.tabs(['Line Combo ROO', 'Player Projections'])
+                with tab1:
+                    st.dataframe(st.session_state.display_lines.style.background_gradient(axis=0).background_gradient(cmap='RdYlGn').format(precision=2), use_container_width = True)
+                with tab2:
+                    st.dataframe(st.session_state.display_baselines.style.background_gradient(axis=0).background_gradient(cmap='RdYlGn').format(precision=2), use_container_width = True)
 
         with display_dl_container:
                     display_dl_container = st.empty()