Update app.py

app.py

@@ -4,7 +4,8 @@ import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
-from tools.visit_webpage import VisitWebpageTool
+import os
+
 from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
@@ -33,9 +34,57 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
+@tool
+def get_ai_safety_updates(category: str = "general", max_results: int = 5) -> str:
+    """
+    A tool that fetches the latest news updates on AI safety developments and categorizes them.
+    
+    Args:
+        category: The topic or category for the news updates (e.g., 'research', 'policy', 'industry').
+        max_results: The maximum number of news articles to retrieve.
+    
+    Returns:
+        A string summarizing the latest AI safety news updates for the specified category.
+    """
+
+    api_key = os.getenv("NEWS_API_KEY")
+    if not api_key:
+        return "Error: NEWS_API_KEY is not set in your environment."
+
+    # Prepare the query; you can customize it further as needed
+    query = f" Latest AI safety research News {category}"
+    
+    # News API endpoint
+    url = "https://newsapi.org/v2/everything"
+    
+    params = {
+        "q": query,
+        "sortBy": "publishedAt",
+        "pageSize": max_results,
+        "apiKey": api_key
+    }
+    
+    response = requests.get(url, params=params)
+    
+    if response.status_code == 200:
+        articles = response.json().get("articles", [])
+        if not articles:
+            return f"No recent updates found for category '{category}'."
+        
+        # Build a summary of the news articles
+        summary = f"Latest AI safety updates for category '{category}':\n"
+        for article in articles:
+            title = article.get("title", "No title")
+            source = article.get("source", {}).get("name", "Unknown source")
+            published = article.get("publishedAt", "No date")
+            summary += f"- {title} (Source: {source}, Date: {published})\n"
+        return summary
+    else:
+        return f"Error fetching news: {response.status_code} - {response.text}"
+
 
 final_answer = FinalAnswerTool()
-web = VisitWebpageTool()
+
 
 # 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' 
@@ -56,7 +105,8 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, web, get_current_time_in_timezone], ## add your tools here (don't remove final answer)
+    tools=[final_answer, get_current_time_in_timezone,
+          get_ai_safety_updates], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,