Update app.py

app.py

@@ -80,8 +80,8 @@ if "feedback_log" not in st.session_state:
 def load_environment():
     load_dotenv()
     # Ensure HF_TOKEN is available
-    if "HF_TOKEN" not in os.environ and "HUGGINGFACEHUB_API_TOKEN" in os.environ:
-        os.environ["HF_TOKEN"] = os.environ["HUGGINGFACEHUB_API_TOKEN"]
+    if "HUGGINGFACEHUB_API_TOKEN" not in os.environ and "HF_TOKEN" in os.environ:
+        os.environ["HUGGINGFACEHUB_API_TOKEN"] = os.environ["HF_TOKEN"] 
     genai.configure(api_key=os.environ["GOOGLE_API_KEY"])
 
 from keybert import KeyBERT
@@ -158,7 +158,13 @@ def set_global_vectorstore(vectorstore):
     global vectorstore_global
     vectorstore_global = vectorstore
 
-kw_model = KeyBERT()
+kw_model = None
+
+def get_kw_model():
+    global kw_model
+    if kw_model is None:
+        kw_model = KeyBERT()
+    return kw_model
 
 def self_reasoning(query, context):
     llm = GeminiLLM()
@@ -191,6 +197,7 @@ def faiss_search_with_keywords(query):
     global vectorstore_global
     if vectorstore_global is None:
         raise ValueError("FAISS vectorstore is not initialized.")
+    kw_model = get_kw_model()
     keywords = kw_model.extract_keywords(query, keyphrase_ngram_range=(1,2), stop_words='english', top_n=5)
     refined_query = " ".join([keyword[0] for keyword in keywords])
     retriever = vectorstore_global.as_retriever(search_kwargs={"k": 13})