app: auto initialization

app.py

@@ -16,15 +16,15 @@ def initialize_manual():
         message = f"Error initializing Stock-Analysis Agent:\n{e}\n"
     return message, stock_analyst
 
-def initialize_auto():
-    print(f'Initializing Stock-Analysis Agent')
+def find_matching_tickers(agent_instance, ticker):
+    if agent_instance is None:
+        return 'Stock-Analysis Agent is not initialized. Please initialize first.'
     try:
-        stock_analyst = StockAnalyst()
-        print(f"Stock-Analysis Agent initialized successfully.")
+        df = agent_instance.tickerfinder.find_best_matching_ticker_or_title(ticker)
     except Exception as e:
-        stock_analyst = None
-        print(f"Error initializing Stock-Analysis Agent:\n{e}")
-    return stock_analyst
+        print(f'Error finding matching tickers for {ticker}: :\n{e}')
+        df = None
+    return df
 
 def ask_stock_agent(agent_instance, ticker):
     if agent_instance is None:
@@ -66,27 +66,31 @@ def plot_stock(agent_instance, ticker):
 with gr.Blocks() as demo:
     gr.Markdown("# Stock Analysis Agent")
     # State to store the agent
-    # auto-initialize (not working)
-    # stock_analyst = initialize_auto()
-    # agent_instance = gr.State(stock_analyst)
-    # manual-initialize with button
-    message, stock_analyst = initialize_manual() # is this needed?
-    agent_instance = gr.State(None)
-    init_status = gr.Textbox(label="Initialization Status", interactive=False)
-    init_button = gr.Button("Initialize")
-    # Event handlers
-    init_button.click(
-        initialize_manual,
-        outputs=[init_status, agent_instance]  # Output: status message and the agent
+    stock_analyst_state  = gr.State()
+    # Auto-initialize
+    init_status = gr.Textbox(label="Init Status", interactive=False, value="Initializing…")
+    # Initialize automatically when the page loads
+    demo.load(
+        fn=initialize_manual,
+        inputs=None,
+        outputs=[init_status, stock_analyst_state]
     )
+    # manual-initialize with button
+    # message, stock_analyst = initialize_manual() # is this needed?
+    # init_status = gr.Textbox(label="Initialization Status", interactive=False)
+    # init_button = gr.Button("Initialize")
+    # init_button.click(
+    #    fn=initialize_manual,
+    #    outputs=[init_status, stock_analyst_state]  # Output: status message and the agent
+    # )
     print('Agent initialized successfully')
     # TickerFinder tool
     ticker_to_find = gr.Textbox(label="Ticker Finder: not sure what's the stock ticker symbol? Enter the company name you want to analyze, and I will try to help.")
     tf_output = gr.Markdown(label="Match Results")
     search_button = gr.Button("Search")
     search_button.click(
-        fn=stock_analyst.tickerfinder.find_best_matching_ticker_or_title,
-        inputs=[ticker_to_find],
+        fn=find_matching_tickers,
+        inputs=[stock_analyst_state, ticker_to_find],
         outputs=[tf_output],
         api_name="ticker_finder"
     )
@@ -96,13 +100,13 @@ with gr.Blocks() as demo:
     md_output = gr.Markdown(label="Output Box")
     analyze_button.click(
             fn=ask_stock_agent,
-            inputs=[agent_instance, ticker],
+            inputs=[stock_analyst_state, ticker],
             outputs=md_output, 
             api_name="ask_stock_agent")
     plot_output = gr.Plot(label=ticker, format="png")
     analyze_button.click(
             fn=plot_stock,
-            inputs=[agent_instance, ticker],
+            inputs=[stock_analyst_state, ticker],
             outputs=[plot_output],
             api_name="plot_stock")