Update app.py

app.py

@@ -148,7 +148,7 @@ with tab1:
 with tab2:
     with st.container():
         hold_container = st.empty()
-        col1, col2, col3 = st.columns([3, 3, 3])
+        col1, col2, col3, col4, col5 = st.columns([2, 2, 2, 2, 2])
         with col1:
             if st.button("Load/Reset Data", key='reset1'):
                 for key in st.session_state.keys():
@@ -177,10 +177,15 @@ with tab2:
         with col3:
             if proj_file is not None:
                 player_check = st.selectbox('Select player to create comps', options = proj_dataframe['Player'].unique(), key='dk_player')
+        with col4:
+            if proj_file is not None:
+                pos_var_list = st.multiselect('Which positions would you like to include?', options = proj_dataframe['Position'].unique(), key='pos_var_list')
+        with col5:
             if st.button('Simulate appropriate pivots'):
                 with hold_container:
                     
                     working_roo = proj_dataframe
+                    working_roo = working_roo[working_roo['Position'].isin(pos_var_list)]
                     working_roo.rename(columns={"Minutes Proj": "Minutes_Proj"}, inplace = True)
                     own_dict = dict(zip(working_roo.Player, working_roo.Own))
                     min_dict = dict(zip(working_roo.Player, working_roo.Minutes_Proj))
@@ -268,7 +273,11 @@ with tab2:
                     st.session_state.final_Proj = final_Proj
                     
                     hold_container = st.empty()
+    
     with st.container():
+        if 'final_Proj' in st.session_state:
+            st.dataframe(st.session_state.final_Proj.style.background_gradient(axis=0).background_gradient(cmap='RdYlGn').format(precision=2), use_container_width = True)
+        
         col1, col2 = st.columns([7, 2])         
         with col1:
             if 'display_portfolio' in st.session_state:
@@ -276,7 +285,4 @@ with tab2:
 
         with col2:
             if 'player_freq' in st.session_state:
-                st.dataframe(st.session_state.player_freq.style.background_gradient(axis=0).background_gradient(cmap='RdYlGn').format(precision=2), use_container_width = True)
-        
-        if 'final_Proj' in st.session_state:
-            st.dataframe(st.session_state.final_Proj.style.background_gradient(axis=0).background_gradient(cmap='RdYlGn').format(precision=2), use_container_width = True)
+                st.dataframe(st.session_state.player_freq.style.background_gradient(axis=0).background_gradient(cmap='RdYlGn').format(precision=2), use_container_width = True)