Update app.py

app.py

@@ -2,6 +2,7 @@ from smolagents import CodeAgent, HfApiModel, tool
 import requests
 from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
+from duckduckgo_search import duckduckgo_search
 
 @tool
 def get_btc_price() -> str:
@@ -35,6 +36,22 @@ def get_btc_confidence_conseils(price: str) -> str:
     else:
         return f"Market analysis ({price}): Bearish trend detected. Consider selling."
 
+@tool
+def get_btc_info(query: str) -> str:
+    """Gathers information on Bitcoin using DuckDuckGo search.
+    
+    Args:
+        query (str): Search query related to Bitcoin (e.g., "Bitcoin price", "Bitcoin news")
+        
+    Returns:
+        str: Relevant information on Bitcoin based on the search query
+    """
+    results = duckduckgo_search(query, user_agent='Mozilla/5.0', max_results=5)
+    info = ""
+    for result in results:
+        info += f"{result['title']}: {result['url']}\n"
+    return info
+
 # Initialize the agent with all tools
 agent = CodeAgent(
     model=HfApiModel(
@@ -42,7 +59,7 @@ agent = CodeAgent(
         max_tokens=2096,
         temperature=0.5
     ),
-    tools=[get_btc_price, get_btc_confidence_conseils, FinalAnswerTool()],
+    tools=[get_btc_price, get_btc_confidence_conseils, get_btc_info, FinalAnswerTool()],
     max_steps=6,
     verbosity_level=1
 )