Use persistent bucket storage when available

app.py

@@ -30,7 +30,15 @@ HF_TOKEN = os.environ.get("HF_TOKEN", "")
 DATASET_REPO = "huanx/daniya-voice-gptsovits"
 GPT_SOVITS_REPO = "https://github.com/RVC-Boss/GPT-SoVITS.git"
 
-WORK_DIR = Path("/tmp/daniya_trainer")
+PERSISTENT_ROOT = Path(os.environ.get("PERSISTENT_ROOT", "/data"))
+EPHEMERAL_ROOT = Path("/tmp")
+ACTIVE_STORAGE_ROOT = (
+    PERSISTENT_ROOT if PERSISTENT_ROOT.is_dir() and os.access(PERSISTENT_ROOT, os.W_OK) else EPHEMERAL_ROOT
+)
+STORAGE_MODE = "persistent" if ACTIVE_STORAGE_ROOT == PERSISTENT_ROOT else "ephemeral"
+
+WORK_DIR = ACTIVE_STORAGE_ROOT / "daniya_trainer"
+TMP_ROOT = WORK_DIR / "tmp"
 HF_HOME = WORK_DIR / "hf_home"
 GPT_SOVITS_DIR = WORK_DIR / "GPT-SoVITS"
 DATASET_DIR = WORK_DIR / "dataset"
@@ -65,6 +73,9 @@ log = logging.getLogger(__name__)
 os.environ.setdefault("HF_HOME", str(HF_HOME))
 os.environ.setdefault("TRANSFORMERS_CACHE", str(HF_HOME / "transformers"))
 os.environ.setdefault("HF_HUB_DISABLE_TELEMETRY", "1")
+os.environ.setdefault("TMPDIR", str(TMP_ROOT))
+os.environ.setdefault("TMP", str(TMP_ROOT))
+os.environ.setdefault("TEMP", str(TMP_ROOT))
 
 
 @dataclass(frozen=True)
@@ -168,6 +179,7 @@ def get_version_context(version: str):
 def ensure_dirs():
     base_dirs = [
         WORK_DIR,
+        TMP_ROOT,
         HF_HOME,
         DATASET_DIR,
         EXP_ROOT,
@@ -264,8 +276,11 @@ def directory_overview():
 def directory_overview_for(ctx: VersionContext):
     return "\n".join(
         [
+            f"存储模式: {STORAGE_MODE}",
+            f"存储根目录: {ACTIVE_STORAGE_ROOT}",
             f"当前版本: {ctx.spec.version}",
             f"工作目录: {WORK_DIR}",
+            f"临时目录: {TMP_ROOT}",
             f"数据集目录: {DATASET_DIR}",
             f"日志目录: {ctx.exp_dir}",
             f"SoVITS 导出目录: {ctx.sovits_output_dir}",
@@ -796,8 +811,15 @@ def agent_status(version=DEFAULT_VERSION):
         "recommended_default": DEFAULT_VERSION,
         "stability": ctx.spec.stability,
         "dataset_prepared": dataset_prepared(ctx),
+        "storage": {
+            "mode": STORAGE_MODE,
+            "root": str(ACTIVE_STORAGE_ROOT),
+            "persistent_root": str(PERSISTENT_ROOT),
+            "tmp_root": str(TMP_ROOT),
+        },
         "paths": {
             "work_dir": str(WORK_DIR),
+            "tmp_root": str(TMP_ROOT),
             "dataset_dir": str(DATASET_DIR),
             "exp_dir": str(ctx.exp_dir),
             "sovits_output_dir": str(ctx.sovits_output_dir),