from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
import datetime
import requests
import pytz
import yaml
from tools.final_answer import FinalAnswerTool

from Gradio_UI import GradioUI

# Below is an example of a tool that does nothing. Amaze us with your creativity !
@tool
def blackjack_strategy(player_cards: list[str], dealer_card:str)-> str:
    """A tool that determines optimal Blackjack strategy based on player's cards and dealer's face-up card. 
    Args:
        player_cards: A list of strings representing the player's cards.
                      Each card is represented as two character string (ex: 'AH' for Ace of hearts, '10S' for 10 of Spades).     
        dealer_card: A string representing delaler's face-up card

    Returns: Recommended action ("Hit", "Stand", or "Double Down"). 
    """
    def card_value(card: str) -> int:
        """Converts a card string into its numeric value for Blackjack.
        Args:
            card: A string representing a playing card, where the first part is the rank 
                  (2-10, J, Q, K, A) and the last character is the suit (H, D, C, S).

        Returns:
            int: The numeric value of the card:
                 - A (Ace): 11
                 - J, Q, K (Face cards): 10
                 - Numeric cards: Their respective integer value
        """
        if card[0] == 'A': return 11
        elif card[0] in 'KQJ': return 10
        else: return int(card[:-1])

    player_values = [card_value(card) for card in player_cards]
    player_total = sum(player_values)
    dealer_total = card_value(dealer_card)

    if 11 in player_values and player_total > 21:
        player_total -= 10
    
    if player_total >= 17:
        return 'Stand'
    elif player_total >= 13 and player_total <= 16:
        if dealer_total >= 7:
            return 'Hit'
        else:
            return 'Stand'
    elif player_total == 12:
        if dealer_total >= 4 and dealer_total <= 6:
            return 'Stand'
        else:
            return 'Hit'
    elif player_total <= 11:
        # Double down condition (9, 10, or 11) - Only allowed if it's the first two cards
        if len(player_cards) == 2 and (player_total in [9,10,11]):
            if dealer_total >= 2 and dealer_total <= 9:  # Typical condition for doubling down
                return 'Double Down'
            else:
                return 'Hit'
        else:
            return 'Hit'

@tool
def get_current_time_in_timezone(timezone: str) -> str:
    """A tool that fetches the current local time in a specified timezone.
    Args:
        timezone: A string representing a valid timezone (e.g., 'America/New_York').
    """
    try:
        # Create timezone object
        tz = pytz.timezone(timezone)
        # Get current time in that timezone
        local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
        return f"The current local time in {timezone} is: {local_time}"
    except Exception as e:
        return f"Error fetching time for timezone '{timezone}': {str(e)}"


final_answer = FinalAnswerTool()

# If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
# model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud' 

model = HfApiModel(
max_tokens=2096,
temperature=0.5,
model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
custom_role_conversions=None,
)


# Import tool from Hub
image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)

with open("prompts.yaml", 'r') as stream:
    prompt_templates = yaml.safe_load(stream)
    
agent = CodeAgent(
    model=model,
    tools=[final_answer, blackjack_strategy], ## add your tools here (don't remove final answer)
    max_steps=6,
    verbosity_level=1,
    grammar=None,
    planning_interval=None,
    name=None,
    description=None,
    prompt_templates=prompt_templates
)


GradioUI(agent).launch()