Update embed.py

embed.py

@@ -3,8 +3,13 @@ from docx import Document
 from sentence_transformers import SentenceTransformer
 import chromadb
 
+
+CHROMA_PATH = os.getenv("CHROMA_PATH", "./chroma_db")
+os.makedirs(CHROMA_PATH, exist_ok=True)
+print("Chroma path:", CHROMA_PATH)
+
 # Initialize ChromaDB client (persistent storage)
-client = chromadb.PersistentClient(path="./chroma_db")
+client = chromadb.PersistentClient(path=CHROMA_PATH)
 collection = client.get_or_create_collection(name="knowledge_base")
 
 # Load MiniLM model for embeddings