Update agent.py

agent.py

@@ -430,21 +430,24 @@ embedding_model = HuggingFaceEmbeddings(
 # Create FAISS index and save it
 # -----------------------------
 try:
-    vector_store = FAISS.load_local(
-        "/home/wendy/my_hf_agent_course_projects/faiss_index",
-        embedding_model,
-        allow_dangerous_deserialization=True
-    )
+    # Make sure to pass the directory, not the file path
+    index_dir = "/home/wendy/my_hf_agent_course_projects/faiss_index"
+    vector_store = FAISS.load_local(index_dir, embedding_model, allow_dangerous_deserialization=True)
 except Exception as e:
     print(f"Error loading index: {str(e)}")
-    # Fallback to rebuilding index
+    # Fallback to rebuilding the index if loading fails
     vector_store = FAISS.from_documents(docs, embedding_model)
 
-
 # -----------------------------
 # Load FAISS index properly
 # -----------------------------
-loaded_store = FAISS.load_local("/home/wendy/my_hf_agent_course_projects/faiss_index/index.faiss", embedding_model, allow_dangerous_deserialization=True)
+# Ensure the directory path is correct and points to the index file directly
+index_path = "/home/wendy/my_hf_agent_course_projects/faiss_index/index.faiss"
+try:
+    loaded_store = FAISS.load_local(index_path, embedding_model, allow_dangerous_deserialization=True)
+    print("FAISS index loaded successfully")
+except Exception as e:
+    print(f"Error loading FAISS index: {str(e)}")
 
 
 # -----------------------------