Update app.py

app.py

@@ -9,20 +9,20 @@ from tools.web_search import DuckDuckGoSearchTool
 from bs4 import BeautifulSoup
 from duckduckgo_search import DDGS
 import re
-from typing import List, Dict, Any, Optional
+from typing import List, Dict, Any
 from Gradio_UI import GradioUI
 import time  # For rate limiting
 
 
 @tool
-def get_healthy_cheat_meal(diet_type: str) -> str:
+def get_healthy_cheat_meal(diet_type: str) -> List[str]:
     """Search for healthy sweet dish alternatives based on specific diet types.
 
     Args:
         diet_type: The specific diet type to search for (e.g., Keto, Paleo, Vegan, Vegetarian, Non-Vegetarian).
 
     Returns:
-        A string containing the recipe name/title and the website url.
+        A list of strings containing the recipe name/title and the website url.
             
     """
     query = f"healthy {diet_type} sweet dish recipe"
@@ -64,7 +64,35 @@ def get_healthy_cheat_meal(diet_type: str) -> str:
     except Exception as e:
         return f"An error occurred: {str(e)}"
 
+@tool
+def visit_webpage(url: str) -> str:
+    """Visits a webpage and extracts its text content.
+
+    Args:
+        url: The URL of the webpage to visit.
 
+    Returns:
+        The text content of the webpage, or an error message if the page could not be accessed or parsed.
+    """
+    try:
+        response = requests.get(url, timeout=10, headers={
+            'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36'
+        })
+        response.raise_for_status()  # Raise an exception for bad status codes (4xx or 5xx)
+
+        soup = BeautifulSoup(response.content, 'html.parser')
+
+        # Extract all text, removing script and style elements
+        for script in soup(["script", "style"]):
+            script.extract()
+        text = soup.get_text(strip=True)
+
+        return text
+
+    except requests.exceptions.RequestException as e:
+        return f"Error accessing webpage: {e}"
+    except Exception as e:
+        return f"An error occurred while processing the webpage: {e}"
         
 # Initialize tools
 final_answer = FinalAnswerTool()
@@ -89,7 +117,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[get_healthy_cheat_meal], ## add your tools here (don't remove final answer)
+    tools=[get_healthy_cheat_meal,visit_webpage,final_answer], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,