Update app.py

app.py

@@ -90,9 +90,6 @@ async def handle_action(request: Request):
 
     game = games[pin]
 
-    if action == 'getStatus':
-        return {"success": True, "challenge": game.get("challenge", None)}
-
     if action == 'steal':
         game["challenge"] = {
             "action": 'steal',
@@ -116,19 +113,24 @@ async def handle_action(request: Request):
             challenger["coins"] += coins_to_steal
             game["challenge"] = None
             return {"success": True, "message": f"Steal accepted. {coins_to_steal} coins transferred from {target_player['name']} to {challenger['name']}."}
+        
         elif response == 'challenge':
             if "Captain" in challenger["cards"]:
                 coins_to_steal = min(target_player["coins"], 2)
                 target_player["coins"] -= coins_to_steal
                 challenger["coins"] += coins_to_steal
-                game["challenge"] = None
-                return {"success": True, "message": f"Challenge failed. {target_player['name']} loses {coins_to_steal} coins to {challenger['name']}."}
+                game["challenge"]["status"] = 'chooseChallenger'
+                return {"success": True, "message": f"Challenge failed. {target_player['name']} loses {coins_to_steal} coins to {challenger['name']}. The challenger must now choose a card to lose.", "challenge": game["challenge"]}
             else:
                 game["challenge"]["status"] = 'choose'
                 return {"success": True, "message": f"Challenge successful. {challenger['name']} must choose a card to lose.", "challenge": game["challenge"]}
 
     if action == 'choose':
-        challenger = next(p for p in game["players"] if p["name"] == game["challenge"]["challenger"])
+        if game["challenge"]["status"] == 'choose':
+            challenger = next(p for p in game["players"] if p["name"] == game["challenge"]["challenger"])
+        elif game["challenge"]["status"] == 'chooseChallenger':
+            challenger = next(p for p in game["players"] if p["name"] == game["challenge"]["target"])
+        
         if target in challenger["cards"]:
             challenger["cards"].remove(target)
             game["challenge"] = None