streamlit changes

streamlit_app.py

@@ -1,4 +1,28 @@
 import os
+
+# Ensure Streamlit and ML caches write to a writable location (e.g., on HF Spaces)
+os.environ.setdefault("HOME", "/tmp")
+os.environ.setdefault("STREAMLIT_BROWSER_GATHER_USAGE_STATS", "false")
+os.environ.setdefault("XDG_CACHE_HOME", "/tmp/.cache")
+os.environ.setdefault("HF_HOME", "/tmp/hf")
+os.environ.setdefault("TRANSFORMERS_CACHE", "/tmp/hf/transformers")
+os.environ.setdefault("SENTENCE_TRANSFORMERS_HOME", "/tmp/hf/sentence-transformers")
+os.environ.setdefault("TORCH_HOME", "/tmp/torch")
+
+# Create the cache directories to avoid PermissionError on first write
+for _d in [
+    os.environ["XDG_CACHE_HOME"],
+    os.environ["HF_HOME"],
+    os.environ["TRANSFORMERS_CACHE"],
+    os.environ["SENTENCE_TRANSFORMERS_HOME"],
+    os.environ["TORCH_HOME"],
+    os.path.join(os.environ.get("HOME", "/tmp"), ".streamlit"),
+]:
+    try:
+        os.makedirs(_d, exist_ok=True)
+    except Exception:
+        pass
+
 import streamlit as st
 from dotenv import load_dotenv, find_dotenv
 from langchain.prompts import PromptTemplate