Update src/streamlit_app.py

src/streamlit_app.py

@@ -27,13 +27,25 @@ import shap
 # Config & paths
 # -------------------------
 st.set_page_config(page_title="AI Feature Universe Explorer — Advanced + SHAP", layout="wide")
-DATA_DIR = os.getenv("DATA_DIR", "./data")
+
+# Use Hugging Face persistent path if available
+PERSISTENT_DIR = "/data" if os.path.exists("/data") else "./data"
+DATA_DIR = os.getenv("DATA_DIR", PERSISTENT_DIR)
+
 os.makedirs(DATA_DIR, exist_ok=True)
+
 CSV_PATH = os.path.join(DATA_DIR, "flatfile_universe_advanced.csv")
 META_PATH = os.path.join(DATA_DIR, "feature_metadata_advanced.json")
 PDF_PATH = os.path.join(DATA_DIR, "annotated_bibliography.pdf")
 ENSEMBLE_ARTIFACT = os.path.join(DATA_DIR, "ensemble_models.joblib")
 
+# Confirm storage mount
+if os.path.exists("/data"):
+    st.sidebar.success(f"✅ Using persistent storage: {DATA_DIR}")
+else:
+    st.sidebar.warning(f"⚠️ Using ephemeral storage: {DATA_DIR}. Data will be lost on rebuild.")
+
+
 # -------------------------
 # Utility: generate advanced dataset if missing
 # -------------------------