import gradio as gr
from transformers import pipeline

# Load the text generation model
pipe = pipeline("text-generation", model="microsoft/Phi-3.5-mini-instruct", trust_remote_code=True)

# Initialize global variables for points and game status
points = 0
current_emoji = ""

# Function to reset the game and initialize introduction
def reset_game():
    global points, current_emoji
    points = 0
    # LLM introduces the game and asks the first emoji puzzle
    introduction = "Welcome to 'Guess the Word from Emojis'! 🎮 Here's how it works:\n\n- You'll get an emoji-based puzzle.\n- For each correct guess, you'll earn 1 point.\n- Wrong guesses will deduct 1 point.\n- Win by reaching 10 points, or lose if your score drops to -10 points.\n\nLet's begin!\n"
    
    # Ask LLM for the first emoji puzzle
    emoji_prompt = "Give me an emoji-based puzzle for a word or phrase guessing game."
    response = pipe([{"role": "user", "content": emoji_prompt}])
    current_emoji = response[0]['generated_text']
    
    return introduction + f"Here's your first puzzle: {current_emoji}"

# Function to handle the game logic and user interactions
def emoji_game(user_input=""):
    global points, current_emoji

    # Normalize the user input
    user_input = user_input.strip().lower()
    
    # Handle specific phrases from the user
    if user_input in ["hi", "hello"]:
        return "Hello! Ready to play 'Guess the Word from Emojis'? Type your guess or ask for a hint!"
    elif user_input in ["how does the game work?", "can you explain the rules?", "how do I play?"]:
        return "In this game, you'll guess the word or phrase based on emojis. For each correct guess, you earn 1 point. If you guess wrong, you lose 1 point. Reach 10 points to win or drop to -10 points to lose!"
    elif user_input in ["give me a hint", "i need a hint"]:
        # Provide a hint (a simple example could be the length of the answer)
        hint = f"The word or phrase has {len(current_emoji)} characters."
        return f"Hint: {hint}"

    # If no user input (game start), LLM asks the question first
    if user_input == "":  
        return reset_game()
    
    # Check if the user made a guess; LLM evaluates the guess
    guess_prompt = f"User guessed '{user_input}'. Was the guess correct for this emoji puzzle: {current_emoji}?"
    response = pipe([{"role": "user", "content": guess_prompt}])
    response_text = response[0]['generated_text'].lower()
    
    # Process response: if it's correct, increase points; otherwise, decrease points
    if "correct" in response_text:  # Correct guess
        points += 1
        if points >= 10:
            return f"🎉 Correct! You've reached 10 points! You win the game! 🏆"
        else:
            # Get next emoji puzzle for the user
            next_emoji_prompt = "Give me another emoji-based puzzle."
            response = pipe([{"role": "user", "content": next_emoji_prompt}])
            current_emoji = response[0]['generated_text']
            return f"✅ Correct! Your current points: {points}. Here's your next puzzle: {current_emoji}"
    else:  # Incorrect guess
        points -= 1
        if points <= -10:
            return f"❌ Incorrect! You've dropped to -10 points. Game over! 😢"
        else:
            return f"❌ Incorrect! Your current points: {points}. Try again: {current_emoji}"

# Create a Gradio interface for the game
interface = gr.Interface(
    fn=emoji_game,  # Game logic function
    inputs="text",  # Text input for user guess
    outputs="text",  # Text output for LLM responses
    title="Guess the Word from Emojis Game",
    description="Try to guess the word or phrase based on the emojis! 🎮 Reach 10 points to win or drop to -10 points to lose!"
)

# Launch the Gradio interface
interface.launch()