Update app.py

app.py

@@ -13,14 +13,13 @@ save_file = 'cascadia_game_state.csv'
 def initialize_game():
     st.session_state['habitat_stack'] = random.sample(habitat_tiles * 10, 50)
     st.session_state['wildlife_stack'] = random.sample(wildlife_tokens * 10, 50)
-    for player in players:
-        st.session_state[player] = {'habitat': [], 'wildlife': [], 'nature_tokens': 3}
+    st.session_state['players'] = {player: {'habitat': [], 'wildlife': [], 'nature_tokens': 3} for player in players}
 
 def load_game_state():
     if os.path.exists(save_file):
         df = pd.read_csv(save_file)
         for player in players:
-            st.session_state[player] = {
+            st.session_state['players'][player] = {
                 'habitat': df[player + '_habitat'].dropna().tolist(),
                 'wildlife': df[player + '_wildlife'].dropna().tolist(),
                 'nature_tokens': df[player + '_nature_tokens'][0]
@@ -34,15 +33,16 @@ def save_game_state():
         'wildlife_stack': pd.Series(st.session_state['wildlife_stack'])
     }
     for player in players:
-        data[player + '_habitat'] = pd.Series(st.session_state[player]['habitat'])
-        data[player + '_wildlife'] = pd.Series(st.session_state[player]['wildlife'])
-        data[player + '_nature_tokens'] = pd.Series([st.session_state[player]['nature_tokens']])
+        player_state = st.session_state['players'][player]
+        data[player + '_habitat'] = pd.Series(player_state['habitat'])
+        data[player + '_wildlife'] = pd.Series(player_state['wildlife'])
+        data[player + '_nature_tokens'] = pd.Series([player_state['nature_tokens']])
     df = pd.DataFrame(data)
     df.to_csv(save_file, index=False)
 
-if 'habitat_stack' not in st.session_state:
+if 'players' not in st.session_state:
     load_game_state()
-    if 'habitat_stack' not in st.session_state:
+    if 'players' not in st.session_state:
         initialize_game()
 
 # Streamlit Interface