Update app.py

app.py

@@ -10,12 +10,11 @@ from Gradio_UI import GradioUI
 import yfinance as yf
 
 @tool
-def get_nse_stock_price(stock_name: str) -> dict:
-    """
-    Fetch the latest NSE stock price, price change, and percentage change.
+def get_stock_price(stock_symbol: str) -> dict:
+    """Fetch the latest stock price, date, current Price, price change, and percentage change.
 
     Args:
-        stock_name (str): Name of the stock
+        stock_symbol (str): Symbol of the stock
 
     Returns:
         dict: A dictionary containing:
@@ -25,12 +24,12 @@ def get_nse_stock_price(stock_name: str) -> dict:
             - "Price Change" (float): The price difference from the previous close.
             - "% Change" (float): The percentage change from the previous close.
     """
-    stock_name = stock_name.upper()  # Ensure uppercase
-    stock = yf.Ticker(stock_name + ".NS")
+    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."}
+        return {"error": f"No data found for {stock_symbol}."}
 
     latest_data = data.iloc[-1]
     stock_date = latest_data.name.strftime('%Y-%m-%d')
@@ -101,7 +100,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[get_nse_stock_price, current_temprature_in_planet, final_answer], ## add your tools here (don't remove final answer)
+    tools=[get_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,