Update modules/utils.py

modules/utils.py

@@ -12,6 +12,7 @@ from typing import Any, Dict, Optional, List
 # ------------------------------------------------------------
 _DATA_DIR: Optional[Path] = None
 _EXPORT_DIR: Optional[Path] = None
+_HF_INIT_DONE: bool = False
 
 def _is_writable(p: Path) -> bool:
     try:
@@ -33,6 +34,34 @@ def _pick_writable_dir(candidates: List[Path]) -> Path:
     fallback.mkdir(parents=True, exist_ok=True)
     return fallback
 
+def _init_hf_env(base: Path) -> None:
+    """
+    Hugging Face / Transformers / Sentence-Transformers の
+    キャッシュ＆ホームをすべて base 配下に固定して PermissionError を回避。
+    """
+    global _HF_INIT_DONE
+    if _HF_INIT_DONE:
+        return
+
+    hf_home     = base / "hf_home"
+    hf_cache    = base / "hf_cache"
+    hf_home.mkdir(parents=True, exist_ok=True)
+    hf_cache.mkdir(parents=True, exist_ok=True)
+
+    # 主要な環境変数を強制設定（既存設定より優先）
+    os.environ.setdefault("HF_HOME", str(hf_home))
+    os.environ.setdefault("HUGGINGFACE_HUB_CACHE", str(hf_cache))
+    os.environ.setdefault("TRANSFORMERS_CACHE", str(hf_cache))
+    os.environ.setdefault("SENTENCE_TRANSFORMERS_HOME", str(hf_cache))
+
+    # 余計な参照を抑制
+    os.environ.setdefault("HF_HUB_DISABLE_TELEMETRY", "1")
+    os.environ.setdefault("HF_HUB_DISABLE_PROGRESS_BARS", "1")
+    # 既定のトークン探索を避けるため空文字に（公開モデルの匿名DLを想定）
+    os.environ.setdefault("HF_TOKEN", "")
+
+    _HF_INIT_DONE = True
+
 def ensure_dirs() -> None:
     """書き込み可能なデータディレクトリを決定して作成。"""
     global _DATA_DIR, _EXPORT_DIR
@@ -61,6 +90,9 @@ def ensure_dirs() -> None:
     _DATA_DIR = chosen
     _EXPORT_DIR = export
 
+    # ★ Hugging Face 関連のホーム/キャッシュを、この書き込み可能ベースに固定
+    _init_hf_env(chosen)
+
 def data_dir() -> Path:
     ensure_dirs()
     return _DATA_DIR  # type: ignore