sahilursa commited on
Commit
a148d79
·
verified ·
1 Parent(s): 04ff36f

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +6 -2
src/streamlit_app.py CHANGED
@@ -34,11 +34,15 @@ st.set_page_config(
34
  layout="wide",
35
  )
36
 
37
- # ========= Paths & knobs =========
38
- DATA_DIR = Path(os.getenv("DATA_DIR", "data"))
 
 
 
39
  VECTOR_PATH = DATA_DIR / "vector_store.index"
40
  PKL_PATH = DATA_DIR / "data.pkl"
41
 
 
42
  EMBED_MODEL_NAME = os.getenv("EMBED_MODEL_NAME", "BAAI/bge-large-en-v1.5")
43
  TOP_K = int(os.getenv("TOP_K", "5"))
44
  MAX_CONTEXT_CHARS = int(os.getenv("MAX_CONTEXT_CHARS", "12000"))
 
34
  layout="wide",
35
  )
36
 
37
+ from pathlib import Path
38
+
39
+ BASE_DIR = Path(__file__).resolve().parent # points to src/
40
+ DATA_DIR = BASE_DIR / "data"
41
+
42
  VECTOR_PATH = DATA_DIR / "vector_store.index"
43
  PKL_PATH = DATA_DIR / "data.pkl"
44
 
45
+
46
  EMBED_MODEL_NAME = os.getenv("EMBED_MODEL_NAME", "BAAI/bge-large-en-v1.5")
47
  TOP_K = int(os.getenv("TOP_K", "5"))
48
  MAX_CONTEXT_CHARS = int(os.getenv("MAX_CONTEXT_CHARS", "12000"))