implement serpapi youtube and other tools properly

app.py

@@ -4,13 +4,15 @@ import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
+from tools.visit_webpage import VisitWebpageTool
+from tools.web_search import DuckDuckGoSearchTool
 import requests
 from Gradio_UI import GradioUI
 import os
 from dotenv import load_dotenv
 load_dotenv()
 
-HUGGINGFACE_API_KEY = os.getenv("HUGGINGFACE_API_KEY")
+SERPAPI_KEY = os.getenv("SERPAPI_KEY")
 
 @tool
 def youtube_search(query: str, max_results: int = 5) -> str:
@@ -27,7 +29,7 @@ def youtube_search(query: str, max_results: int = 5) -> str:
     
     try:
         # Replace with your actual SerpAPI key or set it as an environment variable
-        api_key = HUGGINGFACE_API_KEY  # Add your SerpAPI key here
+        api_key = SERPAPI_KEY  # Add your SerpAPI key here
         url = "https://serpapi.com/search.json"
         params = {
             "engine": "youtube",
@@ -95,7 +97,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, DuckDuckGoSearchTool, get_current_time_in_timezone], ## add your tools here (don't remove final answer)
+    tools=[final_answer, get_current_time_in_timezone, youtube_search, DuckDuckGoSearchTool(), VisitWebpageTool()], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,

tools/visit_webpage.py

@@ -3,6 +3,7 @@ from smolagents.tools import Tool
 import requests
 import markdownify
 import smolagents
+import re
 
 class VisitWebpageTool(Tool):
     name = "visit_webpage"