Update app.py

app.py

@@ -1,8 +1,7 @@
-from smolagents import CodeAgent, HfApiModel, tool
+from smolagents import CodeAgent, HfApiModel, load_tool, 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:
@@ -15,53 +14,50 @@ def get_btc_price() -> str:
     price = data.get('bitcoin', {}).get('usd')
     if price is None:
         return '{"error": "Bitcoin price not found"}'
-    return f"The current Bitcoin price is ${price:,.2f} USD."
+    return f"The current Bitcoin price is ${price:.2f} USD."
 
 @tool
-def get_btc_confidence_conseils(price: str) -> str:
-    """Analyzes Bitcoin market confidence using price data.
-    
-    Args:
-        price (str): The current Bitcoin price in USD format (e.g., "$50,000.00")
-        
-    Returns:
-        str: Market analysis and trading advice based on current conditions
-    """
-    numeric_price = float(price.replace("$", "").replace(",", ""))
-    
-    if numeric_price > 50000:
-        return f"Market analysis ({price}): Bullish trend detected. Consider buying."
-    elif numeric_price > 30000:
-        return f"Market analysis ({price}): Neutral trend detected. Consider holding."
+def get_btc_data(query: str) -> str:
+    """Fetches Bitcoin data using the CoinGecko API."""
+    url = f"https://api.coingecko.com/api/v3/coins/bitcoin?localization=false&tickers=false&market_data=true&community_data=false&developer_data=false&sparkline=false"
+    response = requests.get(url)
+    if response.status_code != 200:
+        return '{"error": "Failed to fetch Bitcoin data"}'
+    data = response.json()
+    if query == "price":
+        return f"The current Bitcoin price is ${data['market_data']['current_price']['usd']:.2f} USD."
+    elif query == "market_cap":
+        return f"The current Bitcoin market capitalization is ${data['market_data']['market_cap']['usd']:.2f} USD."
+    elif query == "circulating_supply":
+        return f"The current Bitcoin circulating supply is {data['market_data']['circulating_supply']:.2f} BTC."
     else:
-        return f"Market analysis ({price}): Bearish trend detected. Consider selling."
+        return '{"error": "Invalid query"}'
 
-@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 FinalAnswerTool
+final_answer = FinalAnswerTool()
+
+# Load the DuckDuckGoSearchTool
+duckduckgo_search_tool = load_tool("smolai/duckduckgo-search-tool", trust_remote_code=True)
+
+# Initialize the model
+model = HfApiModel(
+    max_tokens=2096,
+    temperature=0.5,
+    model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
+    custom_role_conversions=None,
+)
 
-# Initialize the agent with all tools
+# Create the agent
 agent = CodeAgent(
-    model=HfApiModel(
-        model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
-        max_tokens=2096,
-        temperature=0.5
-    ),
-    tools=[get_btc_price, get_btc_confidence_conseils, get_btc_info, FinalAnswerTool()],
+    model=model,
+    tools=[get_btc_price, get_btc_data, duckduckgo_search_tool, final_answer],  # Add your tools here
     max_steps=6,
-    verbosity_level=1
+    verbosity_level=1,
+    grammar=None,
+    planning_interval=None,
+    name=None,
+    description=None,
+    prompt_templates=None  # Set to None if not using prompt templates
 )
 
 # Launch the Gradio interface