Update main.py

main.py

@@ -1,3 +1,4 @@
+import os
 from typing import List
 from pathlib import Path
 from langchain_community.embeddings import HuggingFaceEmbeddings
@@ -10,7 +11,8 @@ from langchain_community.document_loaders import (
     PyMuPDFLoader,
 )
 from langchain.text_splitter import RecursiveCharacterTextSplitter
-from langchain.vectorstores.chroma import Chroma
+from langchain_community.vectorstores import Chroma
+
 from langchain.indexes import SQLRecordManager, index
 from langchain.schema import Document
 from langchain.schema.runnable import Runnable, RunnablePassthrough, RunnableConfig