Update app.py

app.py

@@ -1,11 +1,12 @@
 from langchain_community.document_loaders import UnstructuredMarkdownLoader
 from langchain.text_splitter import RecursiveCharacterTextSplitter
-from langchain.schema import Document
+from langchain_core.documents import Document  # Updated import
 # from langchain_openai import OpenAIEmbeddings
-from langchain.embeddings import HuggingFaceEmbeddings
+from langchain_huggingface import HuggingFaceEmbeddings  # Updated import
 from langchain_community.vectorstores import Chroma
 from dotenv import load_dotenv
 import os
+import shutil  # Added import
 
 # Load environment variables
 load_dotenv()
@@ -29,7 +30,8 @@ def load_documents():
     if not os.path.exists(file_path):
         print(f"Error: File {file_path} not found.")
         return []
-    loader = UnstructuredMarkdownLoader(file_path)
+    loader
+ = UnstructuredMarkdownLoader(file_path)
     documents = loader.load()
     return documents
 
@@ -57,7 +59,9 @@ def save_to_chroma(chunks: list[Document]):
 
     # Create a new DB from the documents.
     db = Chroma.from_documents(
-        chunks, HuggingFaceEmbeddings(model_name="BAAI/bge-small-en-v1.5"), persist_directory=CHROMA_PATH
+        chunks, 
+        HuggingFaceEmbeddings(model_name="BAAI/bge-small-en-v1.5"), 
+        persist_directory=CHROMA_PATH
     )
     db.persist()
     print(f"Saved {len(chunks)} chunks to {CHROMA_PATH}.")