Upload app.py

app.py

@@ -136,6 +136,7 @@ import os
 # import pinecone
 
 import pickle
+@st.cache_data
 def get_faiss_semantic_index():
     try:
         index_path = "./HuggingFaceEmbeddings.pkl"
@@ -147,6 +148,7 @@ def get_faiss_semantic_index():
         return faiss_index
     except Exception as e:
         st.error(f"Error loading embeddings: {e}")
+        return None
 
 # def promt_engineer(text):
 PROMPT_TEMPLATE = """
@@ -194,7 +196,12 @@ def chat_actions():
 
     # query_embedding = model.encode(st.session_state["chat_input"])
     query = st.session_state["chat_input"]
-    docs = faiss_index.similarity_search(query, k=2)
+    if faiss_index is not None:
+        docs = faiss_index.similarity_search(query, k=2)
+    else:
+        st.error("Failed to load embeddings.")
+    # docs = faiss_index.similarity_search(query, k=2)
+
     for doc in docs:
         print("\n")
         print(str(doc.metadata["page"]+1) + ":", doc.page_content)