init

app.py

@@ -1,25 +1,42 @@
-# app.py
-import gradio as gr
-import subprocess
-import spaces
 import os
 import sys
-import datetime
-import shutil
-import time # Moved import time to the top for global access
 import argparse
+import datetime
+import subprocess
+import gradio as gr
+import spaces
+from huggingface_hub import snapshot_download
+
+# ----------------------------------------------------------------------------
+# ▶ 运行时环境探测 & 路径配置
+# ----------------------------------------------------------------------------
+PERSIST_ROOT = "/data" if os.path.isdir("/data") else "/tmp"
+HF_CACHE_DIR = os.path.join(PERSIST_ROOT, ".huggingface")
+MODEL_REPO = "Wan-AI/Wan2.1-T2V-1.3B"
+MODEL_DIR = os.path.join(PERSIST_ROOT, "checkpoints", "Wan2.1-T2V-1.3B")
+
+os.makedirs(HF_CACHE_DIR, exist_ok=True)
+os.makedirs(MODEL_DIR, exist_ok=True)
+os.environ["HF_HOME"] = HF_CACHE_DIR
+
+if not os.path.exists(os.path.join(MODEL_DIR, "model_index.json")):
+    print(f"[Warm‑up] Downloading model {MODEL_REPO} to {MODEL_DIR} …")
+    snapshot_download(
+        repo_id=MODEL_REPO,
+        local_dir=MODEL_DIR,
+        local_dir_use_symlinks=False,
+        resume_download=True,
+    )
+    print("[Warm‑up] Model download complete.")
 
-# --- Configuration ---
-# !!! IMPORTANT: Ensure this path is correct for your environment !!!
-CKPT_DIR = "./checkpoints/Wan2.1-T2V-1.3B"
-EDIT_SCRIPT_PATH = "edit.py"  # Assumes edit.py is in the same directory
+CKPT_DIR = MODEL_DIR
+EDIT_SCRIPT_PATH = "edit.py"
 OUTPUT_DIR = "gradio_outputs"
-PYTHON_EXECUTABLE = sys.executable # Uses the same python that runs gradio
-VIDEO_EXAMPLES_DIR = "video_list" # Directory for example videos
+VIDEO_EXAMPLES_DIR = "video_list"
+PYTHON_EXECUTABLE = sys.executable
 
-# Create output directory if it doesn't exist
 os.makedirs(OUTPUT_DIR, exist_ok=True)
-os.makedirs(VIDEO_EXAMPLES_DIR, exist_ok=True) # Ensure video_list exists for clarity
+os.makedirs(VIDEO_EXAMPLES_DIR, exist_ok=True)
 
 def _parse_args():
     parser = argparse.ArgumentParser(