added ticker finder in the app, match results in html

app.py

@@ -24,7 +24,7 @@ def ask_stock_agent(agent_instance, ticker):
         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.
+        If you are not sure about the ticker symbol use the Ticker Finder to find the correct ticker.
         """
     return response
 
@@ -50,17 +50,26 @@ with gr.Blocks() as demo:
     # State to store the RAG instance
     message, stock_analyst = initialize()
     agent_instance = gr.State(None)
-    init_button = gr.Button("Initialize")
     init_status = gr.Textbox(label="Initialization Status", interactive=False)
+    init_button = gr.Button("Initialize")
     # Event handlers
     init_button.click(
         initialize,
         outputs=[init_status, agent_instance]  # Output: status message and the RAG instance
     )
     # gr.Text(label='Status', value=message)
-    gr.Markdown("Enter a stock symbol ('ticker') to be analyzed (GOOG, MSFT, etc.)")
-    ticker = gr.Textbox(label="ticker")
-    # plot_button = gr.Button('Plot')
+    # 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],
+        outputs=[tf_output],
+        api_name="ticker_finder"
+    )
+    # Enter stock symbol
+    ticker = gr.Textbox(label="Enter a stock ticker symbol to be analyzed (GOOG, MSFT, etc.)")
     analyze_button = gr.Button('Analyze Stock')
     plot_output = gr.Plot(label=ticker, format="png")
     analyze_button.click(

src/ticker_finder.py

@@ -98,7 +98,7 @@ class TickerFinder():
     def find_best_matching_title(
             self,
             input_name:str,
-            top_n=3) -> pd.DataFrame:
+            top_n=5) -> pd.DataFrame:
         """
         Find the best matching company title for a given company name.
         Args:
@@ -124,7 +124,7 @@ class TickerFinder():
     def find_best_matching_ticker(
             self,
             ticker:str,
-            top_n:int=3
+            top_n:int=5
             ) -> pd.DataFrame:
         """
         Find the best matching company ticker for a given ticker.
@@ -173,12 +173,14 @@ class TickerFinder():
         # sort by score:
         c_matches_sorted = c_matches_dedup.sort_values(by='Score', ascending=False)
         # convert results into a pretty string:
-        results = self.df_to_pretty_string(c_matches_sorted )
+        results = self.df_to_pretty_string(c_matches_sorted)
         return(results)
 
     def df_to_pretty_string(
             self,
-            df):
+            df:pd.DataFrame,
+            num_rows:int=5
+            ) -> str:
         """
         Convert a pd.DataFrame into a pretty string, using the tabulate package.
         Args:
@@ -188,13 +190,16 @@ class TickerFinder():
         pretty_string : str
             A string containing the pretty-formatted dataframe
         """
-        pretty_string = tabulate(df[['Title', 'Ticker']], 
-              # headers='keys', 
-              tablefmt='plain', 
+        df = df.rename(columns={'Title': 'Company Name', 'Ticker': 'Ticker Symbol'})
+        df_subset = df[['Company Name', 'Ticker Symbol']].iloc[0:num_rows]  
+        pretty_table= tabulate(df_subset, 
+              headers='keys', 
+              # tablefmt='plain', 
+              tablefmt='html',
               showindex=False,
               numalign='left',
               stralign='left')
-        return pretty_string
+        return pretty_table
     
     def does_ticker_exist(
             self,
@@ -213,7 +218,7 @@ class TickerFinder():
 
 
 # if __name__ == "__main__":
-      # results = TickerFinder().find_best_matching_ticker_or_title("microsoft")
-      # print(results)
+#       results = TickerFinder().find_best_matching_ticker_or_title("microsoft")
+#       print(results)
       # exists = TickerFinder().does_ticker_exist('bbbbb')
       # print(f'Ticker exists') if exists else print(f'Ticker does not exist')