Update app.py

add visit_webpage,web_search

app.py

@@ -3,7 +3,7 @@ import datetime
 import requests
 import pytz
 import yaml
-from tools.final_answer import FinalAnswerTool
+from tools.final_answer import FinalAnswerTool,VisitWebpageTool,DuckDuckGoSearchTool
 
 from Gradio_UI import GradioUI
 
@@ -36,6 +36,35 @@ def get_current_time_in_timezone(timezone: str) -> str:
 
 final_answer = FinalAnswerTool()
 
+@tool
+def visit_webpage(url: str) -> str:
+    """A tool that visits a webpage at the given URL and reads its content as a markdown string.
+    
+    Args:
+        url: The URL of the webpage to visit.
+    """
+    # Initialize the VisitWebpageTool
+    webpage_tool = VisitWebpageTool()
+    
+    # Call the forward method of VisitWebpageTool to get the markdown content
+    return webpage_tool.forward(url)
+
+
+@tool
+def web_search(query: str, max_results: int = 5) -> str:
+    """A tool that performs a DuckDuckGo web search based on your query and returns the top search results.
+    
+    Args:
+        query: The search query to perform.
+        max_results: The maximum number of results to return.
+    """
+    # Initialize the DuckDuckGoSearchTool
+    search_tool = DuckDuckGoSearchTool(max_results=max_results)
+    
+    # Call the forward method of DuckDuckGoSearchTool to get the search results
+    return search_tool.forward(query)
+    
+
 # If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
 # model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud' 
 
@@ -55,7 +84,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer,get_current_time_in_timezone,image_generation_tool], ## add your tools here (don't remove final answer)
+    tools=[final_answer,get_current_time_in_timezone,image_generation_tool,visit_webpage,web_search], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,