Update app.py

app.py

@@ -1,4 +1,4 @@
-from smolagents import CodeAgent, HfApiModel, load_tool, tool
+from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
 import requests
 from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
@@ -16,6 +16,17 @@ def get_btc_price() -> str:
         return '{"error": "Bitcoin price not found"}'
     return f"The current Bitcoin price is ${price:.2f} USD."
 
+@tool
+def get_btc_news() -> str:
+    """Fetches recent Bitcoin news articles using DuckDuckGo search."""
+    search_query = "Bitcoin news"
+    search_tool = DuckDuckGoSearchTool()
+    results = search_tool(search_query)
+    if not results:
+        return "No news articles found."
+    news = "\n".join([f"{i+1}. {result['title']} - {result['url']}" for i, result in enumerate(results[:5])])
+    return f"Recent Bitcoin News:\n{news}"
+
 # Initialize the FinalAnswerTool
 final_answer = FinalAnswerTool()
 
@@ -27,14 +38,10 @@ model = HfApiModel(
     custom_role_conversions=None,
 )
 
-# Load additional tools if needed
-# For example, to load an image generation tool:
-# image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
-
 # Create the agent
 agent = CodeAgent(
     model=model,
-    tools=[get_btc_price, final_answer],  # Add your tools here
+    tools=[get_btc_price, get_btc_news, final_answer],  # Add your tools here
     max_steps=6,
     verbosity_level=1,
     grammar=None,