Update app.py

app.py

@@ -1,73 +1,77 @@
-from smolagents import CodeAgent, HfApiModel, load_tool, tool
+from smolagents import CodeAgent, HfApiModel, DuckDuckGoSearchTool, tool
 import requests
 from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
+from datetime import datetime, timedelta
 
-# Tool for fetching Bitcoin price
+# Modified CoinGecko Price Tool with crypto parameter
 @tool
-def get_btc_price() -> str:
-    """Fetches the current Bitcoin price in USD using the CoinGecko API."""
-    url = "https://api.coingecko.com/api/v3/simple/price?ids=bitcoin&vs_currencies=usd"
+def get_crypto_price(crypto: str) -> str:
+    """Fetches current price for any cryptocurrency using CoinGecko API."""
+    url = f"https://api.coingecko.com/api/v3/simple/price?ids={crypto}&vs_currencies=usd"
     response = requests.get(url)
     if response.status_code != 200:
-        return '{"error": "Failed to fetch Bitcoin price"}'
+        return f'{{"error": "Failed to fetch {crypto} price"}}'
     data = response.json()
-    price = data.get('bitcoin', {}).get('usd')
+    price = data.get(crypto, {}).get('usd')
     if price is None:
-        return '{"error": "Bitcoin price not found"}'
-    return f"The current Bitcoin price is ${price:.2f} USD."
+        return f'{{"error": "{crypto} price not found"}}'
+    return f"Current {crypto} price: ${price:,.2f} USD"
 
-# Tool for fetching Bitcoin news using NewsData.io API
+# DuckDuckGo Search Tool with crypto parameter
 @tool
-def get_btc_news() -> str:
-    """Fetches the latest Bitcoin news from NewsData.io API."""
-    url = "https://newsdata.io/api/1/news"
-    params = {
-        'apikey': 'YOUR_API_KEY',  # Replace with your actual API key
-        'q': 'Bitcoin',
-        'category': 'crypto',
-        'language': 'en',
-        'country': 'us',
-        'page': 1,
-        'size': 5  # Number of news articles to fetch
-    }
-    response = requests.get(url, params=params)
-    if response.status_code != 200:
-        return '{"error": "Failed to fetch Bitcoin news"}'
-    data = response.json()
-    articles = data.get('results', [])
-    if not articles:
-        return "No relevant news found."
-    news_list = []
-    for article in articles:
-        title = article.get('title', 'No title')
-        description = article.get('description', 'No description')
-        news_list.append(f"Title: {title}\nDescription: {description}\n")
-    return "\n\n".join(news_list)
-
-# Initialize the FinalAnswerTool
-final_answer = FinalAnswerTool()
+def get_crypto_news(crypto: str) -> str:
+    """Fetches latest news for any cryptocurrency using DuckDuckGo."""
+    date_filter = (datetime.now() - timedelta(days=3)).strftime("%Y-%m-%d")
+    search_query = (
+        f"{crypto} cryptocurrency "
+        f"news after:{date_filter} "
+        "site:coindesk.com OR site:cointelegraph.com OR site:bitcoin.com"
+    )
+    
+    results = DuckDuckGoSearchTool().run(
+        query=search_query,
+        max_results=5,
+        region="us-en",
+        time_limit="d"
+    )
+    
+    if not results:
+        return "No recent news found"
+    
+    news_items = []
+    for item in results:
+        news_items.append(
+            f"**{item.get('title', 'Untitled')}**\n"
+            f"{item.get('body', 'No description available')}\n"
+            f"Source: {item.get('href', '')}\n"
+        )
+    
+    return "\n\n".join(news_items)
 
-# Initialize the model
+# Initialize agent with enhanced configuration
 model = HfApiModel(
-    max_tokens=2096,
-    temperature=0.5,
     model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
-    custom_role_conversions=None,
+    max_tokens=2096,
+    temperature=0.3,
+    top_p=0.95,
 )
 
-# Create the agent
 agent = CodeAgent(
     model=model,
-    tools=[get_btc_price, get_btc_news, final_answer],  # Add your tools here
-    max_steps=6,
-    verbosity_level=1,
-    grammar=None,
-    planning_interval=None,
-    name=None,
-    description=None,
-    prompt_templates=None  # Set to None if not using prompt templates
+    tools=[get_crypto_price, get_crypto_news, DuckDuckGoSearchTool(), FinalAnswerTool()],
+    max_steps=8,
+    verbosity_level=2,
+    safety_guardrails={
+        "web_access": "filter",
+        "data_privacy": "strict"
+    },
+    rate_limit=5  # Requests per minute
 )
 
-# Launch the Gradio interface
-GradioUI(agent).launch()
+# Launch Gradio interface with crypto parameter support
+GradioUI(
+    agent,
+    default_prompt="Fetch latest price and news for:",
+    input_fields=["crypto (e.g., bitcoin)"]
+).launch()