Update app.py

app.py

@@ -9,41 +9,47 @@ wildlife_tokens = ['🐻', '🦅', '🐟', '🦌', '🐿️']
 players = ['Player 1', 'Player 2']
 save_file = 'cascadia_game_state.csv'
 
-# Initialize or load game state
-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)
-    st.session_state['players'] = {player: {'habitat': [], 'wildlife': [], 'nature_tokens': 3} for player in players}
-
+# Function to load game state from CSV
 def load_game_state():
     if os.path.exists(save_file):
         df = pd.read_csv(save_file)
+        game_state = {
+            'habitat_stack': df['habitat_stack'].dropna().tolist(),
+            'wildlife_stack': df['wildlife_stack'].dropna().tolist(),
+            'players': {}
+        }
         for player in players:
-            st.session_state['players'][player] = {
+            game_state['players'][player] = {
                 'habitat': df[player + '_habitat'].dropna().tolist(),
                 'wildlife': df[player + '_wildlife'].dropna().tolist(),
-                'nature_tokens': df[player + '_nature_tokens'][0]
+                'nature_tokens': int(df[player + '_nature_tokens'][0])
             }
-        st.session_state['habitat_stack'] = df['habitat_stack'].dropna().tolist()
-        st.session_state['wildlife_stack'] = df['wildlife_stack'].dropna().tolist()
+        return game_state
+    else:
+        return None
 
-def save_game_state():
+# Function to save game state to CSV
+def save_game_state(game_state):
     data = {
-        'habitat_stack': pd.Series(st.session_state['habitat_stack']),
-        'wildlife_stack': pd.Series(st.session_state['wildlife_stack'])
+        'habitat_stack': pd.Series(game_state['habitat_stack']),
+        'wildlife_stack': pd.Series(game_state['wildlife_stack'])
     }
     for player in players:
-        player_state = st.session_state['players'][player]
+        player_state = game_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 'players' not in st.session_state:
-    load_game_state()
-    if 'players' not in st.session_state:
-        initialize_game()
+# Initialize or load game state
+game_state = load_game_state()
+if game_state is None:
+    game_state = {
+        'habitat_stack': random.sample(habitat_tiles * 10, 50),
+        'wildlife_stack': random.sample(wildlife_tokens * 10, 50),
+        'players': {player: {'habitat': [], 'wildlife': [], 'nature_tokens': 3} for player in players}
+    }
 
 # Streamlit Interface
 st.title("🌲 Cascadia Lite 🌲")