Update app.py

app.py

@@ -11,7 +11,31 @@ from Gradio_UI import GradioUI
 
 search_tool = DuckDuckGoSearchTool()
 
-   
+news_sites = ["https://www.cnn.com/"]
+
+site_config = {
+    "https://www.cnn.com/": {'tag': 'h2', 'class': 'headline'}
+    }
+}
+
+@tool
+def get_latest_news(news_sites) ->dict:
+    """ Tool returns news headlines"""
+    headlines = {}
+    for site in news_sites:
+        try:
+           config = site_config.get(site,{'tag': 'h2', 'class:':'headline'})
+            response = requests.get(site)
+            response.raise_for_status()
+            soup = BeautifulSoup(response.content, 'html.parser')
+            site_headlines=soup.find_all(config['tag'], class = config['class'])
+            headlines[site] = [headline.text for headline in site_headlines]
+        except requests.RequestException as e:
+            headlines[site] = f"Error fetching news:{e}"
+    return headlines
+        
+
+
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -51,7 +75,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, get_current_time_in_timezone,image_generation_tool, search_tool], ## add your tools here (don't remove final answer)
+    tools=[final_answer, get_current_time_in_timezone,image_generation_tool, search_tool, get_latest_news], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,