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

# 两个适合练习 Agent Tool 机制的简单工具

from smolagents import tool
import random


@tool
def add_numbers(a: int, b: int) -> int:
    """
    Add two integers together.
    Args:
        a: first integer
        b: second integer
    """
    return a + b


@tool
def random_motivation(name: str) -> str:
    """
    Return a random motivational message for the user.
    Args:
        name: user's name
    """
    messages = [
        f"{name}, keep going — progress compounds.",
        f"{name}, consistency beats intensity.",
        f"{name}, today is a good day to build something.",
        f"{name}, small steps still move you forward.",
    ]
    return random.choice(messages)


final_answer = FinalAnswerTool()
model = HfApiModel(
    max_tokens=2096,
    temperature=0.5,
    model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
    custom_role_conversions=None,
)


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

# Load system prompt from prompt.yaml file
with open("prompts.yaml", 'r') as stream:
    prompt_templates = yaml.safe_load(stream)
    
agent = CodeAgent(
    model=model,
    tools=[final_answer,add_numbers,random_motivation], # 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 # Pass system prompt to CodeAgent
)


GradioUI(agent).launch()