"""
Script to initialise the ChromaDB with documents from ./documents.
"""
import os
from vectorstore import create_vectorstore, load_documents

if __name__ == "__main__":
    db_dir = os.getenv("CHROMA_DB_DIR", "./chroma_db")
    store = create_vectorstore(db_dir)
    docs_path = os.getenv("DOCS_PATH", "./documents")
    load_documents(docs_path, store)
    print("Initialisation complete.")