Update app.py

app.py

@@ -1,5 +1,8 @@
+#!pip install -q yfinance
+
+
+import yfinance as yf
 from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
-import os
 import datetime
 import requests
 import pytz
@@ -10,35 +13,33 @@ from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
-def get_coin_price(symbol: str, currency: str = "usd")-> str: #it's import to specify the return type
+def my_custom_tool(arg1:str, arg2:str)-> str: #it's import to specify the return type
     #Keep this format for the description / args / args description but feel free to modify the tool
-    """A tool that fetches the latest price on a cryptocurrency symbol from Coingecko API.
+    """A tool fetches stock information of the provided symbol from Yahoo Finance API
     Args:
-        symbol: A string representing a valid symbol as CoinGecko's ID, not ticker (e.g., 'bitcoin' not BTC).
-        currency: A string representing the currency the value is returned in (e.g., 'usd').
+        arg1: company ticker symbol
+        arg2: information requested from the company data dict
     """
-    # set to lowercase for api compatibility and consistency
-    symbol = symbol.lower()
-    currency = currency.lower()
+
+    symbol: str = arg1.upper()
+    information: str = arg2.lower()
+
+    try:
+
+        tick = yf.Ticker(symbol)
+        company_data = tick.info
     
-    #define the endpoint with return headers - https://docs.coingecko.com/v3.0.1/reference/simple-price
-    url = "https://api.coingecko.com/api/v3/simple/price?ids=" + symbol + "&vs_currencies=" + currency
-    headers = {"accept": "application/json"}
+        if information in company_data:
+            return f"{symbol} : {information} = {company_data[information]}"
     
-    try:
-        #make request to endpoint
-        response = requests.get(url, headers=headers)
-        
-        #process response
-        if response.status_code == 200:
-            data = response.json()
-            usd_value = data[symbol][currency]
-            return f"The price of " + symbol + " in " + currency + " is " + f"${usd_value:,.2f}" #format as currency
         else:
-            # returrn verbose error, so the model can observe and ReAct
-            return f"Error: {response.status_code}, {response.text}"
+            available_keys = ", ".join(company_data.keys())
+            return f"Request information '{information}' is not available for {symbol}. Available informtion {available_keys} "
+
     except Exception as e:
-        return f"Error fetching price for '{symbol}' valued in currency '{currency}': {str(e)}"
+        return f"Error fetching data for {symbol}: {str(e)}"
+    
+    # return "What magic will you build ?"
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -68,12 +69,6 @@ model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may
 custom_role_conversions=None,
 )
 
-# model = HfApiModel(
-#     model_id="deepseek-ai/DeepSeek-R1",
-#     provider="together",
-#     api_key=os.getenv("together_key")  # Fetch API key from environment
-# )
-
 
 # Import tool from Hub
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
@@ -83,7 +78,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, get_coin_price, get_current_time_in_timezone], ## add your tools here (don't remove final answer)
+    tools=[final_answer, get_current_time_in_timezone, image_generation_tool,my_custom_tool], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,