Update src/streamlit_app.py

src/streamlit_app.py

@@ -6,6 +6,7 @@ from pathlib import Path
 from typing import List, Annotated, Any
 import operator
 import pandas as pd
+from dotenv import load_dotenv
 from tqdm import tqdm
 from pydantic import BaseModel
 from langchain.embeddings.cohere import CohereEmbeddings
@@ -19,13 +20,19 @@ from langgraph.graph import StateGraph, START, END, add_messages
 from langgraph.constants import Send
 from langgraph.checkpoint.memory import MemorySaver
 
-
 COHERE_API_KEY = os.environ["COHERE_API_KEY"]
 co = cohere.Client(COHERE_API_KEY)
-persist_dir = "./chroma_store"
+
+base_dir = Path(__file__).resolve().parent.parent
+documents_path = base_dir / "documents"
+
+if not documents_path.exists():
+    raise FileNotFoundError(f"Documents folder not found at {documents_path}")
+
+persist_dir = str(base_dir / "chroma_store")
 
 def prepare_vectorstore():
-    loader = DirectoryLoader("./documents", glob="**/*.txt", loader_cls=TextLoader)
+    loader = DirectoryLoader(str(documents_path), glob="**/*.txt", loader_cls=TextLoader)
     documents = loader.load()
     splitter = RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=50)
     docs = splitter.split_documents(documents)