Update app.py

app.py

@@ -105,7 +105,6 @@ async def handle_action(request: Request):
                 "challengeType": 'steal',
                 "status": 'pending'
             }
-            game["permissions"][player]["steal"] = False
             return {"success": True, "message": f"Steal initiated by {player} targeting {target}. Awaiting response from {target}."}
         else:
             return {"success": False, "message": "You can only steal once per turn."}
@@ -120,6 +119,7 @@ async def handle_action(request: Request):
             return {"success": False, "message": "Player not found in game."}
         next_index = (current_index + 1) % len(players_list)
         game["turn"] = players_list[next_index]["name"]
+        game["permissions"][player]["gain"] = True
         return {"success": True, "message": f"Turn ended. Next turn: {game['turn']}"}
 
     
@@ -161,8 +161,6 @@ async def handle_action(request: Request):
         player_data = next((p for p in game["players"] if p["name"] == player), None)
         if not game["permissions"].get(player, {}).get("gain", True):
             return {"success": False, "message": "You can only earn money once per turn."}
-        # Set gain to false immediately.
-        game["permissions"][player]["gain"] = False
         game["challenge"] = {
             "action": "duke",
             "challenger": player,
@@ -184,6 +182,7 @@ async def handle_action(request: Request):
                 coins_to_steal = min(target_player["coins"], 2)
                 target_player["coins"] -= coins_to_steal
                 challenger_player["coins"] += coins_to_steal
+                game["permissions"][player]["steal"] = False
                 game["challenge"] = None
                 return {"success": True, "message": f"Steal accepted. {coins_to_steal} coins transferred from {target_player['name']} to {challenger_player['name']}."}
             elif response == 'challenge':
@@ -191,6 +190,7 @@ async def handle_action(request: Request):
                     coins_to_steal = min(target_player["coins"], 2)
                     target_player["coins"] -= coins_to_steal
                     challenger_player["coins"] += coins_to_steal
+                    game["permissions"][player]["steal"] = False
                     game["challenge"]["status"] = 'choose'
                     game["challenge"]["challenger"] = target_player['name']
                     game["challenge"]["target"] = challenger_player['name']
@@ -310,6 +310,7 @@ async def handle_action(request: Request):
                 if len(game["challenge"]["responses"]) == total_opponents:
                     acting_player = next(p for p in game["players"] if p["name"] == game["challenge"]["challenger"])
                     acting_player["coins"] += 3
+                    game["permissions"][player]["gain"] = False
                     game["challenge"] = None
                     return {"success": True, "message": f"Duke action accepted. {acting_player['name']} gains 3 coins."}
                 else:
@@ -321,6 +322,7 @@ async def handle_action(request: Request):
                     game["challenge"]["challenger"] = player
                     game["challenge"]["target"] = acting_player["name"]
                     acting_player["coins"] += 3
+                    game["permissions"][player]["gain"] = False
                     return {"success": True, "message": f"Challenge failed. {player} must choose a card to lose.", "challenge": game["challenge"]}
                 else:
                     game["challenge"]["status"] = "choose"