Enable price and ratios

src/app.py

@@ -46,9 +46,11 @@ def main():
     
     if st.button("Generate"):
         if input_text:
-            with st.spinner('Generating response...'):
+            with st.status('Generating response...') as status:
+                config.status = status
                 agent_resp = get_response(input_text)
                 st.write(get_output(agent_resp, input_text))
+                config.status.update(label="Finished!", state="complete", expanded=True)
         else:
             st.warning("Please enter some text to generate a response.")
 

src/config.py

@@ -1,11 +1,14 @@
 from pydantic import Field
 from pydantic_settings import BaseSettings
+from typing import Dict, Any
 
 class Config(BaseSettings):
     hf_token: str = Field(...)
     hf_model: str = Field("InvestmentResearchAI/LLM-ADE-dev")
     headless: bool = Field(False, description="Run in headless mode.")
 
+    status: Any = None # Hold the status
+
     az_search_endpoint: str = Field("https://analysis-bank.search.windows.net")
     az_search_api_key: str = Field(...)
     az_search_idx_name: str = Field("analysis-index")

src/functioncall.py

@@ -72,6 +72,7 @@ class ModelInference:
             raise ValueError("Assistant message is None")
         
     def execute_function_call(self, tool_call):
+        config.status.update(label=":mag: Gathering information..")
         function_name = tool_call.get("name")
         function_to_call = getattr(functions, function_name, None)
         function_args = tool_call.get("arguments", {})
@@ -82,6 +83,7 @@ class ModelInference:
         return results_dict
     
     def run_inference(self, prompt: List[Dict[str, str]]):
+        config.status.update(label=":brain: Thinking..")
         inputs = self.tokenizer.apply_chat_template(
             prompt,
             add_generation_prompt=True,

src/functions.py

@@ -102,6 +102,7 @@ def get_current_stock_price(symbol: str) -> float:
     float: The current stock price, or None if an error occurs.
   """
   try:
+    config.status.update(label=":chart_with_upwards_trend: Getting price")
     stock = yf.Ticker(symbol)
     # Use "regularMarketPrice" for regular market hours, or "currentPrice" for pre/post market
     current_price = stock.info.get("regularMarketPrice", stock.info.get("currentPrice"))
@@ -187,6 +188,7 @@ def get_key_financial_ratios(symbol: str) -> dict:
     dict: Dictionary containing key financial ratios.
     """
     try:
+        config.status.update(label=":chart_with_upwards_trend: Gathering financial data")
         stock = yf.Ticker(symbol)
         key_ratios = stock.info
         return key_ratios
@@ -290,12 +292,12 @@ def get_openai_tools() -> List[dict]:
     functions = [
         get_analysis,  
         # google_search_and_scrape,
-        # get_current_stock_price,
+        get_current_stock_price,
         # get_company_news,
         # get_company_profile,
         # get_stock_fundamentals,
         # get_financial_statements,
-        # get_key_financial_ratios,
+        get_key_financial_ratios,
         # get_analyst_recommendations,
         # get_dividend_data,
         # get_technical_indicators