Update app.py

app.py

@@ -2,6 +2,8 @@ from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
 import datetime
 import requests
 import pytz
+import yfinance as yf
+import pandas as pd
 import yaml
 from tools.final_answer import FinalAnswerTool
 
@@ -18,6 +20,37 @@ def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return
     """
     return "What magic will you build ?"
 
+@tool
+def get_sp500_price(date_str)-> str:
+    """
+    Fetches the S&P 500 price for a given date.
+
+    Parameters:
+    date_str (str): The date in 'YYYY-MM-DD' format.
+
+    Returns:
+    float: The closing price of the S&P 500 on the given date.
+    """
+    try:
+        # Convert the input string to a datetime object
+        date = datetime.datetime.strptime(date_str, '%Y-%m-%d')
+        
+        # Define the ticker symbol for S&P 500
+        sp500 = '^GSPC'
+        
+        # Fetch the data for the given date
+        data = yf.download(sp500, start=date_str, end=(date + datetime.timedelta(days=1)).strftime('%Y-%m-%d'))
+        
+        # Check if data is available
+        if not data.empty:
+            # Return the closing price
+            return data['Close'].iloc[0]
+        else:
+            return None
+    except Exception as e:
+        print(f"An error occurred: {e}")
+        return None
+    
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.
@@ -55,7 +88,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[get_current_time_in_timezone,final_answer], ## add your tools here (don't remove final answer)
+    tools=[get_sp500_price,get_current_time_in_timezone,final_answer], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,