Update app.py

app.py

@@ -26,6 +26,7 @@ def init_game(username, loadout):
         "ammo": 100,
         "armor": 100,
         "inventory": [],
+        "equipment": [],      # extra items like UAV, Air Strike
         "in_gulag": False
     }
 
@@ -33,7 +34,6 @@ def init_game(username, loadout):
 def home():
     return "Welcome to the BR game API! Use /start, /fight, or /gulag."
 
-
 @app.route("/start", methods=["GET"])
 def start():
     user = request.args.get("user")
@@ -51,22 +51,32 @@ def fight():
     if not user or user not in game_states:
         return limit_response("Start game first with !start"), 400
     state = game_states[user]
-    if state["status"] != "active":
+    if state["status"] not in ["active"]:
         return limit_response(f"Game over. Final Kills: {state['kills']}."), 400
     state["round"] += 1
     r = state["round"]
+    final_round = state["players"] <= 3  # Final round condition
     event = random.choice(["fight", "loot", "buy", "ambush"])
     resp = f"R{r}: "
+    
     if event == "fight":
         roll = random.random() * state["bonus"]
-        if roll > 0.7:
+        # In final round, force kill outcome (no enemy fleeing)
+        if final_round or roll > 0.7:
             kills = random.randint(1, 3)
             state["kills"] += kills
             state["players"] = max(state["players"] - kills, 1)
             state["ammo"] = max(state["ammo"] - random.randint(5,15), 0)
             resp += f"Fought & got {kills} kill(s)! "
         elif roll > 0.5:
-            resp += "Enemy fled! "
+            # In non-final rounds, enemy may flee—but warn if it's final round.
+            if final_round:
+                kills = random.randint(1, 3)
+                state["kills"] += kills
+                state["players"] = max(state["players"] - kills, 1)
+                resp += f"Fought & got {kills} kill(s)! "
+            else:
+                resp += "Enemy fled! (Avoid fleeing in final rounds!) "
         else:
             if random.random() < 0.3 and state["revives"] < 1:
                 state["revives"] += 1
@@ -74,7 +84,8 @@ def fight():
             else:
                 state["in_gulag"] = True
                 state["status"] = "gulag"
-                return limit_response(resp + "Downed! Now in Gulag. Use !gulag.")
+                return limit_response(resp + "Downed! Enter Gulag. Use !gulag to fight your opponent.")
+    
     elif event == "loot":
         loot = random.choice(["weapon", "ammo", "armor"])
         if loot == "weapon":
@@ -91,18 +102,32 @@ def fight():
             add_arm = random.randint(5,20)
             state["armor"] += add_arm
             resp += f"Found {add_arm} armor. "
+    
     elif event == "buy":
         item = random.choice(["UAV", "Air Strike", "Cluster Strike", "Armor Box"])
-        if item == "Armor Box":
+        if item in ["UAV", "Air Strike"]:
+            state["equipment"].append(item)
+            state["bonus"] += 0.3  # High-value equipment gives a bigger bonus
+            resp += f"Bought {item} (boosts kills)! "
+        elif item == "Armor Box":
             state["armor"] += 10
+            resp += "Bought Armor Box. "
         else:
             state["bonus"] += 0.15
-        resp += f"Bought {item}. "
+            resp += f"Bought {item}. "
+    
     elif event == "ambush":
         resp += "Ambushed! "
         amb = random.choice(["escaped", "lost ammo", "got a kill"])
         if amb == "escaped":
-            resp += "Escaped. "
+            # In final rounds, don't allow escaping
+            if final_round:
+                kills = random.randint(1,2)
+                state["kills"] += kills
+                state["players"] = max(state["players"] - kills, 1)
+                resp += f"Ambush turned to fight: got {kills} kill(s)! "
+            else:
+                resp += "Escaped an ambush. "
         elif amb == "lost ammo":
             lost = random.randint(10,20)
             state["ammo"] = max(state["ammo"] - lost, 0)
@@ -111,8 +136,9 @@ def fight():
             kills = random.randint(1,2)
             state["kills"] += kills
             state["players"] = max(state["players"] - kills, 1)
-            resp += f"Got {kills} kill(s)! "
-    # Natural eliminations
+            resp += f"Got {kills} kill(s) in ambush! "
+    
+    # Natural eliminations happen regardless of event
     nat = random.randint(5,15)
     state["players"] = max(state["players"] - nat, 1)
     resp += f"Nat:{nat}. "
@@ -130,15 +156,18 @@ def gulag():
         return limit_response("Start game first with !start"), 400
     state = game_states[user]
     if not state.get("in_gulag"):
-        return limit_response("Not in Gulag."), 400
+        return limit_response("You're not currently in Gulag. If you've been downed, use !gulag to fight your opponent."), 400
+    
+    # Simulate the Gulag fight:
+    waiting_message = "In Gulag: waiting for an opponent... "
     if random.random() < (0.6 + state["bonus"] * 0.05):
         state["status"] = "active"
         state["in_gulag"] = False
         state["players"] = max(state["players"] - 5, 1)
-        resp = "Won Gulag fight! Back in action. Use !fight next."
+        resp = waiting_message + "Opponent engaged! You won the Gulag fight. Back in action. Use !fight."
     else:
         state["status"] = "eliminated"
-        resp = f"Lost Gulag fight. Game over. Final Kills: {state['kills']}."
+        resp = waiting_message + f"Opponent overpowered you. Game over. Final Kills: {state['kills']}."
     return limit_response(resp)
 
 if __name__ == "__main__":