"""Инициализация ChromaDB: загрузка документов из папки documents/."""
from pathlib import Path
from vectorstore import create_vectorstore, load_documents

CHROMA_DIR = "./chroma_db"
DOCS_DIR   = "./documents"

if not Path(DOCS_DIR).exists():
    Path(DOCS_DIR).mkdir(parents=True)
    sample = Path(DOCS_DIR) / "intro.txt"
    sample.write_text("LangChain — фреймворк для создания AI-приложений с LLM.
"
                      "RAG (Retrieval Augmented Generation) позволяет давать LLM доступ к внешним документам.
"
                      "ChromaDB — векторная база данных с открытым исходным кодом.", encoding="utf-8")
    print(f"Создан тестовый файл: {sample}")

vs = create_vectorstore(CHROMA_DIR)
count = load_documents(DOCS_DIR, vs)
print(f"Загружено {count} чанков из {DOCS_DIR}/ в ChromaDB ({CHROMA_DIR})")