Update app.py

app.py

@@ -124,11 +124,26 @@ async def handle_action(request: Request):
         }
         return {"success": True, "message": f"Coup initiated by {player} targeting {target}."}
 
+    if action == 'assassin':
+        # Assassin action: must be the player's turn.
+        if game["turn"] != player:
+            return {"success": False, "message": "Not your turn."}
+        # Create an assassin challenge waiting for the target's decision.
+        game["challenge"] = {
+            "action": "assassin",
+            "challenger": player,    # the assassin
+            "target": target,        # the chosen target
+            "challengeType": "assassin",
+            "status": "pending",
+            "phase": "target_decision"  # awaiting target's decision: allow/challenge/contessa
+        }
+        return {"success": True, "message": f"Assassin action initiated by {player} targeting {target}. Awaiting target's response."}
+
     if action == 'challengeResponse':
         if not game.get("challenge"):
             return {"success": False, "message": "No challenge pending."}
 
-        # Handle steal challengeResponse as before.
+        # --- Handle existing steal response ---
         if game["challenge"]["challengeType"] == "steal":
             challenger_player = next(p for p in game["players"] if p["name"] == game["challenge"]["challenger"])
             target_player = next(p for p in game["players"] if p["name"] == game["challenge"]["target"])
@@ -151,18 +166,15 @@ async def handle_action(request: Request):
                     game["challenge"]["status"] = 'choose'
                     return {"success": True, "message": f"Challenge successful. {challenger_player['name']} must choose a card to lose.", "challenge": game["challenge"]}
 
-        # New: Handle Ambassador challenge/allow responses.
+        # --- Handle Ambassador response (existing) ---
         elif game["challenge"]["challengeType"] == "ambassador":
-            # Opponents respond with allow or challenge.
             if response == 'allow':
                 if "responses" not in game["challenge"]:
                     game["challenge"]["responses"] = {}
                 game["challenge"]["responses"][player] = 'allow'
                 total_opponents = len(game["players"]) - 1
                 if len(game["challenge"]["responses"]) == total_opponents:
-                    # All opponents allowed; perform the swap.
                     ambassador_player = next(p for p in game["players"] if p["name"] == game["challenge"]["challenger"])
-                    # Swap cards with the deck:
                     ambassador_player["cards"] = generate_cards()
                     game["challenge"] = None
                     return {"success": True, "message": f"Ambassador action accepted. {ambassador_player['name']} swaps cards with the deck."}
@@ -170,22 +182,89 @@ async def handle_action(request: Request):
                     return {"success": True, "message": f"{player} allowed the Ambassador action. Awaiting other responses."}
             elif response == 'challenge':
                 ambassador_player = next(p for p in game["players"] if p["name"] == game["challenge"]["challenger"])
-                # If the acting player really had Ambassador, then the challenger loses a card.
                 if "Ambassador" in ambassador_player["cards"]:
                     game["challenge"]["status"] = "choose"
-                    game["challenge"]["challenger"] = player   # The challenging opponent now must choose a card to lose.
+                    game["challenge"]["challenger"] = player  # challenger loses card
                     game["challenge"]["target"] = ambassador_player["name"]
                     ambassador_player["cards"] = generate_cards()
                     game["challenge"] = None
-                    return {"success": True, "message": f"Challenge failed. Challenger must choose a card to lose. {ambassador_player['name']} swaps cards with the deck."}
+                    return {"success": True, "message": f"Challenge failed. {player} must lose a card, while {ambassador_player['name']} swaps cards with the deck."}
                 else:
                     game["challenge"]["status"] = "choose"
                     return {"success": True, "message": "Challenge successful. Ambassador user must choose a card to lose.", "challenge": game["challenge"]}
