Update app.py

app.py

@@ -3,13 +3,17 @@ import datetime
 import requests
 import pytz
 import yaml
-from tools.final_answer import FinalAnswerTool
-#import yfinance as yf
+import requests
 import logging
 from typing import Optional, Dict, Any
+from tools.final_answer import FinalAnswerTool  # Adjust if FinalAnswerTool is defined elsewhere
 
 from Gradio_UI import GradioUI
 
+logger = logging.getLogger(__name__)
+logger.setLevel(logging.DEBUG)
+
+
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
 def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return type
@@ -42,9 +46,6 @@ final_answer = FinalAnswerTool()
 
 @tool
 def get_current_stock_price(ticker_symbol: str) -> Optional[Dict[str, Any]]:
-    logger = logging.getLogger(__name__)
-    logger.setLevel(logging.DEBUG)
-    
     """
     Get the latest stock price for the given ticker symbol by querying Yahoo Finance directly,
     then calculate the price change over the last 5 days (percentage and absolute).
@@ -53,30 +54,23 @@ def get_current_stock_price(ticker_symbol: str) -> Optional[Dict[str, Any]]:
         ticker_symbol: The stock ticker symbol to fetch the price for.
 
     Returns:
-        A dictionary containing:
+        A dictionary with:
             - "price"                : current closing price (float)
             - "price_change"         : absolute change over 5 days (float)
             - "price_change_percent" : percent change over 5 days (float)
-
-        If there's an error or no data is available, returns a dict with an "error" key, e.g.
-            {"error": "No historical data available for TSLA"}
-        Or returns None for unexpected issues.
+        If there's an error, returns a dict with an "error" key.
     """
     try:
-        # Build the Yahoo Finance “chart” API URL for the last 5 calendar days.
         url = f"https://query1.finance.yahoo.com/v8/finance/chart/{ticker_symbol}"
         params = {
             "range": "5d",
             "interval": "1d",
-            # We only need daily closes, so the default interval=1d is fine.
         }
 
         resp = requests.get(url, params=params, timeout=10)
         resp.raise_for_status()
         data = resp.json()
 
-        # Drill into the JSON structure:
-        #   data["chart"]["result"][0]["indicators"]["quote"][0]["close"]
         chart = data.get("chart", {})
         if not chart or chart.get("error"):
             return {"error": f"Yahoo Finance returned an error for {ticker_symbol}."}
@@ -89,23 +83,16 @@ def get_current_stock_price(ticker_symbol: str) -> Optional[Dict[str, Any]]:
         if not quote_section:
             return {"error": f"No quote information in chart for {ticker_symbol}."}
 
-        closes = quote_section[0].get("close", [])
-        # The “close” array may contain nulls if the market was closed on a given day.
-        # Filter out any None values:
-        closes = [c for c in closes if (c is not None)]
+        closes = [c for c in quote_section[0].get("close", []) if c is not None]
 
         if len(closes) < 2:
-            return {"error": f"Not enough close‐price data (need at least 2 days) for {ticker_symbol}."}
+            return {"error": f"Not enough close‐price data for {ticker_symbol}."}
 
-        # The “closes” list is chronological; last element is the most recent closing price:
         first_price = closes[0]
         current_price = closes[-1]
-
-        # Calculate changes:
         price_change = current_price - first_price
         price_change_percent = (price_change / first_price) * 100 if first_price != 0 else 0.0
 
-        # Round everything to two decimals:
         return {
             "price": round(current_price, 2),
             "price_change": round(price_change, 2),
@@ -125,9 +112,12 @@ def get_current_stock_price(ticker_symbol: str) -> Optional[Dict[str, Any]]:
         logger.error(f"Unexpected error fetching price for {ticker_symbol}: {e}")
         return None
 
+
+# Required for SmolAgent to emit a final structured response
 final_answer = FinalAnswerTool()
 
 
+
 # If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
 # model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud'