Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,4 +1,20 @@
 import os
+# ✅ Set all relevant cache directories to a writable location
+os.environ["HF_HOME"] = "/tmp/cache"
+os.environ["TRANSFORMERS_CACHE"] = "/tmp/cache/transformers"
+os.environ["SENTENCE_TRANSFORMERS_HOME"] = "/tmp/cache/sentence_transformers"
+os.environ["HF_DATASETS_CACHE"] = "/tmp/cache/hf_datasets"
+os.environ["TORCH_HOME"] = "/tmp/cache/torch"
+
+# ✅ Create the directories if they don't exist
+for path in [
+    "/tmp/cache",
+    "/tmp/cache/transformers",
+    "/tmp/cache/sentence_transformers",
+    "/tmp/cache/hf_datasets",
+    "/tmp/cache/torch"
+]:
+    os.makedirs(path, exist_ok=True)
 import streamlit as st
 import torch
 from sentence_transformers import SentenceTransformer, util
@@ -9,18 +25,6 @@ import openai
 from io import BytesIO
 import requests
 
-# ========== ✅ ENV SETUP FOR HF SPACES ==========
-os.environ['HF_HOME'] = './cache'
-os.environ['TRANSFORMERS_CACHE'] = './cache/transformers'
-os.environ['SENTENCE_TRANSFORMERS_HOME'] = './cache/sentence_transformers'
-os.environ['HF_DATASETS_CACHE'] = './cache/hf_datasets'
-os.environ['TORCH_HOME'] = './cache/torch'
-
-# Ensure required cache directories exist
-os.makedirs('./cache/transformers', exist_ok=True)
-os.makedirs('./cache/sentence_transformers', exist_ok=True)
-os.makedirs('./cache/hf_datasets', exist_ok=True)
-os.makedirs('./cache/torch', exist_ok=True)
 
 # ========== 🔑 API Key ==========
 openai.api_key = os.getenv("OPENAI_API_KEY")