-    
+
+        # --- Handle Assassin/Contessa responses ---
+        elif game["challenge"]["challengeType"] == "assassin":
+            # Phase 1: target's decision on assassination
+            if game["challenge"].get("phase") == "target_decision":
+                if response == "allow":
+                    # Target accepts assassination: must lose one card.
+                    game["challenge"]["status"] = "choose"
+                    # Set the losing player to the target.
+                    game["challenge"]["challenger"] = game["challenge"]["target"]
+                    return {"success": True, "message": f"{game['challenge']['target']} must now choose a card to lose."}
+                elif response == "challenge":
+                    # Target challenges the assassin's claim.
+                    assassin_player = next(p for p in game["players"] if p["name"] == game["challenge"]["challenger"])
+                    if "Assassin" in assassin_player["cards"]:
+                        # Assassin is truthful: target loses both cards.
+                        target_player = next(p for p in game["players"] if p["name"] == game["challenge"]["target"])
+                        target_player["cards"] = []  # remove all cards (eliminated)
+                        game["challenge"] = None
+                        return {"success": True, "message": f"Challenge failed. {target_player['name']} loses both cards."}
+                    else:
+                        # Assassin is lying: assassin must lose one card.
+                        game["challenge"]["status"] = "choose"
+                        game["challenge"]["challenger"] = assassin_player["name"]
+                        return {"success": True, "message": f"Challenge successful. {assassin_player['name']} must choose a card to lose.", "challenge": game["challenge"]}
+                elif response == "contessa":
+                    # Target opts to block with Contessa. Shift phase.
+                    game["challenge"]["phase"] = "assassin_contessa"
+                    return {"success": True, "message": f"{game['challenge']['challenger']} must now choose to challenge or accept the Contessa."}
+            # Phase 2: assassin responding to contessa claim.
+            elif game["challenge"].get("phase") == "assassin_contessa":
+                if response == "accept":
+                    # Assassin accepts contessa block: target loses one card.
+                    game["challenge"]["status"] = "choose"
+                    game["challenge"]["challenger"] = game["challenge"]["target"]
+                    return {"success": True, "message": f"{game['challenge']['target']} must now choose a card to lose."}
+                elif response == "challenge":
+                    # Assassin challenges contessa.
+                    target_player = next(p for p in game["players"] if p["name"] == game["challenge"]["target"])
+                    if "Contessa" in target_player["cards"]:
+                        # Contessa is truthful: assassin loses one card.
+                        game["challenge"]["status"] = "choose"
+                        # Losing player is the assassin.
+                        game["challenge"]["challenger"] = game["challenge"]["challenger"]
+                        return {"success": True, "message": f"Contessa challenge successful. {game['challenge']['challenger']} must choose a card to lose.", "challenge": game["challenge"]}
+                    else:
+                        # Contessa is bluffing: target loses both cards.
+                        target_player["cards"] = []
+                        game["challenge"] = None
+                        return {"success": True, "message": f"Contessa challenge failed. {target_player['name']} loses both cards."}
+
+        # --- Handle Foreign Aid responses ---
+        elif game["challenge"]["challengeType"] == "foreignAid":
+            if player == game["challenge"]["challenger"]:
+                return {"success": False, "message": "Challenger cannot respond to their own action."}
+            game["challenge"]["responses"][player] = response
+            if response == "block":
+                game["challenge"]["status"] = "blocked"
+                game["challenge"] = None
+                return {"success": True, "message": f"Foreign Aid blocked by {player}."}
+            else:
+                total_opponents = len(game["players"]) - 1
+                if len(game["challenge"]["responses"]) == total_opponents:
+                    challenger_player = next(p for p in game["players"] if p["name"] == game["challenge"]["challenger"])
+                    challenger_player["coins"] += 2
+                    game["challenge"] = None
+                    return {"success": True, "message": f"Foreign Aid accepted. {challenger_player['name']} gains 2 coins."}
+                else:
+                    return {"success": True, "message": f"{player} allowed Foreign Aid. Awaiting other responses."}
+
     if action == 'ambassador':
         if game["turn"] != player:
             return {"success": False, "message": "Not your turn."}
-        # Initiate the ambassador action with a pending challenge.
         game["challenge"] = {
             "action": "ambassador",
             "challenger": player,
@@ -196,7 +275,6 @@ async def handle_action(request: Request):
         return {"success": True, "message": f"Ambassador action initiated by {player}. Waiting for opponents to respond."}
 
     if action == 'choose':
-        # Pre-existing choose logic to lose a card.
         acting_player = next((p for p in game["players"] if p["name"] == game["challenge"]["challenger"]), None)
         if acting_player and target in acting_player["cards"]:
             acting_player["cards"].remove(target)
@@ -218,14 +296,11 @@ async def handle_action(request: Request):
         if not player_found:
             raise HTTPException(status_code=404, detail="Player not found in the game.")
 
-    # --- NEW: Foreign Aid ---
     if action == 'foreignAid':
-        # Ensure it is the player's turn and they haven't already performed a gain action.
         if game["turn"] != player:
             return {"success": False, "message": "Not your turn."}
         if not game["permissions"].get(player, {}).get("gain", True):
             return {"success": False, "message": "You can only earn money once per turn."}
-        # Create a pending foreign aid challenge.
         game["challenge"] = {
             "action": "foreignAid",
             "challenger": player,
@@ -237,25 +312,18 @@ async def handle_action(request: Request):
         return {"success": True, "message": f"Foreign Aid initiated by {player}. Waiting for opponents to respond."}
 
     if action == 'foreignAidResponse':
-        # This branch handles responses from opponents for the foreign aid action.
         if not game.get("challenge") or game["challenge"].get("challengeType") != "foreignAid":
             return {"success": False, "message": "No foreign aid action pending."}
-        # Only opponents may respond.
         if player == game["challenge"]["challenger"]:
             return {"success": False, "message": "Challenger cannot respond to their own action."}
-        # Record the response.
         game["challenge"]["responses"][player] = response
-        # If any opponent chooses to block, immediately cancel the foreign aid.
         if response == "block":
             game["challenge"]["status"] = "blocked"
-            challenger = game["challenge"]["challenger"]
             game["challenge"] = None
             return {"success": True, "message": f"Foreign Aid blocked by {player}."}
         else:
-            # Check if all opponents have responded (all players except the challenger).
             total_opponents = len(game["players"]) - 1
             if len(game["challenge"]["responses"]) == total_opponents:
-                # No opponent blocked; grant the acting player +2 coins.
                 challenger_player = next(p for p in game["players"] if p["name"] == game["challenge"]["challenger"])
                 challenger_player["coins"] += 2
                 game["challenge"] = None
@@ -263,7 +331,6 @@ async def handle_action(request: Request):
             else:
                 return {"success": True, "message": f"{player} allowed Foreign Aid. Awaiting other responses."}
 
-    # Default response.
     return {"success": True, "message": f"Action '{action}' processed for player {player}."}
 
 @app.post("/api/joinGame")