feat: add kw extraction to session state

sorbobotapp/app.py

@@ -3,12 +3,12 @@ import os
 
 import streamlit as st
 import streamlit.components.v1 as components
-from langchain.callbacks import get_openai_callback
-
 from chain import get_chain
 from chat_history import insert_chat_history, insert_chat_history_articles
 from connection import connect
 from css import load_css
+from keyword_extraction import KeywordExtractor
+from langchain.callbacks import get_openai_callback
 from message import Message
 
 st.set_page_config(layout="wide")
@@ -27,6 +27,8 @@ def initialize_session_state():
         st.session_state.token_count = 0
     if "conversation" not in st.session_state:
         st.session_state.conversation = get_chain(conn)
+    if "keyword_extractor" not in st.session_state:
+        st.session_state.keyword_extractor = KeywordExtractor()
 
 
 def send_message_callback():