Update src/streamlit_app.py

src/streamlit_app.py

@@ -12,6 +12,18 @@ import streamlit as st
 import pandas as pd
 import rag  # Import the rag module
 import os   # Import os for file path handling
+cache_dir_base = "/app/.cache_app"
+
+# Create the base directory if it doesn't exist
+os.makedirs(cache_dir_base, exist_ok=True)
+
+# Set environment variables for Hugging Face libraries
+os.environ['HF_HOME'] = os.path.join(cache_dir_base, 'huggingface_hub') # For hub downloads
+os.environ['TRANSFORMERS_CACHE'] = os.path.join(cache_dir_base, 'transformers') # For transformers models
+os.environ['SENTENCE_TRANSFORMERS_HOME'] = os.path.join(cache_dir_base, 'sentence_transformers') # For sentence-transformers
+
+# Set environment variable for pip cache (optional but good practice)
+os.environ['PIP_CACHE_DIR'] = os.path.join(cache_dir_base, 'pip')
 
 
 st.set_page_config(layout="wide", page_title="Communication Board")