Simplify all a bit

agents/analysis_agent.py

@@ -8,7 +8,7 @@ in a user-friendly way, showing lyrics with comments underneath them.
 from smolagents import CodeAgent, VisitWebpageTool
 from loguru import logger
 
-from config import AGENT_CONFIG, load_prompt_templates
+from config import load_prompt_templates
 from tools.search_tools import ThrottledDuckDuckGoSearchTool
 from tools.analysis_tools import analyze_lyrics_tool
 
@@ -23,11 +23,10 @@ def create_analysis_agent(model):
     Returns:
         A configured CodeAgent for lyrics analysis
     """
-    # Get configuration values
-    config = AGENT_CONFIG['analysis_agent']
+    # Load prompt templates
     prompt_templates = load_prompt_templates()
     
-    # Create the throttled search tool
+    # Create the throttled search tool with hardcoded values
     throttled_search_tool = ThrottledDuckDuckGoSearchTool(
         min_delay=3.0, 
         max_delay=7.0
@@ -47,8 +46,8 @@ def create_analysis_agent(model):
         name="lyrics_analysis_agent",
         description="Specialized agent for analyzing song lyrics and providing detailed commentary",
         additional_authorized_imports=["numpy", "bs4", "json", "re"],
-        max_steps=config['max_steps'],
-        verbosity_level=config['verbosity_level'],
+        max_steps=5,  # Hardcoded from config
+        verbosity_level=2,  # Hardcoded from config
         prompt_templates=prompt_templates
     )
     

agents/manager_agent.py

@@ -5,7 +5,7 @@ Manager agent for coordinating the lyrics search and analysis process.
 from smolagents import CodeAgent, FinalAnswerTool
 from loguru import logger
 
-from config import AGENT_CONFIG, load_prompt_templates
+from config import load_prompt_templates
 from agents.web_agent import create_web_agent
 from agents.analysis_agent import create_analysis_agent
 
@@ -20,8 +20,7 @@ def create_manager_agent(model):
     Returns:
         A configured CodeAgent manager for coordinating other agents
     """
-    # Get configuration values
-    config = AGENT_CONFIG['manager_agent']
+    # Load prompt templates
     prompt_templates = load_prompt_templates()
     
     # Create sub-agents
@@ -43,9 +42,9 @@ def create_manager_agent(model):
         description="Specialized agent for coordinating lyrics search and analysis",
         managed_agents=[web_agent, analysis_agent],
         additional_authorized_imports=["json", "re"],
-        planning_interval=config['planning_interval'],
-        verbosity_level=config['verbosity_level'],
-        max_steps=config['max_steps'],
+        planning_interval=3,  # Hardcoded from config
+        verbosity_level=3,  # Hardcoded from config
+        max_steps=30,  # Hardcoded from config
         prompt_templates=prompt_templates
     )
     

agents/single_agent.py

@@ -0,0 +1,43 @@
+"""
+Web agent for finding and extracting song lyrics from online sources.
+"""
+
+from loguru import logger
+from smolagents import CodeAgent, DuckDuckGoSearchTool, VisitWebpageTool, FinalAnswerTool
+
+from config import load_prompt_templates
+from tools.analysis_tools import analyze_lyrics_tool
+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
+    """
+    # Load prompt templates
+    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(),
+            VisitWebpageTool(),
+            analyze_lyrics_tool
+        ],
+        model=model,
+        additional_authorized_imports=['numpy', 'bs4'],
+        max_steps=22,
+        verbosity_level=1,
+        description="Specialized agent for finding and extracting song, specified by the user. Performs lyrics search and analysys using given tools. Provides detailed commentary using FinalAnswerTool meaning.",
+    )
+    
+    logger.info("Web agent (lyrics search) created successfully")
+    return agent

agents/web_agent.py

@@ -5,7 +5,7 @@ Web agent for finding and extracting song lyrics from online sources.
 from smolagents import CodeAgent, VisitWebpageTool
 from loguru import logger
 
-from config import AGENT_CONFIG, load_prompt_templates
+from config import load_prompt_templates
 from tools.search_tools import ThrottledDuckDuckGoSearchTool
 
 
@@ -19,8 +19,7 @@ def create_web_agent(model):
     Returns:
         A configured CodeAgent for web searches
     """
-    # Get configuration values
-    config = AGENT_CONFIG['web_agent']
+    # Load prompt templates
     prompt_templates = load_prompt_templates()
     
     # Create the throttled search tool
@@ -43,8 +42,8 @@ def create_web_agent(model):
         name="lyrics_search_agent",
         description="Specialized agent for finding and extracting complete song lyrics",
         additional_authorized_imports=["numpy", "bs4", "json", "re"],
-        max_steps=config['max_steps'],
-        verbosity_level=config['verbosity_level'],
+        max_steps=10,  # Hardcoded from config
+        verbosity_level=2,  # Hardcoded from config
         prompt_templates=prompt_templates
     )
     

app.py

@@ -5,13 +5,14 @@ Lyrics Analyzer Agent - Main Entry Point
 This module serves as the entry point for the Lyrics Analyzer application, which
 uses a system of specialized agents to search for and analyze song lyrics.  
 """
-
+import os
 from loguru import logger
 from Gradio_UI import GradioUI
 from smolagents import LiteLLMModel
 
 from config import setup_logger, load_api_keys, get_model_id, get_gradio_config
 from agents.manager_agent import create_manager_agent
+from agents.single_agent import create_single_agent
 
 
 def main():
@@ -31,14 +32,14 @@ def main():
     model = LiteLLMModel(model_id=model_id)
     
     # Create the manager agent which will create and manage the other agents
-    manager_agent = create_manager_agent(model)
+    single_agent = create_single_agent(model)
     
     # Start the Gradio UI server
     logger.info("Initializing Gradio UI and launching server")
     
     # Determine if we're in test mode (local) or production (HuggingFace)
     # HuggingFace environment has SPACE_ID environment variable
-    import os
+    
     is_test = os.environ.get('SPACE_ID') is None
     gradio_config = get_gradio_config(is_test)
     
@@ -55,7 +56,7 @@ def main():
             "server_port": gradio_config["server_port"]
         })
     
-    GradioUI(manager_agent).launch(**launch_kwargs)
+    GradioUI(single_agent).launch(**launch_kwargs)
     logger.success("Server started successfully")
 
 

config.py

@@ -57,25 +57,6 @@ SEARCH_TOOL_CONFIG = {
     "max_delay": 7.0
 }
 
-# Agent configuration
-AGENT_CONFIG = {
-    "web_agent": {
-        "max_steps": 50,
-        "verbosity_level": 2,
-        "description": "Browses the web to find original full lyrics and scrape them. Excels at building effective search queries"
-    },
-    "analysis_agent": {
-        "max_steps": 5,
-        "verbosity_level": 2,
-        "description": "You are a Song Analysis Expert with deep knowledge of music theory, lyrical interpretation, cultural contexts, and music history. Your role is to analyze song lyrics to uncover their deeper meaning, artistic significance, and historical context."
-    },
-    "manager_agent": {
-        "max_steps": 30,
-        "verbosity_level": 3, 
-        "planning_interval": 3,
-        "description": "Manages the search process and coordinates the search and analysis of song lyrics. Summarizes the results in a user-friendly format."
-    }
-}
 
 # Gradio UI configuration
 def get_gradio_config(is_test=True):