Update app.py

app.py

@@ -1,24 +1,23 @@
-from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
+from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
 import datetime
 import requests
 import pytz
-import yaml
-from tools.final_answer import FinalAnswerTool
-
 from Gradio_UI import GradioUI
 
-# Below is an example of a tool that does nothing. Amaze us with your creativity !
+# ==========================================
+# 1. DEFINE YOUR TOOLS
+# ==========================================
+
+base_search_tool = DuckDuckGoSearchTool()
+
 @tool
 def custom_duckduckgo_search(query: str) -> str:
     """A tool that performs a web search using DuckDuckGo to find current information.
     Args:
         query: The search query string to look up on the web.
     """
-    base_search_tool = DuckDuckGoSearchTool()
     try:
-        # Call the underlying smolagents search tool
-        result = base_search_tool(query)
-        return str(result)
+        return str(base_search_tool(query))
     except Exception as e:
         return f"Error performing search: {str(e)}"
 
@@ -29,45 +28,95 @@ def get_current_time_in_timezone(timezone: str) -> str:
         timezone: A string representing a valid timezone (e.g., 'America/New_York').
     """
     try:
-        # Create timezone object
         tz = pytz.timezone(timezone)
-        # Get current time in that timezone
         local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
         return f"The current local time in {timezone} is: {local_time}"
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
+@tool
+def get_weather(location: str) -> str:
+    """A tool that fetches the current weather for a given city or location.
+    Args:
+        location: The name of the city (e.g., 'Tokyo', 'New York', 'Paris').
+    """
+    try:
+        # Step 1: Get latitude and longitude
+        geo_url = f"https://geocoding-api.open-meteo.com/v1/search?name={location}&count=1"
+        geo_data = requests.get(geo_url).json()
+        if "results" not in geo_data:
+            return f"Could not find coordinates for {location}."
+        
+        lat = geo_data["results"][0]["latitude"]
+        lon = geo_data["results"][0]["longitude"]
+        
+        # Step 2: Fetch the weather using the coordinates
+        weather_url = f"https://api.open-meteo.com/v1/forecast?latitude={lat}&longitude={lon}&current_weather=true"
+        weather_data = requests.get(weather_url).json()
+        
+        current = weather_data["current_weather"]
+        temp = current["temperature"]
+        wind = current["windspeed"]
+        
+        return f"The current weather in {location} is {temp}°C with wind speeds of {wind} km/h."
+    except Exception as e:
+        return f"Error fetching weather: {str(e)}"
 
-final_answer = FinalAnswerTool()
+@tool
+def get_crypto_price(crypto_id: str) -> str:
+    """Fetches the current live price of a cryptocurrency in USD.
+    Args:
+        crypto_id: The full CoinGecko ID of the coin (e.g., 'bitcoin', 'ethereum', 'dogecoin').
+    """
+    try:
+        url = f"https://api.coingecko.com/api/v3/simple/price?ids={crypto_id}&vs_currencies=usd"
+        response = requests.get(url).json()
+        if crypto_id.lower() in response:
+            price = response[crypto_id.lower()]["usd"]
+            return f"The current price of {crypto_id.capitalize()} is ${price} USD."
+        else:
+            return f"Could not find price for '{crypto_id}'. Make sure to use the full name (e.g., 'bitcoin' not 'BTC')."
+    except Exception as e:
+        return f"Error fetching price: {str(e)}"
 
-# If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
-# model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud' 
+# ==========================================
+# 2. SETUP MODEL AND EXTERNAL TOOLS
+# ==========================================
 
+# Initialize the model
 model = HfApiModel(
-max_tokens=2096,
-temperature=0.5,
-model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
-custom_role_conversions=None,
+    max_tokens=2096,
+    temperature=0.5,
+    model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
 )
 
-
-# Import tool from Hub
+# Load the image generation tool from Hugging Face Hub
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
-with open("prompts.yaml", 'r') as stream:
-    prompt_templates = yaml.safe_load(stream)
-    
+# ==========================================
+# 3. CONFIGURE AND LAUNCH THE AGENT
+# ==========================================
+
+# Compile all our tools into a single list
+my_tools = [
+    custom_duckduckgo_search,
+    get_current_time_in_timezone,
+    get_weather,
+    get_crypto_price,
+    image_generation_tool
+]
+
+# Note: Removed `prompt_templates` here to use the built-in, error-free defaults
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, custom_duckduckgo_search, get_current_time_in_timezone], ## add your tools here (don't remove final answer)
-    max_steps=50,
+    tools=my_tools,
+    max_steps=10,       # Bumped up to 10 to give the agent more room to think!
     verbosity_level=1,
     grammar=None,
     planning_interval=None,
-    name=None,
-    description=None,
-    prompt_templates=prompt_templates
+    name="Super_Analyst",
+    description="An agent capable of checking weather, stock prices, web searching, and drawing images."
 )
 
-
+# Launch the UI
 GradioUI(agent).launch()