Update app.py

app.py

@@ -13,7 +13,7 @@ api_key = os.getenv("ALPHAVANTAGE")
 if not api_key:
     raise ValueError("\u26a0\ufe0f Key is missing! Add it as a Secret in Hugging Face Spaces.")
 
-@spaces.GPU
+@tool
 # Helper functions for DCF calculation using Alpha Vantage API
 def get_company_ticker(api_key: str, company_name: str) -> str:
     """
@@ -26,7 +26,7 @@ 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.
@@ -34,7 +34,7 @@ def get_company_overview(api_key: str, symbol: str) -> dict:
     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.
@@ -46,6 +46,7 @@ def calculate_dcf(free_cash_flow: float, wacc: float, growth_rate: float, years:
         dcf_value += discounted_cash_flow
     return dcf_value
 
+@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.
@@ -103,7 +104,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, 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 your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,