Update app.py

app.py

@@ -2,6 +2,7 @@ from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
 import datetime
 import requests
 import pytz
+import yfinance as yf
 import yaml
 from tools.final_answer import FinalAnswerTool
 
@@ -18,6 +19,26 @@ def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return
     """
     return "What magic will you build ?"
 
+# Below is the rool to get the current price of the Stock
+@tool
+def get_stock_price(stock_code: str) -> str:
+    """A tool that Fetches the current stock price for a given stock code.
+
+    Args:
+        stock_code: The stock ticker symbol (e.g., 'NVDA').
+    """
+    try:
+        ticker = yf.Ticker(stock_code)
+        todays_data = ticker.history(period='1d')
+        if not todays_data.empty:
+            current_price = todays_data['Close'][-1]
+            return current_price
+        else:
+            return None
+    except Exception as e:
+        print(f"Error fetching stock price for {stock_code}: {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.