Update rag.py

rag.py

@@ -1,5 +1,4 @@
 import os
-import logging
 import streamlit as st
 from dotenv import load_dotenv
 import pickle
@@ -16,7 +15,12 @@ from llama_index.core import get_response_synthesizer
 from llama_index.core.agent import ReActAgent
 from chromadb import PersistentClient
 
-logger = logging.getLogger(__name__)
+# writable cache directories
+os.environ["TRANSFORMERS_CACHE"] = "/app/.cache"
+os.environ["HF_HOME"] = "/app/.cache"
+os.environ["XDG_CACHE_HOME"] = "/app/.cache"  # Some libs use this
+
+os.makedirs("/app/.cache", exist_ok=True)
 
 @st.cache_resource
 def setup_rag_system(debug=False):
@@ -167,8 +171,6 @@ def setup_rag_system(debug=False):
         system_prompt=system_prompt
     )
 
-    logger.info("RAG system setup complete.")
-
     if debug:
         return agent, llm, hybrid_retrievers