Upload app.py

app.py

@@ -90,6 +90,9 @@ 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',
@@ -113,24 +116,19 @@ 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"]["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"]}
+                game["challenge"] = None
+                return {"success": True, "message": f"Challenge failed. {target_player['name']} loses {coins_to_steal} coins to {challenger['name']}."}
             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':
-        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"])
-        
+        challenger = next(p for p in game["players"] if p["name"] == game["challenge"]["challenger"])
         if target in challenger["cards"]:
             challenger["cards"].remove(target)
             game["challenge"] = None