Update app.py

app.py

@@ -26,9 +26,21 @@ class Character:
 
     def attack_target(self, target):
         """Performs an attack on a target."""
-        # Add a little randomness to the attack damage
+        # Add randomness, critical hits, and misses!
+        miss_chance = 0.05  # 5% chance to miss
+        if random.random() < miss_chance:
+            return f"{self.name} swings and MISSES {target.name}!"
+
         damage = random.randint(self.attack - 2, self.attack + 2)
-        message = f"{self.name} attacks {target.name} for {damage} damage!"
+
+        critical_hit = random.random() < 0.1  # 10% chance for a crit
+        if critical_hit:
+            damage = int(damage * 1.5)
+            crit_text = " CRITICAL HIT!"
+        else:
+            crit_text = ""
+
+        message = f"{self.name} attacks {target.name} for {damage} damage!{crit_text}"
         message += "\n" + target.take_damage(damage)
         return message
 
@@ -39,7 +51,8 @@ class PlayerCharacter(Character):
         super().__init__(name, max_hp, attack, defense, character_type="player")
         self.xp = 0
         self.level = 1
-        self.money = 0 # For future features like buying items
+        # Start players with a little spending cash
+        self.money = 25
 
     def gain_xp(self, amount):
         """Awards experience points and checks for level up."""
@@ -109,6 +122,14 @@ class ScottPilgrimGame:
             ("The Subspace Highway", EvilEx("Young Neil", 160, 21, 8, 160, 75)) # Fictional final boss
         ]
 
+        # --- In-game shop items ---
+        self.shop_items = {
+            "Energy Drink":  {"cost": 10, "type": "heal", "amount": 20,  "desc": "Restore 20 HP"},
+            "Protein Bar":   {"cost": 15, "type": "attack", "amount": 2,   "desc": "+2 Attack"},
+            "Armor Patch":   {"cost": 15, "type": "defense", "amount": 1,  "desc": "+1 Defense"},
+            "Max Potion":    {"cost": 20, "type": "heal_full", "amount": 0, "desc": "Fully heal"},
+        }
+
     def reset_game(self):
         """Resets the game state for a new playthrough."""
         self.__init__() # Re-initialize the game object to its default state
@@ -155,6 +176,7 @@ class ScottPilgrimGame:
         if not self.player:
             return ""
         status = f"{self.player.name} (Lvl {self.player.level}) | HP: {self.player.current_hp}/{self.player.max_hp} | XP: {self.player.xp}/{self.player.level * 50}"
+        status += f" | $: {self.player.money}"
         if self.current_enemy:
             status += f"\n{self.current_enemy.name} | HP: {self.current_enemy.current_hp}/{self.current_enemy.max_hp}"
         return status
@@ -231,6 +253,50 @@ class ScottPilgrimGame:
             # Enemy gets a free hit as a penalty for a failed escape
             self.enemy_attack_turn()
 
+    # --------- Shop Mechanics ---------
+    def get_shop_choices(self):
+        """Return list of item names with price for UI dropdown."""
+        return [f"{name} - ${data['cost']}" for name, data in self.shop_items.items()]
+
+    def buy_item(self, item_display_name):
+        """Process purchasing an item given the dropdown display string."""
+        if not self.player:
+            return
+
+        # Extract item key (before ' - $')
+        item_name = item_display_name.split(" - $")[0]
+        if item_name not in self.shop_items:
+            self.add_message("That item doesn't exist!")
+            return
+
+        item = self.shop_items[item_name]
+        cost = item["cost"]
+
+        if self.player.money < cost:
+            self.add_message("Not enough money!")
+            return
+
+        # Deduct money
+        self.player.money -= cost
+
+        # Apply item effect
+        if item["type"] == "heal":
+            self.player.current_hp = min(self.player.max_hp, self.player.current_hp + item["amount"])
+            self.add_message(f"You used {item_name} and healed {item['amount']} HP!")
+        elif item["type"] == "heal_full":
+            healed = self.player.max_hp - self.player.current_hp
+            self.player.current_hp = self.player.max_hp
+            self.add_message(f"{item_name} fully restored your HP (+{healed})!")
+        elif item["type"] == "attack":
+            self.player.attack += item["amount"]
+            self.add_message(f"{item_name} consumed! Attack increased by {item['amount']}.")
+        elif item["type"] == "defense":
+            self.player.defense += item["amount"]
+            self.add_message(f"{item_name} equipped! Defense increased by {item['amount']}.")
+
+        # Confirm purchase
+        self.add_message(f"You bought {item_name} for ${cost}. Remaining money: ${self.player.money}.")
+
 # --- Gradio Interface Logic ---
 
 # Create a single game instance to be managed by gr.State
