Update app.py

app.py

@@ -1,4 +1,4 @@
-from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
+from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
 import datetime
 import requests
 import pytz
@@ -18,6 +18,13 @@ if not api_key:
 def get_company_ticker(api_key: str, company_name: str) -> str:
     """
     Retrieves the ticker symbol for a given company name using Alpha Vantage's SYMBOL_SEARCH.
+    
+    Args:
+        api_key (str): The API key required for authentication with the Alpha Vantage API.
+        company_name (str): The name of the company to search for.
+    
+    Returns:
+        str: The ticker symbol of the company if found, otherwise None.
     """
     search_url = f'https://www.alphavantage.co/query?function=SYMBOL_SEARCH&keywords={company_name}&apikey={api_key}'
     response = requests.get(search_url)
@@ -26,18 +33,36 @@ def get_company_ticker(api_key: str, company_name: str) -> str:
         return data['bestMatches'][0]['1. symbol']
     else:
         return None
+
 @tool
 def get_company_overview(api_key: str, symbol: str) -> dict:
     """
     Retrieves the company overview (financial fundamentals) for a given ticker using Alpha Vantage.
+
+    Args:
+        api_key (str): The API key required for authentication with the Alpha Vantage API.
+        symbol (str): The stock ticker symbol of the company.
+
+    Returns:
+        dict: A dictionary containing financial overview data for the company.
     """
     overview_url = f'https://www.alphavantage.co/query?function=OVERVIEW&symbol={symbol}&apikey={api_key}'
     response = requests.get(overview_url)
     return response.json()
+
 @tool
 def calculate_dcf(free_cash_flow: float, wacc: float, growth_rate: float, years: int = 5) -> float:
     """
     Calculates the Discounted Cash Flow (DCF) given the free cash flow, WACC and growth rate.
+
+    Args:
+        free_cash_flow (float): The company's free cash flow.
+        wacc (float): The Weighted Average Cost of Capital (WACC).
+        growth_rate (float): The estimated growth rate of free cash flow.
+        years (int, optional): The number of years to project cash flows. Defaults to 5.
+
+    Returns:
+        float: The total DCF value.
     """
     dcf_value = 0
     for year in range(1, years + 1):
@@ -49,13 +74,17 @@ def calculate_dcf(free_cash_flow: float, wacc: float, growth_rate: float, years:
 @spaces.GPU
 @tool
 def get_company_dcf(company_name: str) -> str:
-    """A tool that calculates the Discounted Cash Flow (DCF) of a company's stock.
+    """
+    A tool that calculates the Discounted Cash Flow (DCF) of a company's stock.
     
     Args:
-        company_name: A string representing the company's name (e.g., 'Apple Inc').
+        company_name (str): The name of the company (e.g., 'Apple Inc').
+
+    Returns:
+        str: The calculated DCF value for the given company.
     """
     try:
-        api_key = api_key
+        global api_key  # Ensure we are using the correct API key variable
         # Step 1: Retrieve the ticker symbol for the company
         symbol = get_company_ticker(api_key, company_name)
         if not symbol:
@@ -67,7 +96,6 @@ def get_company_dcf(company_name: str) -> str:
             return f"Could not retrieve financial data for ticker '{symbol}'."
         
         # Extract required financial metrics.
-        # Σημείωση: Τα ονόματα πεδίων μπορεί να διαφέρουν στο Alpha Vantage Overview.
         free_cash_flow = float(overview.get('FreeCashFlow', 0))
         wacc = float(overview.get('WeightedAverageCostOfCapital', 0)) / 100  # converting percentage to decimal
         growth_rate = float(overview.get('GrowthRate', 0)) / 100             # converting percentage to decimal
@@ -85,17 +113,13 @@ def get_company_dcf(company_name: str) -> str:
 
 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' 
-
 model = HfApiModel(
-max_tokens=2096,
-temperature=0.5,
-model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
-custom_role_conversions=None,
+    max_tokens=2096,
+    temperature=0.5,
+    model_id='Qwen/Qwen2.5-Coder-32B-Instruct',  # it is possible that this model may be overloaded
+    custom_role_conversions=None,
 )
 
-
 # Import tool from Hub
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
@@ -104,7 +128,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer,get_company_ticker, get_company_overview, calculate_dcf, get_company_dcf], ## add your tools here (don't remove final answer)
+    tools=[final_answer, get_company_ticker, get_company_overview, calculate_dcf, get_company_dcf],  # add tools here
     max_steps=6,
     verbosity_level=1,
     grammar=None,
@@ -114,5 +138,4 @@ agent = CodeAgent(
     prompt_templates=prompt_templates
 )
 
-
 GradioUI(agent).launch()