Update main.py

main.py

@@ -95,8 +95,28 @@ from sentence_transformers import SentenceTransformer
 
 
 # ─────────────────────────── Constants ───────────────────────────────────────
-DIM       = 768   # jina-embeddings-v2-base-code output dimension
-STORE_DIR = pathlib.Path("/data/indexes")
+DIM = 768   # jina-embeddings-v2-base-code output dimension
+
+def _resolve_store_dir() -> pathlib.Path:
+    """
+    Try /data/indexes (HF Spaces persistent volume).
+    Fall back to ~/.cache/code-search/indexes if /data is not writable
+    (local dev, or volume not yet mounted with correct permissions).
+    """
+    primary = pathlib.Path("/data/indexes")
+    try:
+        primary.mkdir(parents=True, exist_ok=True)
+        probe = primary / ".write_probe"
+        probe.touch()
+        probe.unlink()
+        return primary
+    except OSError:
+        fallback = pathlib.Path.home() / ".cache" / "code-search" / "indexes"
+        fallback.mkdir(parents=True, exist_ok=True)
+        print(f"Warning: /data/indexes not writable — using fallback: {fallback}")
+        return fallback
+
+STORE_DIR = _resolve_store_dir()
 
 LANGUAGE_MAP = {
     ".py":   "python",