Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,18 +1,18 @@
 import os
+import streamlit as st
 
-# ✅ Fix for Streamlit writing to /.streamlit by setting HOME
-os.environ["HOME"] = "/tmp"
-
-# Optional: Redirect Hugging Face + other cache directories
+# Fix permission errors by forcing writable directories
 os.environ["HF_HOME"] = "/tmp/hf_home"
+# os.environ["TRANSFORMERS_CACHE"] = "/tmp/hf_cache"
 os.environ["SENTENCE_TRANSFORMERS_HOME"] = "/tmp/sbert"
 os.environ["XDG_CONFIG_HOME"] = "/tmp/.config"
+os.environ["STREAMLIT_HOME"] = "/tmp/.streamlit"
 
-# Optional: Create those folders if needed (not mandatory in Spaces, but safe)
-# for path in ["/tmp/hf_home", "/tmp/sbert", "/tmp/.config", "/tmp/.config/streamlit"]:
+# Make those folders if not already present
+# for path in ["/tmp/hf_home", "/tmp/hf_cache", "/tmp/sbert", "/tmp/.config", "/tmp/.streamlit"]:
 #     os.makedirs(path, exist_ok=True)
 
-# Hugging Face login using secret
+# HuggingFace login (requires HF_TOKEN to be added as secret in Hugging Face Spaces)
 from huggingface_hub import login
 login(os.environ.get("HF_TOKEN", ""))