full read and write permissions to the working directory

Dockerfile

@@ -5,8 +5,17 @@ FROM node:20-alpine AS builder
 RUN apk add --no-cache libc6-compat
 WORKDIR /app
 
-# Give permissions to the app directory
-RUN chmod -R 777 /app
+# Create writable directories for Hugging Face Spaces
+RUN mkdir -p /tmp/huggingface && \
+    chmod -R 777 /tmp/huggingface && \
+    mkdir -p /app/workspace && \
+    chmod -R 777 /app/workspace
+
+# Set cache environment variables at build time
+ENV HF_HOME=/tmp/huggingface \
+    TRANSFORMERS_CACHE=/tmp/huggingface \
+    XDG_CACHE_HOME=/tmp \
+    WRITABLE_DIR=/app/workspace
 
 # Copy the 'frontend' folder from the project root into the container
 COPY frontend ./frontend

main.py

@@ -17,6 +17,8 @@ from google.api_core.exceptions import ResourceExhausted
 logger = logging.getLogger()
 logger.setLevel(logging.INFO)
 
+ENV_FILE_PATH = os.getenv("WRITABLE_DIR", "/tmp") + "/.env"
+
 CONTEXT_LENGTH = 128000
 BUFFER = 10000
 MAX_TOKENS_ALLOWED = CONTEXT_LENGTH - BUFFER
@@ -35,7 +37,7 @@ def format_error_sse(event_type: str, data: str) -> str:
 
 # Initialize the components
 def initialize_components():
-    load_dotenv(override=True)
+    load_dotenv(ENV_FILE_PATH, override=True)
 
     from src.search.search_engine import SearchEngine
     from src.query_processing.query_processor import QueryProcessor

src/helpers/helper.py

@@ -5,7 +5,7 @@ import torch
 import transformers
 from langchain.text_splitter import RecursiveCharacterTextSplitter, TokenTextSplitter
 
-ENV_FILE_PATH = os.path.join(os.path.dirname(__file__), "../../.env").replace("\\", "/")
+ENV_FILE_PATH = os.path.join(os.getenv("WRITABLE_DIR", "/tmp"), ".env")
 
 def remove_markdown(text: str) -> str:
         # Remove code block format type and the code block itself