Update app.py

app.py

@@ -72,6 +72,8 @@ def init_baselines():
     raw_display['Team Date'] = raw_display['Team'] + " " + raw_display['Date']
     raw_display = raw_display.drop(columns=['Day of Season', 'Team', 'Opp', 'Date Num', 'DR Team', 'In Minutes File'])
     game_model = raw_display[raw_display['Injury and Rotation Adjusted Win %'] != ""]
+    just_win_probs = game_model[['Team Date', 'Time', 'Acro', 'Opponent', 'Injury and Rotation Adjusted Win %', 'Total Proj', 'Projected Points']]
+    just_win_probs['Projected Spread'] = (just_win_probs['Total Proj'] / 2) - just_win_probs['Projected Points']
     
 
     worksheet = sh.worksheet('SeasonExport')
@@ -89,12 +91,12 @@ def init_baselines():
                                'Playoff Odds', 'Division Odds', 'Top 4 Seed Odds', '1 Seed Odds', 'Win 1st Round', 'Win 2nd Round', 'Win Conference', 'Win Title']]
     seed_probs = season_model[['Team', 'Conference', 'Division', 'Avg Seed', '1', '2', '3', '4', '5', '6', '7', '8', '9', '10', '11', '12', '13', '14', '15']]
 
-    return game_model, season_model, seed_probs, title_sims
+    return game_model, season_model, seed_probs, title_sims, just_win_probs
 
 def convert_df_to_csv(df):
     return df.to_csv().encode('utf-8')
 
-game_model, season_model, seed_probs, title_sims = init_baselines()
+game_model, season_model, seed_probs, title_sims, just_win_probs = init_baselines()
 
 tab1, tab2 = st.tabs(["Game Betting Model", "Season and Futures"])
 
@@ -103,7 +105,8 @@ with tab1:
     with col1:
         if st.button("Reset Data", key='reset1'):
                   st.cache_data.clear()
-                  game_model, season_model, seed_probs, title_sims = init_baselines()
+                  game_model, season_model, seed_probs, title_sims, just_win_probs = init_baselines()
+        view_var1 = st.radio("Would you like to view math and stuff or just the win percentages and margins?", ('Just win probs', 'Gimme details'), key='view_var1')
         split_var1 = st.radio("Would you like to view all teams or specific ones?", ('All', 'Specific Teams'), key='split_var1')
         if split_var1 == 'Specific Teams':
             team_var1 = st.multiselect('Which teams would you like to include in the tables?', options = game_model['Acro'].unique(), key='team_var1')
@@ -115,23 +118,35 @@ with tab1:
         elif date_split_var1 == 'All':
             date_var1 = game_model.Date.values.tolist()
     with col2:
-        game_display = game_model[game_model['Acro'].isin(team_var1)]
-        game_display = game_display[game_display['Date'].isin(date_var1)]
-        game_display = game_display.set_index('Team Date')
-        st.dataframe(game_display.style.background_gradient(axis=0).background_gradient(cmap='RdYlGn').format(precision=2), use_container_width = True)
-        st.download_button(
-            label="Export Game Model",
-            data=convert_df_to_csv(game_model),
-            file_name='AmericanNumbers_Game_Model_export.csv',
-            mime='text/csv',
-        )
+        if view_var1 == 'Just win probs':
+            game_display = just_win_probs[just_win_probs['Acro'].isin(team_var1)]
+            game_display = game_display[game_display['Date'].isin(date_var1)]
+            game_display = game_display.set_index('Team Date')
+            st.dataframe(game_display.style.background_gradient(axis=0).background_gradient(cmap='RdYlGn').format(precision=2), use_container_width = True)
+            st.download_button(
+                label="Export Game Model",
+                data=convert_df_to_csv(game_model),
+                file_name='AmericanNumbers_Game_Model_export.csv',
+                mime='text/csv',
+            )
+        elif view_var1 == 'Gimme details':
+            game_display = game_model[game_model['Acro'].isin(team_var1)]
+            game_display = game_display[game_display['Date'].isin(date_var1)]
+            game_display = game_display.set_index('Team Date')
+            st.dataframe(game_display.style.background_gradient(axis=0).background_gradient(cmap='RdYlGn').format(precision=2), use_container_width = True)
+            st.download_button(
+                label="Export Game Model",
+                data=convert_df_to_csv(game_model),
+                file_name='AmericanNumbers_Game_Model_export.csv',
+                mime='text/csv',
+            )
 
 with tab2:
     col1, col2 = st.columns([1, 9])
     with col1:
         if st.button("Reset Data", key='reset2'):
                   st.cache_data.clear()
-                  game_model, season_model, seed_probs, title_sims = init_baselines()
+                  game_model, season_model, seed_probs, title_sims, just_win_probs = init_baselines()
         view_var2 = st.radio("Would you like to view title odds and win projections or seeding probabilities?", ('Win Odds', 'Seed Probabilities'), key='view_var2')
         split_var2 = st.radio("Would you like to view all teams or specific ones?", ('All', 'Specific Teams'), key='split_var2')
         if split_var2 == 'Specific Teams':