Update app.py

app.py

@@ -4,6 +4,7 @@ import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
+from bs4 import BeautifulSoup
 
 from Gradio_UI import GradioUI
 
@@ -18,8 +19,57 @@ def my_cutom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return
     """
     return "What magic will you build ?"
 
+# Tool to fetch latest news headlines from different news outlets
 @tool
-def get_current_time_in_timezone(timezone: str) -> dict:
+def get_latest_news(source: str) -> str:
+    """Fetches latest headlines from specified news source.
+    
+    Args:
+        source: Name of news source from supported list (BBC News, CNN, Reuters, AP, 
+               Fox News, CBS News, Wall Street Journal, Daily Mail)
+    
+    Returns:
+        String containing latest headlines or error message
+    """
+    news_urls = {
+        "BBC News": "https://www.bbc.com/news",
+        "CNN": "https://www.cnn.com",
+        "Reuters": "https://www.reuters.com",
+        "AP": "https://apnews.com",
+        "Fox News": "https://www.foxnews.com",
+        "CBS News": "https://www.cbsnews.com",
+        "Wall Street Journal": "https://www.wsj.com",
+        "Daily Mail": "https://www.dailymail.co.uk/home/index.html"
+    }
+
+    try:
+        # Get the URL for the requested source
+        url = news_urls.get(source)
+        if not url:
+            return f"Error: {source} is not a supported news source"
+        
+        # Fetch and parse the page
+        response = requests.get(url, timeout=10)
+        response.raise_for_status()
+        soup = BeautifulSoup(response.text, 'html.parser')
+        
+        # Extract headlines based on source-specific patterns
+        headlines = []
+        if source == "BBC News":
+            headlines = [h3.text.strip() for h3 in soup.select('h3[class^="gs-c-promo-heading"]')][:5]
+        elif source == "CNN":
+            headlines = [div.text.strip() for div in soup.select('div[class^="container__headline"]')][:5]
+        elif source == "Reuters":
+            headlines = [a.text.strip() for a in soup.select('a[data-testid="Heading"]')][:5]
+        # Add other sources' parsing logic here...
+        
+        return f"Latest headlines from {source}:\n- " + "\n- ".join(headlines[:5])
+        
+    except Exception as e:
+        return f"Error fetching news from {source}: {str(e)}"
+
+@tool
+def get_current_time_in_timezone(timezone: str) -> str:
     """Fetches the current local time in a specified timezone and provides a human-friendly description.
     
     Args:
@@ -52,7 +102,6 @@ model_id='https://jc26mwg228mkj8dw.us-east-1.aws.endpoints.huggingface.cloud',#
 custom_role_conversions=None,
 )
 
-
 # Import tool from Hub
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
@@ -61,9 +110,9 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, get_current_time_in_timezone], ## add your tools here (don't remove final answer)
+    tools=[final_answer, get_latest_news], ## add your tools here (don't remove final answer)
     max_steps=6,
-    verbosity_level=1,
+    verbosity_level=0,
     grammar=None,
     planning_interval=None,
     name=None,