Add init_db.py

init_db.py

@@ -0,0 +1,12 @@
+"""
+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.")