Update app.py

app.py

@@ -63,6 +63,14 @@ def draw_habitat_and_wildlife(player, game_state):
     game_state['players'][player]['wildlife'].append(wildlife)
     save_game_state(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}
+        }
+
 
 # Display players' areas and handle actions
 col1, col2 = st.columns(2)