import os
from dotenv import load_dotenv
from src.vectorstore import create_vectorstore, load_documents

def main():
    load_dotenv()
    vectorstore = create_vectorstore(persist_directory="./chroma_db")
    load_documents("./documents", vectorstore)
    print("Documents loaded into ChromaDB.")

if __name__ == "__main__":
    main()