Update app.py

app.py

@@ -4,15 +4,13 @@ import random
 app = Flask(__name__)
 
 # ---------------- Configuration ----------------
-# Adjust these values to tune game difficulty:
-AIR_STRIKE_CHANCE = 0.15        # Chance (15%) to call in an Air Strike if available
-SELF_REVIVE_CHANCE = 0.10       # Chance (10%) to self-revive when downed
-EXTRA_LOST_PLATE_CHANCE = 0.30  # Chance (30%) to lose an extra plate from enemy fire
-GULAG_WIN_THRESHOLD = 0.35      # In Gulag: outcome <35% = win (return to battle)
-GULAG_ELIMINATED_THRESHOLD = 0.65  # In Gulag: outcome >=35% and <65% = eliminated; >=65% = stalemate (lose a plate and try again)
+AIR_STRIKE_CHANCE = 0.15        
+SELF_REVIVE_CHANCE = 0.10       
+EXTRA_LOST_PLATE_CHANCE = 0.30  
+GULAG_WIN_THRESHOLD = 0.35      
+GULAG_ELIMINATED_THRESHOLD = 0.65  
 # ------------------------------------------------
 
-# In-memory storage for each player's game state.
 game_states = {}
 
 def limit_response(text, limit=400):
@@ -34,12 +32,12 @@ def init_game(username, loadout):
         "ammo": 100,
         "armor": 100,
         "inventory": [],
-        "equipment": [],   # e.g. "UAV", "Air Strike"
+        "equipment": [],
         "plates": 3,
         "max_plates": 12,
         "in_gulag": False,
-        "gulag_count": 0,  # number of times downed
-        "gulag_tokens": 0  # only obtainable AFTER first gulag
+        "gulag_count": 0,  
+        "gulag_tokens": 0  
     }
 
 @app.route("/")
@@ -63,8 +61,6 @@ def fight():
     if not user or user not in game_states:
         return limit_response("Start game first with !start")
     state = game_states[user]
-    
-    # If game is over, display final result.
     if state["status"] != "active":
         msg = f"Game over. Final Kills: {state['kills']}. "
         if state["status"] == "won":
@@ -75,7 +71,6 @@ def fight():
             msg += "You're in Gulag! Use !gulag to fight your opponent."
         return limit_response(msg)
     
-    # Adjust effective bonus if no plates remain.
     effective_bonus = state["bonus"] if state["plates"] > 0 else state["bonus"] * 0.7
     state["round"] += 1
     r = state["round"]
@@ -84,7 +79,6 @@ def fight():
     resp = f"R{r}: "
     
     if event == "fight":
-        # Air Strike branch (if available)
         if "Air Strike" in state["equipment"] and random.random() < AIR_STRIKE_CHANCE:
             kills = random.randint(2, 3)
             state["kills"] += kills
@@ -93,7 +87,6 @@ def fight():
             resp += f"You called in an Air Strike! It obliterated {kills} enemies. "
             state["equipment"].remove("Air Strike")
         else:
-            # Select one of 15 detailed gunfight scenarios.
             scenario = random.randint(1,15)
             gun_used = random.choice(["M4", "AK-47", "Shotgun", "Sniper", "SMG"])
             downed = False
@@ -121,7 +114,6 @@ def fight():
                 downed = True
             elif scenario == 8:
                 resp += f"A stray round from your {gun_used} grazed an enemy, but nothing decisive happened. "
-                kills = 0
             elif scenario == 9:
                 kills = 1
                 resp += f"An aggressive assault with your {gun_used} got you {kills} kill. "
@@ -146,7 +138,6 @@ def fight():
                 resp += f"Your {gun_used} burst ricocheted, earning {kills} kill but costing you a plate. "
                 state["plates"] = max(state["plates"] - 1, 0)
             elif scenario == 15:
-                kills = 0
                 resp += f"In the chaos, you managed no kills and were left vulnerable."
                 if random.random() < 0.5:
                     downed = True
@@ -157,12 +148,10 @@ def fight():
             if kills > 0:
                 state["players"] = max(state["players"] - kills, 1)
                 state["ammo"] = max(state["ammo"] - random.randint(5,10), 0)
-            # Additional chance to lose a plate from enemy fire.
             if not downed and random.random() < EXTRA_LOST_PLATE_CHANCE:
                 lost = 1
                 state["plates"] = max(state["plates"] - lost, 0)
                 resp += f"Amid the firefight, you lost {lost} plate from enemy fire. "
-            # If downed, try self-revive; if unsuccessful, enter Gulag.
             if downed:
                 if random.random() < SELF_REVIVE_CHANCE:
                     state["revives"] += 1
@@ -290,6 +279,11 @@ def gulag():
     if not user or user not in game_states:
         return limit_response("Start game first with !start")
     state = game_states[user]
+    # If game is over (won or eliminated), just show final result.
+    if state["status"] in ["eliminated", "won"]:
+        msg = f"Game over. Final Kills: {state['kills']}. Use !start to play again."
+        return limit_response(msg)
+    # If not in gulag, inform the user.
     if not state.get("in_gulag"):
         return limit_response("You're not in Gulag. If downed, use !gulag to fight your opponent.")
     
@@ -302,6 +296,7 @@ def gulag():
         resp = waiting_msg + "After a fierce duel, you won the Gulag fight and return to battle. Use !fight to continue."
     elif outcome < GULAG_ELIMINATED_THRESHOLD:
         state["status"] = "eliminated"
+        state["in_gulag"] = False
         resp = waiting_msg + f"Your opponent proved too strong. You were eliminated in Gulag. Final Kills: {state['kills']}. Use !start to play again."
     else:
         if state["plates"] > 0: