vishalkatheriya commited on
Commit
477224c
Β·
verified Β·
1 Parent(s): 24773d4

Upload 2 files

Browse files
Files changed (2) hide show
  1. src/config.py +9 -5
  2. src/streamlit_app.py +2 -2
src/config.py CHANGED
@@ -17,12 +17,16 @@ SUPABASE_KEY = os.getenv("SUPABASE_KEY", "")
17
 
18
  # ── Paths ─────────────────────────────────────────────────────────────
19
  BASE_DIR = Path(__file__).resolve().parent
20
- UPLOADS_DIR = BASE_DIR / "uploads"
21
- UPLOADS_DIR.mkdir(exist_ok=True)
22
 
23
- # LightRAG working directory
24
- WORKING_DIR = BASE_DIR / "lightrag_working"
25
- WORKING_DIR.mkdir(exist_ok=True)
 
 
 
 
 
 
26
 
27
  # ── Chunking (used by pdf_processor) ─────────────────────────────────
28
  CHUNK_SIZE = 1000
 
17
 
18
  # ── Paths ─────────────────────────────────────────────────────────────
19
  BASE_DIR = Path(__file__).resolve().parent
 
 
20
 
21
+ # HuggingFace Spaces has a read-only filesystem β€” use /tmp for writable dirs
22
+ IS_HF_SPACE = bool(os.getenv("SPACE_ID"))
23
+ _DATA_ROOT = Path("/tmp/handbook_data") if IS_HF_SPACE else BASE_DIR
24
+
25
+ UPLOADS_DIR = _DATA_ROOT / "uploads"
26
+ UPLOADS_DIR.mkdir(parents=True, exist_ok=True)
27
+
28
+ WORKING_DIR = _DATA_ROOT / "lightrag_working"
29
+ WORKING_DIR.mkdir(parents=True, exist_ok=True)
30
 
31
  # ── Chunking (used by pdf_processor) ─────────────────────────────────
32
  CHUNK_SIZE = 1000
src/streamlit_app.py CHANGED
@@ -11,12 +11,12 @@ from google.adk.sessions import InMemorySessionService
11
  from google.adk.runners import Runner
12
  from google.genai import types
13
 
14
- from config import OPENAI_API_KEY, UPLOADS_DIR, BASE_DIR
15
  from rag import index_pdf, reset_index
16
  from handbook_generator import build_handbook
17
  from agent import root_agent
18
 
19
- HANDBOOK_EXPORT_PATH = BASE_DIR / "handbook_export.md"
20
 
21
  # ────────────────────────────────────────────────
22
  APP_NAME = "handbook_app"
 
11
  from google.adk.runners import Runner
12
  from google.genai import types
13
 
14
+ from config import OPENAI_API_KEY, UPLOADS_DIR, _DATA_ROOT
15
  from rag import index_pdf, reset_index
16
  from handbook_generator import build_handbook
17
  from agent import root_agent
18
 
19
+ HANDBOOK_EXPORT_PATH = _DATA_ROOT / "handbook_export.md"
20
 
21
  # ────────────────────────────────────────────────
22
  APP_NAME = "handbook_app"