Update app.py

app.py

@@ -199,7 +199,13 @@ with tab2:
     col1, col2 = st.columns([1, 7])
     with col1:
         if st.button("Load/Reset Data", key='reset1'):
+            for key in st.session_state.keys():
+                  del st.session_state[key]
             display_portfolio = hold_portfolio
+            st.session_state.display_portfolio = display_portfolio
+        proj_container = st.empty()
+        display_container = st.empty()
+        display_dl_container = st.empty()
         hold_container = st.empty()
         if st.button("Trim Lineups", key='trim1'):
             max_proj = 10000
@@ -207,10 +213,8 @@ with tab2:
             x = 0
             for index, row in display_portfolio.iterrows():
                 if row['Ownership'] > max_own:
-                    st.write(max_own)
                     display_portfolio.drop(index, inplace=True)
                 elif row['Ownership'] <= max_own:
-                    st.write(max_own)
                     max_own = row['Ownership']
         player_check = st.selectbox('Select player to create comps', options = proj_dataframe['Player'].unique(), key='dk_player')
         if st.button('Simulate appropriate pivots'):
@@ -301,13 +305,20 @@ with tab2:
                 final_Proj = final_Proj[['Player', 'Minutes Proj', 'Position', 'Team', 'Salary', 'Floor', 'Median', 'Ceiling', 'Top_finish', 'Top_5_finish', 'Top_10_finish', '20+%', '3x%', '4x%', '5x%', 'GPP%', 'Own', 'LevX', 'ValX']]
                 final_Proj = final_Proj.sort_values(by='Median', ascending=False)
                 final_Proj['Player_swap'] = player_check
+                st.session_state.final_Proj = final_Proj
                 
                 pivot_dict = dict(zip(final_Proj['Player_swap'], final_Proj['Player']))
                 
                 hold_container = st.empty()
                     
     with col2:    
-        with st.container():
-            st.dataframe(display_portfolio.style.background_gradient(axis=0).background_gradient(cmap='RdYlGn').format(precision=2), use_container_width = True)
-        with st.container():
-            st.dataframe(final_Proj.style.background_gradient(axis=0).background_gradient(cmap='RdYlGn').format(precision=2), use_container_width = True)
+        with proj_container:
+             proj_container = st.empty()
+             if 'display_portfolio' in st.session_state:
+                 st.dataframe(st.session_state.display_portfolio.style.background_gradient(axis=0).background_gradient(cmap='RdYlGn').format(precision=2), use_container_width = True)
+        
+        with display_container:
+             display_container = st.empty()
+             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)
+