@@ -271,7 +337,8 @@ def _format_status_html(game_state):
     status_html = [
         f"<h4 style='margin:4px 0'>{player.name}  (Lvl {player.level})</h4>",
         f"HP: <progress value='{player.current_hp}' max='{player.max_hp}' style='width:160px;height:14px;'></progress>  {player.current_hp}/{player.max_hp}<br>",
-        f"XP: <progress value='{player.xp}' max='{player.level * 50}' style='width:160px;height:14px;'></progress>  {player.xp}/{player.level * 50}<br><br>"
+        f"XP: <progress value='{player.xp}' max='{player.level * 50}' style='width:160px;height:14px;'></progress>  {player.xp}/{player.level * 50}<br>",
+        f"Money: ${player.money}<br><br>"
     ]
 
     if game_state.current_enemy:
@@ -309,6 +376,7 @@ def update_ui(game_state):
     char_buttons_visible = (phase == "start")
     action_buttons_visible = (phase == "combat")
     play_again_visible = (phase in ["game_over", "win"])
+    shop_visible = (phase == "combat")
 
     return (
         log_html,
@@ -316,6 +384,7 @@ def update_ui(game_state):
         gr.update(visible=char_buttons_visible),
         gr.update(visible=action_buttons_visible),
         gr.update(visible=play_again_visible),
+        gr.update(visible=shop_visible),
         game_state
     )
 
@@ -339,6 +408,15 @@ def play_again(game_state):
     game_state.reset_game()
     return update_ui(game_state)
 
+# --- Shop callback ---
+def buy_item_action(item_choice, game_state):
+    """Callback for Buy button in the shop."""
+    if item_choice:
+        game_state.buy_item(item_choice)
+    else:
+        game_state.add_message("Select an item first!")
+    return update_ui(game_state)
+
 # --- Gradio UI Layout ---
 
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
@@ -362,9 +440,14 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
 
     play_again_btn = gr.Button("Play Again?", visible=False)
 
+    # --- Shop UI Components ---
+    with gr.Row(visible=False) as shop_row:
+        item_dropdown = gr.Dropdown(choices=game_instance.get_shop_choices(), label="Shop Items")
+        buy_btn = gr.Button("Buy")
+
     # --- Event Handlers ---
     # Define a list of all UI components that need to be updated, including the state
-    outputs = [game_output, status_output, character_selection_buttons, game_action_buttons, play_again_btn, game_state]
+    outputs = [game_output, status_output, character_selection_buttons, game_action_buttons, play_again_btn, shop_row, game_state]
 
     # Connect buttons to their callback functions
     scott_btn.click(choose_character, inputs=[scott_btn, game_state], outputs=outputs)
@@ -372,6 +455,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     attack_btn.click(player_attack, inputs=[game_state], outputs=outputs)
     run_btn.click(player_run, inputs=[game_state], outputs=outputs)
     play_again_btn.click(play_again, inputs=[game_state], outputs=outputs)
+    buy_btn.click(buy_item_action, inputs=[item_dropdown, game_state], outputs=outputs)
 
     # Set the initial UI state when the app loads, ensuring the outputs match the function's return values
     demo.load(update_ui, inputs=[game_state], outputs=outputs)