Update app.py

app.py

@@ -7,6 +7,49 @@ from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
 
+import yfinance as yf
+
+def get_nse_stock_price(stock_symbol: str) -> dict:
+    """
+    Fetch the latest NSE stock price, price change, and percentage change.
+    Use NSE ticker symbol.
+
+    Args:
+        stock_symbol (str): NSE Stock symbol of the company (e.g., 'RELIANCE', 'INFY').
+
+    Returns:
+        dict: Stock details including date, current price, price change, and percentage change.
+    """
+    try:
+        stock_symbol = stock_symbol.upper()  # Ensure uppercase
+        stock = yf.Ticker(stock_symbol + ".NS")
+        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} on NSE."}
+
+        latest_data = data.iloc[-1]
+        stock_date = latest_data.name.strftime('%Y-%m-%d')
+        current_price = latest_data['Close']
+
+        # Get previous close safely
+        if len(data) > 1:
+            previous_close = data['Close'].iloc[-2]  # Use last available close
+        else:
+            previous_close = latest_data['Open']  # Fallback
+
+        price_change = current_price - previous_close
+        percent_change = (price_change / previous_close) * 100
+
+        return {
+            "Stock": stock_symbol,
+            "Date": stock_date,
+            "Current Price": round(current_price, 2),
+            "Price Change": round(price_change, 2),
+            "% Change": round(percent_change, 2)
+        }
+
+
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
 def current_temprature_in_planet(planet: str) -> str: 
@@ -54,7 +97,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[current_temprature_in_planet, final_answer], ## add your tools here (don't remove final answer)
+    tools=[get_nse_stock_price, current_temprature_in_planet, final_answer], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,