Add prompt_templates

app.py

@@ -1,6 +1,7 @@
 import os
 import time
 import random
+import yaml
 import traceback
 from loguru import logger
 from Gradio_UI import GradioUI
@@ -61,6 +62,8 @@ class LyricsSearchTool(Tool):
         # TODO: Implement lyrics search functionality
         return "Lyrics search not implemented yet"
 
+with open("prompts.yaml", 'r') as stream:
+    prompt_templates = yaml.safe_load(stream)
 
 @tool
 def analyze_lyrics_tool(song_title: str, artist: str, lyrics: str) -> str:
@@ -163,7 +166,6 @@ def _make_api_call_with_retry(model: str, prompt: str) -> str:
 
 # TODO: use DuckDuckGoSearchTool to find related information
 # for explanation in case the LLM itself is not confident or doesn't know
-#
 
 # Check if we need to use Anthropic for local testing
 use_anthropic = os.getenv("USE_ANTHROPIC", "false").lower() == "true"
@@ -183,8 +185,9 @@ web_agent = CodeAgent(
     name="lyrics_search_agent",
     description="Browses the web to find original full lyrics and scrape them. Excels at building effective search queries",
     additional_authorized_imports=["numpy", "bs4"],
-    max_steps=22,
+    max_steps=50,
     verbosity_level=2,
+    prompt_templates=prompt_templates
 )
 
 
@@ -194,8 +197,9 @@ analysis_agent = CodeAgent(
     name="lyrics_analysis_agent",
     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.",
     additional_authorized_imports=["numpy", "bs4"],
-    max_steps=50,
+    max_steps=5,
     verbosity_level=2,
+    prompt_templates=prompt_templates
 )
 
 
@@ -210,11 +214,12 @@ manager_agent = CodeAgent(
     planning_interval=5,
     verbosity_level=2,
     max_steps=15,
+    prompt_templates=prompt_templates
 )
 
 logger.info("Initializing Gradio UI and launching server")
 GradioUI(manager_agent).launch(
     debug=True, share=True, 
-    # server_name="127.0.0.1", server_port=3000
+    server_name="127.0.0.1",  server_port=3000
 )
 logger.success("Server started successfully")

pyproject.toml

@@ -9,5 +9,6 @@ dependencies = [
     "huggingface-hub>=0.29.1",
     "litellm>=1.61.20",
     "loguru>=0.7.3",
+    "pyyaml>=6.0.2",
     "smolagents>=1.9.2",
 ]