Update core/image_generator.py

core/image_generator.py

@@ -21,6 +21,8 @@ TMP_DIR = Path("/tmp/generated_images")
 for d in [HF_CACHE_DIR, MODEL_DIR, SEED_DIR, TMP_DIR]:
     d.mkdir(parents=True, exist_ok=True)
 
+
+# Ensure all relevant environment variables are correctly set
 os.environ.update({
     "HF_HOME": str(HF_CACHE_DIR),
     "HF_HUB_CACHE": str(HF_CACHE_DIR),
@@ -32,9 +34,24 @@ os.environ.update({
     "TMPDIR": str(HF_CACHE_DIR),
     "CACHE_DIR": str(HF_CACHE_DIR),
 })
+
+# Force Python's tempfile to use /tmp/hf_cache too
 import tempfile
 tempfile.tempdir = str(HF_CACHE_DIR)
 
+# 🔒 Extra layer: patch os.path.expanduser to block “/.cache”
+import os.path
+def safe_expanduser(path):
+    if path.startswith("~") or path.startswith("/.cache"):
+        return str(HF_CACHE_DIR)
+    return os.path.expanduser_original(path)
+
+os.path.expanduser_original = os.path.expanduser
+os.path.expanduser = safe_expanduser
+
+print("[DEBUG] ✅ Hugging Face and Diffusers cache fully redirected to:", HF_CACHE_DIR)
+
+
 
 print("[DEBUG] ✅ Using persistent Hugging Face cache at:", HF_CACHE_DIR)
 print("[DEBUG] ✅ Model directory:", MODEL_DIR)