Update app.py

app.py

@@ -12,6 +12,8 @@ import requests
 from io import BytesIO
 from PIL import Image
 
+web_serach_tool = DuckDuckGoSearchTool()
+
 @tool
 def generate_image(imageprompt: str) -> Image.Image:
     """A tool that generates an image based on a given prompt.
@@ -30,6 +32,23 @@ def generate_image(imageprompt: str) -> Image.Image:
     else:
         raise Exception("Failed to generate image.")
 
+
+@tool
+def get_news(topic:str) -> str:
+    """A tool that scraps the web to find latest news article related to a specific topic
+
+    Args:
+        A detailed topic that will act as the search query
+    
+    Returns:
+        A latest news article from the web based on the provided topic
+    
+    """
+
+    search_result = web_serach_tool.run(topic)
+
+    return search_result
+
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.
@@ -67,7 +86,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, generate_image], ## add your tools here (don't remove final answer)
+    tools=[final_answer, get_news, generate_image], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,