app: check whether ticker exists before calls

app.py

@@ -4,43 +4,57 @@ from src.stock_analysis_agent import StockAnalyst
 from src.technical_analysis import TechnicalAnalysis
 
 
-def initialize_agent():
+def initialize():
     print(f'Initializing Stock-Analysis Agent')
-    # Initialize RAG instance
     try:
         stock_analyst = StockAnalyst()
-        message = "Stock-Analysis Agent initialized successfully."
+        message = "Stock-Analysis Agent initialized successfully.\n"
     except Exception as e:
         stock_analyst = None
-        message = f"Error initializing Stock-Analysis Agent:\n{e}"
+        message = f"Error initializing Stock-Analysis Agent:\n{e}\n"
     return message, stock_analyst
 
 def ask_stock_agent(agent_instance, ticker):
     if agent_instance is None:
         return 'Stock-Analysis Agent is not initialized. Please initialize first.'
-    response = agent_instance.get_formatted_stock_summary(ticker)
+    ticker_exists = agent_instance.tickerfinder.does_ticker_exist(ticker)
+    if ticker_exists:
+        response = agent_instance.get_formatted_stock_summary(ticker)
+    else:
+        response = f"""
+        Ticker symbol {ticker} is not listed by the U.S. Securities and Exchange Commission database,
+        see: https://www.sec.gov/files/company_tickers.json.
+        If you are not sure about the ticker symbol use the Ticker-Finder tool to find the correct ticker.
+        """
     return response
 
-def plot_stock(ticker):
-    # run the technical analysis
-    _, fig = TechnicalAnalysis(
-        ticker=ticker,
-        fetchperiodinweeks=12,
-        plot_ta=True,
-        savefig=False,
-        debug=False).run()
+def plot_stock(agent_instance, ticker):
+    if agent_instance is None:
+        return 'Stock-Analysis Agent is not initialized. Please initialize first.'
+    ticker_exists = agent_instance.tickerfinder.does_ticker_exist(ticker)
+    if ticker_exists:
+        # run the technical analysis
+        _, fig = TechnicalAnalysis(
+            ticker=ticker,
+            fetchperiodinweeks=12,
+            plot_ta=True,
+            savefig=False,
+            debug=False).run()
+    else:
+        message = f"""Ticker {ticker} does not exist"""
+        fig = TechnicalAnalysis(ticker).get_fetcherror_fig(message)
     return fig
 
 with gr.Blocks() as demo:
     gr.Markdown("# Stock Analysis Agent")
     # State to store the RAG instance
-    message, stock_analyst = initialize_agent()
-    agent_instance = gr.State(None) 
+    message, stock_analyst = initialize()
+    agent_instance = gr.State(None)
     init_button = gr.Button("Initialize")
     init_status = gr.Textbox(label="Initialization Status", interactive=False)
     # Event handlers
     init_button.click(
-        initialize_agent,
+        initialize,
         outputs=[init_status, agent_instance]  # Output: status message and the RAG instance
     )
     # gr.Text(label='Status', value=message)
@@ -51,7 +65,7 @@ with gr.Blocks() as demo:
     plot_output = gr.Plot(label=ticker, format="png")
     analyze_button.click(
             fn=plot_stock,
-            inputs=[ticker],
+            inputs=[agent_instance, ticker],
             outputs=[plot_output],
             api_name="plot_stock")
     md_output = gr.Markdown(label="Output Box")