Adding the band name

README.md

@@ -1,8 +1,8 @@
 ---
-title: First Agent Template
+title: HS First Agent Template
 emoji: ⚡
-colorFrom: pink
-colorTo: yellow
+colorFrom: #14213dff
+colorTo: #737c9dff
 sdk: gradio
 sdk_version: 5.15.0
 app_file: app.py
@@ -15,4 +15,4 @@ tags:
 - agent-course
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -4,8 +4,11 @@ import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
-
+from tools.band_name_generator import generate_band_name
 from Gradio_UI import GradioUI
+from dotenv import load_dotenv
+load_dotenv()
+
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
@@ -55,7 +58,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer], ## add your tools here (don't remove final answer)
+    tools=[final_answer, generate_band_name], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,

tools/band_name_generator.py

@@ -0,0 +1,25 @@
+from smolagents import tool
+import random
+
+@tool
+def generate_band_name(first_name: str) -> str:
+    """Generates a random band name using the provided first name as inspiration.
+    
+    Args:
+        first_name: The first name to use as inspiration for the band name
+    """
+    
+    adjectives = [
+        "Cosmic", "Electric", "Mystic", "Savage", "Silent", 
+        "Neon", "Quantum", "Midnight", "Crystal", "Thunder"
+    ]
+    
+    nouns = [
+        "Dragons", "Warriors", "Prophets", "Wolves", "Riders",
+        "Pirates", "Bandits", "Knights", "Wizards", "Rebels"
+    ]
+    
+    # Generate the band name using the first name and random elements
+    band_name = f"{random.choice(adjectives)} {first_name}'s {random.choice(nouns)}"
+    
+    return f"Your generated band name is: {band_name}"