Update app.py

app.py

@@ -20,12 +20,21 @@ import faiss
 from sentence_transformers import SentenceTransformer
 
 # ── Config ─────────────────────────────────────────────────────────────────────
-ROOT = Path(__file__).resolve().parent.parent
-MODEL_DIR = ROOT / "models" / "charsim_final"  # trained model
-PROC_DIR = ROOT / "data" / "processed"
-INDEX_DIR = Path(__file__).resolve().parent
+import os
+ON_SPACES = os.getenv("SPACE_ID") is not None
+
+if ON_SPACES:
+    MODEL_DIR  = "Phalakk/charsim-model"
+    PROC_DIR   = Path(__file__).resolve().parent
+    INDEX_DIR  = Path(__file__).resolve().parent
+else:
+    _ROOT      = Path(__file__).resolve().parent.parent
+    MODEL_DIR  = _ROOT / "models" / "charsim_final"
+    PROC_DIR   = _ROOT / "data"   / "processed"
+    INDEX_DIR  = Path(__file__).resolve().parent
+
 INDEX_PATH = INDEX_DIR / "char_index.faiss"
-META_PATH = INDEX_DIR / "char_meta.jsonl"
+META_PATH  = INDEX_DIR / "char_meta.jsonl"
 
 
 # ══════════════════════════════════════════════════════════════════════════════
@@ -47,7 +56,7 @@ print("=" * 50)
 # Load model
 print(f"\n[1/3] Loading model from {MODEL_DIR}...")
 
-model = SentenceTransformer("Phalakk/charsim-model")
+model = SentenceTransformer(MODEL_DIR if ON_SPACES else str(MODEL_DIR))
 print("      Model loaded.")
 
 # Load characters — filter out [Story] fallback entries