Update app.py

app.py

@@ -1,9 +1,9 @@
-from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, tool
+from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
 import requests
 from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
-# Fetch the current Bitcoin price
+# Tool for fetching Bitcoin price
 @tool
 def get_btc_price() -> str:
     """Fetches the current Bitcoin price in USD using the CoinGecko API."""
@@ -17,28 +17,35 @@ def get_btc_price() -> str:
         return '{"error": "Bitcoin price not found"}'
     return f"The current Bitcoin price is ${price:.2f} USD."
 
-# Fetch the latest Bitcoin news
+
+# Tool for fetching Bitcoin news using DuckDuckGo search
 @tool
 def get_btc_news() -> str:
-    """Fetches the latest Bitcoin news articles using DuckDuckGoSearchTool."""
+    """Fetches the latest Bitcoin news from DuckDuckGo search."""
     search_query = "latest Bitcoin news"
-    search_results = DuckDuckGoSearchTool()(search_query)
-    if not search_results:
-        return '{"error": "Failed to fetch Bitcoin news"}'
-    
-    # Extract titles and meta descriptions from the search results
-    news_titles = [result['title'] for result in search_results]
-    news_descriptions = [result['snippet'] for result in search_results]
+    # Performing a DuckDuckGo search using the DuckDuckGoSearchTool
+    search_results = DuckDuckGoSearchTool().run(search_query)
     
-    # Format the news list into a readable string
-    news_list = "\n".join([f"**{title}**\n{description}" for title, description in zip(news_titles, news_descriptions)])
-    
-    return f"Here are the latest Bitcoin news articles:\n\n{news_list}"
+    # Ensure search_results is a list of articles with titles and meta descriptions
+    try:
+        # Process the results and extract titles and descriptions
+        articles = []
+        for result in search_results['RelatedTopics']:
+            title = result.get('Text', '')
+            description = result.get('FirstURL', '')
+            articles.append(f"Title: {title}\nDescription: {description}\n")
+        if articles:
+            return "\n\n".join(articles)
+        else:
+            return "No relevant news found."
+    except Exception as e:
+        return f"Error fetching Bitcoin news: {str(e)}"
+
 
-# Initialize the FinalAnswerTool for final response processing
+# Initialize the FinalAnswerTool
 final_answer = FinalAnswerTool()
 
-# Initialize the model (Qwen2.5-Coder-32B-Instruct from Hugging Face)
+# Initialize the model
 model = HfApiModel(
     max_tokens=2096,
     temperature=0.5,
@@ -49,7 +56,7 @@ model = HfApiModel(
 # Create the agent
 agent = CodeAgent(
     model=model,
-    tools=[get_btc_price, get_btc_news, final_answer],  # Add the tools here
+    tools=[get_btc_price, get_btc_news, final_answer],  # Add your tools here
     max_steps=6,
     verbosity_level=1,
     grammar=None,
@@ -59,5 +66,5 @@ agent = CodeAgent(
     prompt_templates=None  # Set to None if not using prompt templates
 )
 
-# Launch the Gradio UI for interaction
+# Launch the Gradio interface
 GradioUI(agent).launch()