Update app.py

app.py

@@ -104,12 +104,12 @@ with col1:
     view_var2 = st.radio("Would you like to view all books or specific ones?", ('All Books', 'Specific Books'), key='view_var2')
     if view_var2 == 'All Books':
         site_view = game_model.booksid.unique()
-    elif view_var2 == 'All Books':
+    elif view_var2 == 'Specific Books':
         site_view = st.multiselect("What books would you like included?", options = game_model.booksid.unique(), key='site_view')
 
 with col2:
     if view_var1 == 'Just win probs':
-        game_display = just_win_probs
+        game_display = just_win_probs[just_win_probs['booksid'].isin(site_view)]
         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",
@@ -118,7 +118,7 @@ with col2:
             mime='text/csv',
         )
     elif view_var1 == 'Gimme details':
-        game_display = game_model
+        game_display = game_model[game_model['booksid'].isin(site_view)]
         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",