Update app.py

app.py

@@ -213,7 +213,7 @@ async def handle_action(request: Request):
                 total_opponents = len(game["players"]) - 1
                 if len(game["challenge"]["responses"]) == total_opponents:
                     ambassador_player = next(p for p in game["players"] if p["name"] == game["challenge"]["challenger"])
-                    ambassador_player["cards"] = generate_cards()
+                    ambassador_player["cards"] = generate_cards(len(ambassador_player["cards"]))
                     game["challenge"] = None
                     return {"success": True, "message": f"Ambassador action accepted. {ambassador_player['name']} swaps cards with the deck."}
                 else:
@@ -224,7 +224,7 @@ async def handle_action(request: Request):
                     game["challenge"]["status"] = "choose"
                     game["challenge"]["challenger"] = player
                     game["challenge"]["target"] = ambassador_player["name"]
-                    ambassador_player["cards"] = generate_cards()
+                    ambassador_player["cards"] = generate_cards(len(ambassador_player["cards"]))
                     game["challenge"] = None
                     return {"success": True, "message": f"Challenge failed. {player} must lose a card, while {ambassador_player['name']} swaps cards with the deck."}
                 else:
@@ -428,7 +428,7 @@ async def start_game(request: Request):
         raise HTTPException(status_code=400, detail="No players in the game.")
 
     # Initialize players with 2 coins and cards.
-    game["players"] = [{"name": name, "coins": 2, "cards": generate_cards()} for name in game["players"]]
+    game["players"] = [{"name": name, "coins": 2, "cards": generate_cards(2)} for name in game["players"]]
     # Reset permissions to a dictionary with proper keys for each player.
     game["permissions"] = {player["name"]: {"steal": True, "gain": True} for player in game["players"]}
     game["gameStarted"] = True
@@ -436,10 +436,13 @@ async def start_game(request: Request):
 
     return {"success": True, "message": "Game started successfully!"}
 
-def generate_cards():
+def generate_cards(numofcards):
     cards = ['Duke', 'Assassin', 'Captain', 'Ambassador', 'Contessa']
     shuffled = sorted(cards, key=lambda x: 0.5 - random.random())
-    return [shuffled[0], shuffled[1]]
+    if numofcards == 1:
+        return [shuffled[0]]
+    else:
+        return [shuffled[0], shuffled[1]]
 
 if __name__ == "__main__":
     uvicorn.run(app, host="0.0.0.0", port=8000)