Update app.py

app.py

@@ -48,6 +48,37 @@ def fact_checker(claim:str)-> str: #it's import to specify the return type
         return f"Error during fact-checking: {str(e)}"
         
 
+@tool
+def stock_price(symbol: str) -> str:
+    """A tool that retrieves the current stock price for a given stock symbol.
+    Args:
+        symbol: The ticker symbol of the stock (e.g., AAPL for Apple Inc.).
+    Returns:
+        A summary indicating the current price of the stock or an error message if unavailable.
+    """
+    try:
+        import yfinance as yf
+
+        # Initialize the ticker for the given symbol
+        ticker = yf.Ticker(symbol)
+        
+        # Try to get the current price from the ticker info
+        price = ticker.info.get("currentPrice")
+        
+        # Fallback: if 'currentPrice' is not available, get the last closing price
+        if price is None:
+            history = ticker.history(period="1d")
+            if not history.empty:
+                price = history["Close"].iloc[-1]
+            else:
+                return f"Could not retrieve the current price for {symbol}. Please verify the ticker symbol."
+
+        # Return the current price in a formatted string
+        return f"The current price for {symbol} is ${price:.2f}."
+    
+    except Exception as e:
+        return f"An error occurred while retrieving the stock price for {symbol}: {e}"
+
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.
@@ -85,7 +116,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[fact_checker, get_current_time_in_timezone, image_generation_tool, final_answer], ## add your tools here (don't remove final answer)
+    tools=[fact_checker, stock_price, get_current_time_in_timezone, image_generation_tool, final_answer], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,