Update app.py

app.py

@@ -43,6 +43,16 @@ percentages_format = {'Shots': '{:.2%}', 'HDCF': '{:.2%}', 'Goals': '{:.2%}', 'A
 def init_baselines():
     sh = gcservice_account.open_by_url(NHL_data)
     
+    worksheet = sh.worksheet('Player_Level_ROO')
+    raw_display = pd.DataFrame(worksheet.get_values())
+    raw_display.columns = raw_display.iloc[0]
+    raw_display = raw_display[1:]
+    raw_display = raw_display.reset_index(drop=True)
+    raw_display = raw_display[raw_display['Opp'] != ""]
+    team_frame = raw_display[['Team', 'Opp']]
+    team_list = team_frame['Team'].unique()
+    team_dict = dict(zip(team_frame['Team'], team_frame['Opp']))
+    
     worksheet = sh.worksheet('Matchups')
     raw_display = pd.DataFrame(worksheet.get_values())
     raw_display.columns = raw_display.iloc[0]
@@ -67,19 +77,25 @@ def init_baselines():
     overall_ms[data_cols] = overall_ms[data_cols].apply(pd.to_numeric, errors='coerce')
     overall_ms = overall_ms.sort_values(by='Shots', ascending=False)
 
-    return matchups, overall_ms
+    return matchups, overall_ms, team_list, team_dict
 
 def convert_df_to_csv(df):
     return df.to_csv().encode('utf-8')
 
-matchups, overall_ms = init_baselines()
+matchups, overall_ms, team_list, team_dict = init_baselines()
 
 col1, col2 = st.columns([1, 9])
 with col1:
     if st.button("Reset Data", key='reset1'):
               st.cache_data.clear()
-              matchups, overall_ms = init_baselines()
+              matchups, overall_ms, team_list, team_dict = init_baselines()
     split_var1 = st.radio("View matchups or line marketshares?", ('Slate Matchups', 'Line Marketshares'), key='split_var1')
+    if split_var1 == "Line Marketshares":
+        team_var = st.radio("View all teams or specific teams?", ('All Teams', 'Specific Teams'), key='team_var')
+        if team_var == "All Teams":
+            team_split = team_list
+        elif team_var == "Specific Teams":
+            team_split = st.multiselect('Which teams would you like to include in the tables?', options = team_list, key='team_var1')
     
 with col2:
     if split_var1 == 'Slate Matchups':