"""
Web agent for finding and extracting song lyrics from online sources.
"""

from loguru import logger
from smolagents import CodeAgent, VisitWebpageTool, FinalAnswerTool, DuckDuckGoSearchTool

from config import load_prompt_templates
from tools.analysis_tools import AnalyzeLyricsTool
from tools.formatting_tools import FormatAnalysisResultsTool
from tools.search_tools import ThrottledDuckDuckGoSearchTool


def create_single_agent(model):
    """ 
    Create an agent specialized in web browsing and lyrics extraction.
    
    Args:
        model: The LLM model to use with this agent
        
    Returns:
        A configured CodeAgent for web searches
    """
    prompt_templates = load_prompt_templates()

    # Define agent parameters directly in the code instead of using config
    # Example usage within the agent
    agent = CodeAgent(
        tools=[
            FinalAnswerTool(),
            DuckDuckGoSearchTool(),
            # ThrottledDuckDuckGoSearchTool(min_delay=7.0, max_delay=15.0),
            VisitWebpageTool(),
            AnalyzeLyricsTool(),
            FormatAnalysisResultsTool()
        ],
        model=model,
        additional_authorized_imports=['numpy', 'bs4', 'rich'],
        max_steps=25,
        verbosity_level=2,
        description="Specialized agent for finding and extracting song lyrics specified by the user. Performs lyrics search and after getting the lyrics full text from the web, it performs the analysis using given tools. Provides detailed commentary with beautiful and human-readable format using rich library formatting. Format analysis results in rich colorful output.",
        prompt_templates=prompt_templates
    )
    
    logger.info("Web agent (lyrics search) created successfully")
    return agent