Update app.py

app.py

@@ -11,38 +11,41 @@ from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
 
-api_key = os.getenv("ALPHAVANTAGE")
-
 # Define a tool to retrieve the Discounted Cash Flow (DCF) of a company
 
 @tool
 def get_dcf_of_company(symbol: str) -> str:
-    """A tool that retrieves the DCF of a company based on its name.
+    """Retrieve the Discounted Cash Flow (DCF) of a company based on its stock symbol.
+    
     Args:
-        symbol: The symbol name of the company in the stock market, always in capitals.
-        e.g. AAPL for apple, IBM for IBM
+        symbol (str): The stock symbol of the company (e.g., 'AAPL' for Apple, 'IBM' for IBM).
+    
+    Returns:
+        str: A message with the DCF value or an error message if the calculation fails.
     """
     try:
-
+        api_key = 'NR9AFISYRYH2B5U3'
         # Retrieve the company's financial statements
         financials = {}
         for statement in ['INCOME_STATEMENT', 'BALANCE_SHEET', 'CASH_FLOW']:
-            url = f'https://www.alphavantage.co/query?function={statement}&symbol={symbol}&apikey=NR9AFISYRYH2B5U3'
+            url = f'https://www.alphavantage.co/query?function={statement}&symbol={symbol}&apikey={api_key}'
             response = requests.get(url)
-            financials[statement] = response.json()
+            data = response.json()
+            if 'annualReports' in data:
+                financials[statement] = data['annualReports'][0]  # Use the most recent annual report
+            else:
+                return f"Error: Unable to retrieve {statement} for symbol '{symbol}'."
 
         # Extract necessary financial data
-        # Note: This is a simplified example. In practice, you'd need to handle
-        # multiple periods, calculate averages, and make growth assumptions.
-        income_statement = financials['INCOME_STATEMENT']['annualReports'][0]
-        cash_flow_statement = financials['CASH_FLOW']['annualReports'][0]
-        balance_sheet = financials['BALANCE_SHEET']['annualReports'][0]
-
-        net_income = float(income_statement['netIncome'])
-        depreciation = float(cash_flow_statement['depreciationDepletionAndAmortization'])
-        capex = float(cash_flow_statement['capitalExpenditures'])
+        income_statement = financials['INCOME_STATEMENT']
+        cash_flow_statement = financials['CASH_FLOW']
+        balance_sheet = financials['BALANCE_SHEET']
+
+        net_income = float(income_statement.get('netIncome', 0))
+        depreciation = float(cash_flow_statement.get('depreciationDepletionAndAmortization', 0))  # Corrected key
+        capex = float(cash_flow_statement.get('capitalExpenditures', 0))
         change_in_working_capital = (
-            float(balance_sheet['totalCurrentAssets']) - float(balance_sheet['totalCurrentLiabilities'])
+            float(balance_sheet.get('totalCurrentAssets', 0)) - float(balance_sheet.get('totalCurrentLiabilities', 0))
         )
 
         # Calculate Free Cash Flow (FCF)