Update app.py

app.py

@@ -1,34 +1,39 @@
 from smolagents import CodeAgent, HfApiModel, tool
 import requests
-from requests.exceptions import RequestException  # Add this import
+from requests.exceptions import RequestException  # Import request exception handling
 from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 import markdownify
 import re
+from smolagents.tools.web import web_search as web_search_tool  # Import web search from SmolAgents
 
 @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"
-    response = requests.get(url)
-    if response.status_code != 200:
-        return '{"error": "Failed to fetch Bitcoin price"}'
-    data = response.json()
-    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."
+def web_search(query: str) -> str:
+    """Performs a web search and returns the top search results.
+    
+    Args:
+        query: The search query string.
+
+    Returns:
+        A formatted string with the top search results, or an error message if the request fails.
+    """
+    try:
+        results = web_search_tool(query)
+        return results if results else "No relevant results found."
+    except Exception as e:
+        return f"Error performing web search: {str(e)}"
 
 @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
 
@@ -40,7 +45,7 @@ def visit_webpage(url: str) -> str:
 
         return markdown_content
 
-    except RequestException as e:  # Updated to use the imported RequestException
+    except RequestException as e:
         return f"Error fetching the webpage: {str(e)}"
     except Exception as e:
         return f"An unexpected error occurred: {str(e)}"
@@ -50,7 +55,7 @@ final_answer = FinalAnswerTool()
 
 # Initialize the model
 model = HfApiModel(
-    max_tokens=1500,  # Reduced from 2096 to avoid token limit errors
+    max_tokens=1500,  # Adjusted to avoid token limit errors
     temperature=0.5,
     model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
     custom_role_conversions=None,
@@ -59,7 +64,7 @@ model = HfApiModel(
 # Create the agent
 agent = CodeAgent(
     model=model,
-    tools=[get_btc_price, visit_webpage, final_answer],  # Add your tools here
+    tools=[web_search, visit_webpage, final_answer],  # Added web_search tool
     max_steps=6,
     verbosity_level=1,
     grammar=None,