Update app.py

app.py

@@ -8,6 +8,97 @@ from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
 
+@tool
+def get_stock_price(stock_symbol: str) -> str:
+    """A tool that fetches the current stock price for given stock symbol.
+
+    Args:
+        stock_symbol: A string representing a valid NYSE (USA) stock ticker symbol (e.g., "AAPL" for Apple Inc., Consumer Electronics, Market Cap $3,674.40B).
+    """
+    import yfinance as yf
+    stock_symbol = stock_symbol.upper()  # Ensure uppercase
+    stock = yf.Ticker(stock_symbol + ".NYSE")
+    data = stock.history(period='5d')  # Get last 5 days to handle non-trading days
+
+    if data.empty:
+        return {"error": f"No data found for {stock_symbol}."}
+
+    latest_data = data.iloc[-1]
+    current_price = round(latest_data['Close'], 2)
+
+    return current_price
+
+
+@tool
+def get_current_time_in_timezone(timezone: str) -> str:
+    """A tool that fetches the current local time in a specified timezone.
+
+    Args:
+        timezone: A string representing a valid timezone (e.g., 'America/New_York').
+
+    Returns:
+        A string displaying the local date and time in the specified timezone.
+    """
+    try:
+        # Validate timezone
+        if timezone not in pytz.all_timezones:
+            return f"❌ Error: '{timezone}' is not a valid timezone. Please provide a correct timezone (e.g., 'America/New_York')."
+
+        # Get the current time in the specified timezone
+        tz = pytz.timezone(timezone)
+        local_time = datetime.datetime.now(tz)
+
+        # Format the output with day, date, and time
+        formatted_time = local_time.strftime("%A, %Y-%m-%d %H:%M:%S %Z")
+
+        return f"🕒 The current local time in **{timezone}** is: **{formatted_time}**"
+
+    except Exception as e:
+        return f"⚠️ Error fetching time for timezone '{timezone}': {str(e)}"
+
+@tool
+def get_weather(location: str = None, zipcode: int = None) -> str:
+    """Fetches the current weather for a specified location or zip code.
+    
+    Args:
+        location: A string representing a valid city name or region (e.g., 'New York').
+        zipcode: An integer representing a valid zip code (e.g., 10001 for NYC).
+    
+    Returns:
+        A string describing the weather conditions, temperature, and humidity.
+    """
+    temperature = data["main"]["temp"]
+    conditions = data["weather"][0]["description"]
+    humidity = data["main"]["humidity"]
+
+    return (f"🌤️ Weather in {location or zipcode} is: {conditions.capitalize()}, "
+                f"Temperature: {temperature}°C, Humidity: {humidity}%")
+    
+   #return f"The current weather in {location,zipcode} is: {temperature,}"
+
+
+@tool
+def getMarsWeather() -> str:
+    """
+    A tool that fetches the current weather on Mars using NASA's InSight API.
+    Returns:
+        A string containing the sol (Martian day), average temperature, and wind speed on Mars.
+    """
+    insightURL = 'https://api.nasa.gov/insight_weather/?api_key={}&feedtype=json&ver=1.0'.format(NASA_API_KEY)
+    response = requests.get(insightURL)
+    marsWeatherRaw = response.json()
+    firstKey = next(iter(marsWeatherRaw))
+    marsWeather = {
+        'sol': firstKey,
+        'temperature': marsWeatherRaw[firstKey]['AT']['av'],
+        'wind_speed': marsWeatherRaw[firstKey]['HWS']['av']
+    }
+    outputStr = "The temperature on Mars on Sol {sol} is {temperature} C and wind speeds of {wind_speed} m/sec.".format(
+    sol=marsWeather['sol'],
+    temperature=marsWeather['temperature'],
+    wind_speed=marsWeather['wind_speed'])
+    
+    return outputStr
 
 
 final_answer = FinalAnswerTool()
@@ -32,7 +123,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer,get_stock_price,get_weather,image_generation_tool,get_current_time_in_timezone,DuckDuckGoSearchTool()], ## add your tools here (don't remove final answer)
+    tools=[final_answer,get_weather,image_generation_tool,get_current_time_in_timezone,get_stock_price,DuckDuckGoSearchTool()], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,