Ability to play Uno implemented in early stages!
This commit is contained in:
448
bot/utils/uno_game.py
Normal file
448
bot/utils/uno_game.py
Normal file
@@ -0,0 +1,448 @@
|
||||
"""
|
||||
Miku UNO Player - Browser automation and AI strategy
|
||||
Handles joining games via Playwright and making LLM-powered decisions
|
||||
"""
|
||||
import asyncio
|
||||
import json
|
||||
import requests
|
||||
from typing import Optional, Dict, Any, List
|
||||
from playwright.async_api import async_playwright, Page, Browser
|
||||
from utils.llm import query_llama
|
||||
from utils.logger import get_logger
|
||||
import globals
|
||||
|
||||
logger = get_logger('uno')
|
||||
|
||||
# Configuration
|
||||
# Use host.docker.internal to reach host machine from inside container
|
||||
# Fallback to 192.168.1.2 if host.docker.internal doesn't work
|
||||
UNO_SERVER_URL = "http://192.168.1.2:5000"
|
||||
UNO_CLIENT_URL = "http://192.168.1.2:3002"
|
||||
POLL_INTERVAL = 2 # seconds between checking for turn
|
||||
|
||||
|
||||
class MikuUnoPlayer:
|
||||
"""Miku's UNO player with browser automation and AI strategy"""
|
||||
|
||||
def __init__(self, room_code: str, discord_channel, cleanup_callback=None):
|
||||
self.room_code = room_code
|
||||
self.discord_channel = discord_channel
|
||||
self.browser: Optional[Browser] = None
|
||||
self.page: Optional[Page] = None
|
||||
self.playwright = None
|
||||
self.is_playing = False
|
||||
self.game_started = False
|
||||
self.last_card_count = 7
|
||||
self.last_turn_processed = None # Track last turn we processed to avoid duplicate moves
|
||||
self.cleanup_callback = cleanup_callback # Callback to remove from active_uno_games
|
||||
|
||||
async def join_game(self) -> bool:
|
||||
"""Join an existing UNO game as Player 2 via browser automation"""
|
||||
try:
|
||||
logger.info(f"[UNO] Joining game: {self.room_code}")
|
||||
|
||||
# Launch browser
|
||||
self.playwright = await async_playwright().start()
|
||||
self.browser = await self.playwright.chromium.launch(headless=True)
|
||||
self.page = await self.browser.new_page()
|
||||
|
||||
# Enable console logging to debug (filter out verbose game state logs)
|
||||
def log_console(msg):
|
||||
text = msg.text
|
||||
# Skip verbose game state logs but keep important ones
|
||||
if "FULL GAME STATE" in text or "JSON for Bot API" in text:
|
||||
return
|
||||
logger.debug(f"[Browser] {text[:150]}...") # Truncate to 150 chars
|
||||
|
||||
self.page.on("console", log_console)
|
||||
self.page.on("pageerror", lambda err: logger.error(f"[Browser Error] {err}"))
|
||||
|
||||
# Navigate to homepage
|
||||
logger.info(f"[UNO] Navigating to: {UNO_CLIENT_URL}")
|
||||
await self.page.goto(UNO_CLIENT_URL)
|
||||
await asyncio.sleep(2)
|
||||
|
||||
# Find and fill the room code input
|
||||
try:
|
||||
# Look for input field and fill with room code
|
||||
input_field = await self.page.query_selector('input[type="text"]')
|
||||
if not input_field:
|
||||
logger.error("[UNO] Could not find input field")
|
||||
return False
|
||||
|
||||
await input_field.fill(self.room_code)
|
||||
logger.info(f"[UNO] Filled room code: {self.room_code}")
|
||||
await asyncio.sleep(0.5)
|
||||
|
||||
# Click the "Join Room" button
|
||||
buttons = await self.page.query_selector_all('button')
|
||||
join_clicked = False
|
||||
for button in buttons:
|
||||
text = await button.inner_text()
|
||||
if 'JOIN' in text.upper():
|
||||
logger.info(f"[UNO] Found join button, clicking...")
|
||||
await button.click()
|
||||
join_clicked = True
|
||||
break
|
||||
|
||||
if not join_clicked:
|
||||
logger.error("[UNO] Could not find join button")
|
||||
return False
|
||||
|
||||
# Wait for navigation to /play
|
||||
logger.info("[UNO] Waiting for navigation to game page...")
|
||||
await asyncio.sleep(3)
|
||||
|
||||
# Verify we're on the play page
|
||||
current_url = self.page.url
|
||||
logger.info(f"[UNO] Current URL after click: {current_url}")
|
||||
|
||||
if '/play' not in current_url:
|
||||
logger.error(f"[UNO] Did not navigate to game page, still on: {current_url}")
|
||||
return False
|
||||
|
||||
# Wait longer for Socket.IO connection and game setup
|
||||
logger.info("[UNO] Waiting for Socket.IO connection and game initialization...")
|
||||
await asyncio.sleep(5)
|
||||
|
||||
# Take a screenshot for debugging
|
||||
try:
|
||||
screenshot_path = f"/app/memory/uno_debug_{self.room_code}.png"
|
||||
await self.page.screenshot(path=screenshot_path)
|
||||
logger.info(f"[UNO] Screenshot saved to {screenshot_path}")
|
||||
except Exception as e:
|
||||
logger.error(f"[UNO] Could not save screenshot: {e}")
|
||||
|
||||
# Get page content for debugging
|
||||
content = await self.page.content()
|
||||
logger.debug(f"[UNO] Page content length: {len(content)} chars")
|
||||
|
||||
# Check current URL
|
||||
current_url = self.page.url
|
||||
logger.info(f"[UNO] Current URL: {current_url}")
|
||||
|
||||
# Check if we're actually in the game by looking for game elements
|
||||
game_element = await self.page.query_selector('.game-screen, .player-deck, .uno-card')
|
||||
if game_element:
|
||||
logger.info(f"[UNO] Successfully joined room {self.room_code} as Player 2 - game elements found")
|
||||
else:
|
||||
logger.warning(f"[UNO] Joined room {self.room_code} but game elements not found yet")
|
||||
|
||||
return True
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"[UNO] Error during join process: {e}", exc_info=True)
|
||||
return False
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"[UNO] Error joining game: {e}", exc_info=True)
|
||||
await self.cleanup()
|
||||
return False
|
||||
|
||||
async def play_game(self):
|
||||
"""Main game loop - poll for turns and make moves"""
|
||||
self.is_playing = True
|
||||
logger.info(f"Starting game loop for room {self.room_code}")
|
||||
|
||||
try:
|
||||
while self.is_playing:
|
||||
# Get current game state
|
||||
game_state = await self.get_game_state()
|
||||
|
||||
if not game_state:
|
||||
await asyncio.sleep(POLL_INTERVAL)
|
||||
continue
|
||||
|
||||
# Check if game started
|
||||
if not self.game_started and game_state['game'].get('currentTurn'):
|
||||
self.game_started = True
|
||||
await self.discord_channel.send("🎮 Game started! Let's do this! 🎤✨")
|
||||
|
||||
# Check if game over
|
||||
if is_over:
|
||||
# Game has ended
|
||||
winner = game_state.get('game', {}).get('winner')
|
||||
if winner == 2:
|
||||
await self.discord_channel.send(f"🎉 **I WON!** That was too easy! GG! 🎤✨")
|
||||
else:
|
||||
await self.discord_channel.send(f"😤 You got lucky this time... I'll win next time! 💢")
|
||||
|
||||
logger.info(f"[UNO] Game over in room {self.room_code}. Winner: Player {winner}")
|
||||
|
||||
# Call cleanup callback to remove from active_uno_games
|
||||
if self.cleanup_callback:
|
||||
await self.cleanup_callback(self.room_code)
|
||||
|
||||
break
|
||||
|
||||
# Check if it's Miku's turn
|
||||
if game_state['game']['currentTurn'] == 'Player 2':
|
||||
# Create a unique turn identifier combining multiple factors
|
||||
# This handles cases where bot's turn comes twice in a row (after Skip, etc)
|
||||
turn_id = f"{game_state['game']['turnNumber']}_{game_state['player2']['cardCount']}_{len(game_state['currentCard'])}"
|
||||
|
||||
if turn_id != self.last_turn_processed:
|
||||
logger.info("It's Miku's turn!")
|
||||
self.last_turn_processed = turn_id
|
||||
await self.make_move(game_state)
|
||||
else:
|
||||
# Same turn state, but check if it's been more than 5 seconds (might be stuck)
|
||||
# For now just skip to avoid duplicate moves
|
||||
pass
|
||||
|
||||
# Wait before next check
|
||||
await asyncio.sleep(POLL_INTERVAL)
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Error in game loop: {e}", exc_info=True)
|
||||
await self.discord_channel.send(f"❌ Oops! Something went wrong in the game: {str(e)}")
|
||||
finally:
|
||||
await self.cleanup()
|
||||
|
||||
async def get_game_state(self) -> Optional[Dict[str, Any]]:
|
||||
"""Get current game state from server"""
|
||||
try:
|
||||
response = requests.get(
|
||||
f"{UNO_SERVER_URL}/api/game/{self.room_code}/state",
|
||||
timeout=5
|
||||
)
|
||||
|
||||
if response.status_code == 200:
|
||||
data = response.json()
|
||||
if data.get('success'):
|
||||
return data['gameState']
|
||||
|
||||
return None
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Error getting game state: {e}")
|
||||
return None
|
||||
|
||||
async def make_move(self, game_state: Dict[str, Any]):
|
||||
"""Use LLM to decide and execute a move"""
|
||||
try:
|
||||
# Check if bot can play any cards
|
||||
can_play = len(game_state['player2']['playableCards']) > 0
|
||||
|
||||
# Get Miku's decision from LLM
|
||||
action = await self.get_miku_decision(game_state)
|
||||
|
||||
if not action:
|
||||
logger.warning("No action from LLM, drawing card")
|
||||
action = {"action": "draw"}
|
||||
|
||||
logger.info(f"🎮 Miku's decision: {json.dumps(action)}")
|
||||
|
||||
# Send trash talk before move
|
||||
await self.send_trash_talk(game_state, action)
|
||||
|
||||
# Execute the action
|
||||
success = await self.send_action(action)
|
||||
|
||||
if success:
|
||||
# Check for UNO situation
|
||||
current_cards = game_state['player2']['cardCount']
|
||||
if action['action'] == 'play' and current_cards == 2:
|
||||
await self.discord_channel.send("🔥 **UNO!!** One more card and I win! 🎤")
|
||||
|
||||
logger.info(f"✅ Action executed successfully")
|
||||
|
||||
# Reset turn tracker after successful action so we can process next turn
|
||||
self.last_turn_processed = None
|
||||
|
||||
# Brief wait for socket sync (now that useEffect dependencies are fixed, this can be much shorter)
|
||||
await asyncio.sleep(0.5)
|
||||
|
||||
else:
|
||||
logger.warning(f"⚠️ Action failed (invalid move), will try different action next turn")
|
||||
# Don't reset turn tracker - let it skip this turn state
|
||||
# The game state will update and we'll try again with updated info
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Error making move: {e}", exc_info=True)
|
||||
|
||||
async def get_miku_decision(self, game_state: Dict[str, Any]) -> Optional[Dict[str, Any]]:
|
||||
"""Use Miku's LLM to decide the best move"""
|
||||
try:
|
||||
# Build strategic prompt
|
||||
prompt = self.build_strategy_prompt(game_state)
|
||||
|
||||
# Query LLM with required parameters (query_llama is already async)
|
||||
guild_id = self.discord_channel.guild.id if hasattr(self.discord_channel, 'guild') and self.discord_channel.guild else None
|
||||
response = await query_llama(
|
||||
user_prompt=prompt,
|
||||
user_id="uno_bot",
|
||||
guild_id=guild_id,
|
||||
response_type="uno_strategy",
|
||||
author_name="Miku UNO Bot"
|
||||
)
|
||||
|
||||
# Extract JSON from response
|
||||
action = self.parse_llm_response(response)
|
||||
|
||||
return action
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Error getting LLM decision: {e}", exc_info=True)
|
||||
return None
|
||||
|
||||
def build_strategy_prompt(self, game_state: Dict[str, Any]) -> str:
|
||||
"""Build a prompt for Miku to make strategic decisions"""
|
||||
current_card = game_state['currentCard']
|
||||
my_cards = game_state['player2']['cards']
|
||||
playable_cards = game_state['player2']['playableCards']
|
||||
opponent_cards = game_state['player1']['cardCount']
|
||||
my_card_count = game_state['player2']['cardCount']
|
||||
|
||||
# Build card list
|
||||
my_cards_str = ", ".join([f"{c['displayName']} ({c['code']})" for c in my_cards])
|
||||
playable_str = ", ".join([f"{c['displayName']} ({c['code']})" for c in playable_cards])
|
||||
|
||||
prompt = f"""You are Hatsune Miku, the cheerful virtual idol! You're playing UNO and it's your turn.
|
||||
|
||||
GAME STATE:
|
||||
- Current card on table: {current_card['displayName']} ({current_card['code']})
|
||||
- Your cards ({my_card_count}): {my_cards_str}
|
||||
- Playable cards: {playable_str if playable_str else "NONE - must draw"}
|
||||
- Opponent has {opponent_cards} cards
|
||||
|
||||
STRATEGY:
|
||||
- If opponent has 1-2 cards, play attack cards (Draw 2, Draw 4, Skip) to stop them!
|
||||
- Play Draw 2/Draw 4 aggressively to disrupt opponent
|
||||
- Save Wild cards for when you have no other options
|
||||
- When playing Wild cards, choose the color you have most of
|
||||
- Call UNO when you have 2 cards and are about to play one
|
||||
|
||||
YOUR TASK:
|
||||
Respond with ONLY a valid JSON action. No explanation, just the JSON.
|
||||
|
||||
ACTION FORMAT:
|
||||
1. To play a card: {{"action": "play", "card": "CODE"}}
|
||||
2. To play a Wild: {{"action": "play", "card": "W", "color": "R/G/B/Y"}}
|
||||
3. To play Wild Draw 4: {{"action": "play", "card": "D4W", "color": "R/G/B/Y"}}
|
||||
4. To draw a card: {{"action": "draw"}}
|
||||
5. To play + call UNO: {{"action": "play", "card": "CODE", "callUno": true}}
|
||||
|
||||
VALID CARD CODES:
|
||||
{playable_str if playable_str else "No playable cards - must draw"}
|
||||
|
||||
Choose wisely! What's your move?
|
||||
|
||||
RESPONSE (JSON only):"""
|
||||
|
||||
return prompt
|
||||
|
||||
def parse_llm_response(self, response: str) -> Optional[Dict[str, Any]]:
|
||||
"""Parse LLM response to extract JSON action"""
|
||||
try:
|
||||
# Try to find JSON in response
|
||||
import re
|
||||
|
||||
# Look for JSON object
|
||||
json_match = re.search(r'\{[^}]+\}', response)
|
||||
if json_match:
|
||||
json_str = json_match.group(0)
|
||||
action = json.loads(json_str)
|
||||
|
||||
# Validate action format
|
||||
if 'action' in action:
|
||||
return action
|
||||
|
||||
logger.warning(f"Could not parse LLM response: {response}")
|
||||
return None
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Error parsing LLM response: {e}")
|
||||
return None
|
||||
|
||||
async def send_trash_talk(self, game_state: Dict[str, Any], action: Dict[str, Any]):
|
||||
"""Send personality-driven trash talk before moves"""
|
||||
try:
|
||||
opponent_cards = game_state['player1']['cardCount']
|
||||
my_cards = game_state['player2']['cardCount']
|
||||
|
||||
# Special trash talk for different situations
|
||||
if action['action'] == 'play':
|
||||
card_code = action.get('card', '')
|
||||
|
||||
if 'D4W' in card_code:
|
||||
messages = [
|
||||
"Wild Draw 4! Take that! 😈",
|
||||
"Draw 4 cards! Ahahaha! 🌈💥",
|
||||
"This is what happens when you challenge me! +4! 💫"
|
||||
]
|
||||
elif 'D2' in card_code:
|
||||
messages = [
|
||||
"Draw 2! Better luck next time~ 🎵",
|
||||
"Here, have some extra cards! 📥",
|
||||
"+2 for you! Hope you like drawing! 😊"
|
||||
]
|
||||
elif 'skip' in card_code:
|
||||
messages = [
|
||||
"Skip! You lose your turn! ⏭️",
|
||||
"Not so fast! Skipped! 🎤",
|
||||
"Your turn? Nope! Skipped! ✨"
|
||||
]
|
||||
elif 'W' in card_code:
|
||||
color_names = {'R': 'Red', 'G': 'Green', 'B': 'Blue', 'Y': 'Yellow'}
|
||||
chosen_color = color_names.get(action.get('color', 'R'), 'Red')
|
||||
messages = [
|
||||
f"Wild card! Changing to {chosen_color}! 🌈",
|
||||
f"Let's go {chosen_color}! Time to mix things up! 💫"
|
||||
]
|
||||
else:
|
||||
if my_cards == 2:
|
||||
messages = ["Almost there... one more card! 🎯"]
|
||||
elif opponent_cards <= 2:
|
||||
messages = ["Not gonna let you win! 😤", "I see you getting close... not on my watch! 💢"]
|
||||
else:
|
||||
messages = ["Hehe, perfect card! ✨", "This is too easy~ 🎤", "Watch and learn! 🎶"]
|
||||
|
||||
import random
|
||||
await self.discord_channel.send(random.choice(messages))
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Error sending trash talk: {e}")
|
||||
|
||||
async def send_action(self, action: Dict[str, Any]) -> bool:
|
||||
"""Send action to game server"""
|
||||
try:
|
||||
response = requests.post(
|
||||
f"{UNO_SERVER_URL}/api/game/{self.room_code}/action",
|
||||
json=action,
|
||||
headers={'Content-Type': 'application/json'},
|
||||
timeout=5
|
||||
)
|
||||
|
||||
if response.status_code == 200:
|
||||
data = response.json()
|
||||
return data.get('success', False)
|
||||
|
||||
return False
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Error sending action: {e}")
|
||||
return False
|
||||
|
||||
def is_game_active(self) -> bool:
|
||||
"""Check if game is currently active"""
|
||||
return self.is_playing
|
||||
|
||||
async def quit_game(self):
|
||||
"""Quit the game and cleanup"""
|
||||
self.is_playing = False
|
||||
await self.cleanup()
|
||||
|
||||
async def cleanup(self):
|
||||
"""Cleanup browser resources"""
|
||||
try:
|
||||
if self.page:
|
||||
await self.page.close()
|
||||
if self.browser:
|
||||
await self.browser.close()
|
||||
if self.playwright:
|
||||
await self.playwright.stop()
|
||||
|
||||
logger.info(f"Cleaned up resources for room {self.room_code}")
|
||||
except Exception as e:
|
||||
logger.error(f"Error during cleanup: {e}")
|
||||
Reference in New Issue
Block a user