Update app.py

app.py

@@ -50,43 +50,47 @@ if game_state is None:
         'wildlife_stack': random.sample(wildlife_tokens * 10, 50),
         'players': {player: {'habitat': [], 'wildlife': [], 'nature_tokens': 3} for player in players}
     }
+    save_game_state(game_state)
 
 # Streamlit Interface
 st.title("🌲 Cascadia Lite 🌲")
 
 # Function to draw habitat and wildlife
-def draw_habitat_and_wildlife(player):
-    habitat = st.session_state['habitat_stack'].pop()
-    wildlife = st.session_state['wildlife_stack'].pop()
-    st.session_state['players'][player]['habitat'].append(habitat)
-    st.session_state['players'][player]['wildlife'].append(wildlife)
+def draw_habitat_and_wildlife(player, game_state):
+    habitat = game_state['habitat_stack'].pop()
+    wildlife = game_state['wildlife_stack'].pop()
+    game_state['players'][player]['habitat'].append(habitat)
+    game_state['players'][player]['wildlife'].append(wildlife)
+    save_game_state(game_state)
 
-# Display players' areas
+# Display players' areas and handle actions
 for player in players:
     st.write(f"## {player}'s Play Area")
-    player_data = pd.DataFrame({'Habitat Tiles': st.session_state['players'][player]['habitat'], 
-                                'Wildlife Tokens': st.session_state['players'][player]['wildlife']})
+    player_data = pd.DataFrame({'Habitat Tiles': game_state['players'][player]['habitat'], 
+                                'Wildlife Tokens': game_state['players'][player]['wildlife']})
     st.dataframe(player_data)
 
     # Drafting Phase
     if st.button(f"{player}: Draw Habitat and Wildlife"):
-        draw_habitat_and_wildlife(player)
+        draw_habitat_and_wildlife(player, game_state)
+        game_state = load_game_state()  # Reload game state after update
 
-    # Tile Placement
-    tile_placement = st.selectbox(f"{player}: Place Habitat Tile", options=['Select Position'] + list(range(1, len(st.session_state['players'][player]['habitat']) + 1)))
-    
-    # Wildlife Placement
-    wildlife_placement = st.selectbox(f"{player}: Place Wildlife Token", options=['Select Habitat'] + list(range(1, len(st.session_state['players'][player]['wildlife']) + 1)))
+    # Tile and Wildlife Placement (Placeholders for actual game logic)
+    # ... (Add selectbox and logic for tile and wildlife placement)
 
-    # Nature Tokens
-    if st.session_state['players'][player]['nature_tokens'] > 0:
-        if st.button(f"{player}: Use a Nature Token"):
-            st.session_state['players'][player]['nature_tokens'] -= 1
-            # Logic for swapping wildlife tokens or other actions
-            st.write(f"{player} used a nature token. Remaining: {st.session_state['players'][player]['nature_tokens']}")
+    # Nature Tokens (Placeholder for actual game logic)
+    # ... (Add button and logic for using nature tokens)
 
-# Save game state
-save_game_state()
+# Reset Button
+if st.button("Reset Game"):
+    os.remove(save_file)  # Delete the save file
+    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}
+    }
+    save_game_state(game_state)
+    st.experimental_rerun()
 
 # Game Controls and Instructions
 st.write("## Game Controls")