add "get_todays_stock" tool

app.py

@@ -4,6 +4,7 @@ import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
+import yfinance as yf
 
 from Gradio_UI import GradioUI
 
@@ -17,21 +18,13 @@ def calculator(a: int, b: int, oper: str)-> str: #it's import to specify the ret
         b: the second argument
         oper: operetion to be performed. Either "add", "sub", "mult" or "div".
     """
-    
     if oper == "add":
         result = a + b
     elif oper == "sub":
         result = a - b
     elif oper == "mult":
         result = a * b
-    elif oper == "div":
-        
-        #try:
-        #    result = a / b
-        #except ZeroDivisionError:
-        #    result = "inf"  
-        
-        
+    elif oper == "div":    
         if b == 0:
             return "Error: Division by zero."
         result = a / b
@@ -40,6 +33,35 @@ def calculator(a: int, b: int, oper: str)-> str: #it's import to specify the ret
 
     return str(result)
 
+@tool
+def get_todays_stock(ticker: str) -> tuple:
+    """A tool that that fetches today's stock data for a selected company. 
+    Args:
+        ticker: The stock's ticker symbol, which is a unique abbreviation assigned to publicly traded companies.
+    """
+    try:
+        stock = yf.Ticker(ticker)
+        data = stock.history(period="1d")
+
+        if data.empty:
+            raise ValueError(f"No data available for {ticker}. Check if the ticker is correct.")
+
+        latest = data.iloc[-1]
+        return {
+            "Date": latest.name.date(),
+            "Open": latest["Open"],
+            "High": latest["High"],
+            "Low": latest["Low"],
+            "Close": latest["Close"],
+            "Volume": latest["Volume"]
+        }
+    
+    except ValueError as ve:
+        return {"Error": str(ve)}
+    
+    except Exception as e:
+        return {"Error": f"An unexpected error occurred: {str(e)}"}
+
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -74,7 +96,12 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, DuckDuckGoSearchTool(), calculator, get_current_time_in_timezone], ## add your tools here (don't remove final answer)
+    tools=[final_answer, 
+           DuckDuckGoSearchTool(), 
+           calculator, 
+           get_current_time_in_timezone,
+           image_generation_tool,
+          ], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,