Update app.py

app.py

@@ -1,8 +1,53 @@
-from smolagents import CodeAgent, HfApiModel, load_tool, tool, DuckDuckGoSearchTool
+from huggingface_hub import login
+from smolagents import (
+    CodeAgent,
+    ToolCallingAgent,
+    HfApiModel,
+    DuckDuckGoSearchTool,
+    tool
+)
+import re
 import requests
-from tools.final_answer import FinalAnswerTool
-from Gradio_UI import GradioUI
+from markdownify import markdownify
+from requests.exceptions import RequestException
+from transformers import pipeline
+
+# Login to Hugging Face Hub
+login()
+
+# Define the model ID
+model_id = "Qwen/Qwen2.5-Coder-32B-Instruct"
+
+# Define the visit_webpage tool
+@tool
+def visit_webpage(url: str) -> str:
+    """Visits a webpage at the given URL and returns its content as a markdown string.
+
+    Args:
+        url: The URL of the webpage to visit.
+
+    Returns:
+        The content of the webpage converted to Markdown, or an error message if the request fails.
+    """
+    try:
+        # Send a GET request to the URL
+        response = requests.get(url)
+        response.raise_for_status()  # Raise an exception for bad status codes
+
+        # Convert the HTML content to Markdown
+        markdown_content = markdownify(response.text).strip()
 
+        # Remove multiple line breaks
+        markdown_content = re.sub(r"\n{3,}", "\n\n", markdown_content)
+
+        return markdown_content
+
+    except RequestException as e:
+        return f"Error fetching the webpage: {str(e)}"
+    except Exception as e:
+        return f"An unexpected error occurred: {str(e)}"
+
+# Define the get_btc_price tool
 @tool
 def get_btc_price() -> str:
     """Fetches the current Bitcoin price in USD using the CoinGecko API."""
@@ -16,54 +61,81 @@ 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_data(query: str) -> str:
-    """
-    Fetches Bitcoin data using the CoinGecko API.
-    
-    Args:
-        query (str): The type of data to fetch. This can be one of the following: "price", "market_cap", or "circulating_supply".
-        
-    Returns:
-        str: The requested Bitcoin data.
-    """
-    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 '{"error": "Invalid query"}'
-
-# Initialize the FinalAnswerTool
-final_answer = FinalAnswerTool()
-
-# Initialize the model
+# Initialize the Hugging Face model
 model = HfApiModel(
     max_tokens=2096,
     temperature=0.5,
-    model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
+    model_id=model_id,
     custom_role_conversions=None,
 )
 
-# Create the agent
-agent = CodeAgent(
+# Create the Web Search Agent
+web_search_agent = ToolCallingAgent(
+    tools=[DuckDuckGoSearchTool(), visit_webpage],
+    model=model,
+    max_steps=10,
+    name="web_search",
+    description="Runs web searches for you. Give it your query as an argument.",
+)
+
+# Create the Manager Agent
+manager_agent = CodeAgent(
+    tools=[get_btc_price],
+    model=model,
+    managed_agents=[web_search_agent],
+    additional_authorized_imports=["time", "numpy", "pandas", "transformers"],
+    max_steps=10,
+    verbosity_level=1,
+    grammar=None,
+    planning_interval=None,
+    name="btc_manager",
+    description="Manages BTC-related tasks, including web searches and price fetching.",
+    prompt_templates=None  # Set to None if not using prompt templates
+)
+
+# Define a function to summarize the content using a Hugging Face model
+def summarize_content(content: str) -> str:
+    """Summarizes the given content using a Hugging Face summarization model."""
+    summarizer = pipeline("summarization")
+    summary = summarizer(content, max_length=130, min_length=30, do_sample=False)
+    return summary[0]['summary_text']
+
+# Define a function to orchestrate the search and summarization
+def search_and_summarize(query: str) -> str:
+    """Searches for the given query, visits the first result, and summarizes the content."""
+    # Run the web search agent
+    search_result = manager_agent.run(f"Search for '{query}' and visit the first result.")
+    
+    # Extract the URL from the search result
+    try:
+        url = re.search(r'https?://[^\s]+', search_result).group(0)
+    except AttributeError:
+        return "Error: Could not extract URL from search result."
+    
+    # Visit the webpage and get the content
+    webpage_content = visit_webpage(url)
+    
+    # Summarize the content
+    summary = summarize_content(webpage_content)
+    
+    return f"Title: {url}\nSummary: {summary}"
+
+# Initialize the FinalAnswerTool
+final_answer = FinalAnswerTool()
+
+# Create the BTC Agent
+btc_agent = CodeAgent(
     model=model,
-    tools=[get_btc_price, get_btc_data, DuckDuckGoSearchTool(), final_answer],  # Add your tools here
-    max_steps=6,
+    tools=[get_btc_price, final_answer, search_and_summarize],
+    max_steps=10,
     verbosity_level=1,
     grammar=None,
     planning_interval=None,
-    name=None,
-    description=None,
+    name="btc_agent",
+    description="Handles BTC-related tasks, including price fetching and news summarization.",
     prompt_templates=None  # Set to None if not using prompt templates
 )
 
 # Launch the Gradio interface
-GradioUI(agent).launch()
+from Gradio_UI import GradioUI
+GradioUI(btc_agent).launch()