Add musubi-tuner integration: clone repository at startup and set default directory and repo URL

.gitignore

@@ -0,0 +1 @@
+/musubi-tuner

app.py

@@ -24,6 +24,12 @@ DEFAULT_DATASET_CONFIG = "/workspace/auto/dataset_QIE.toml"
 DEFAULT_MODELS_ROOT = DEFAULT_MODELS_DIR  # "/workspace/Qwen-Image_models"
 WORKSPACE_AUTO_DIR = "/workspace/auto"
 
+# musubi-tuner settings
+DEFAULT_MUSUBI_TUNER_DIR = os.environ.get("MUSUBI_TUNER_DIR", "/musubi-tuner")
+DEFAULT_MUSUBI_TUNER_REPO = os.environ.get(
+    "MUSUBI_TUNER_REPO", "https://github.com/kohya-ss/musubi-tuner.git"
+)
+
 
 TRAINING_DIR = Path(__file__).resolve().parent
 
@@ -150,6 +156,49 @@ def _pick_shell() -> str:
     raise RuntimeError("No POSIX shell found. Please install bash or sh.")
 
 
+def _is_git_repo(path: str) -> bool:
+    try:
+        out = subprocess.run(
+            ["git", "-C", path, "rev-parse", "--is-inside-work-tree"],
+            capture_output=True,
+            text=True,
+            check=False,
+        )
+        return out.returncode == 0 and out.stdout.strip() == "true"
+    except Exception:
+        return False
+
+
+def _startup_clone_musubi_tuner() -> None:
+    target = DEFAULT_MUSUBI_TUNER_DIR
+    repo = DEFAULT_MUSUBI_TUNER_REPO
+    parent = os.path.dirname(target.rstrip("/\\")) or "/"
+    try:
+        os.makedirs(parent, exist_ok=True)
+    except Exception:
+        pass
+
+    if os.path.isdir(target) and _is_git_repo(target):
+        print(f"[QIE] musubi-tuner exists at {target}; pulling latest...")
+        try:
+            subprocess.run(["git", "-C", target, "fetch", "--all", "--prune"], check=False)
+            subprocess.run(["git", "-C", target, "pull", "--ff-only"], check=False)
+        except Exception as e:
+            print(f"[QIE] git pull failed: {e}")
+        return
+
+    if os.path.exists(target) and not _is_git_repo(target):
+        print(f"[QIE] Warning: {target} exists and is not a git repo. Skipping clone.")
+        return
+
+    print(f"[QIE] Cloning musubi-tuner into {target} from {repo} ...")
+    try:
+        subprocess.run(["git", "clone", "--depth", "1", repo, target], check=True)
+        print("[QIE] Clone completed.")
+    except subprocess.CalledProcessError as e:
+        print(f"[QIE] Clone failed: {e}")
+
+
 def run_training(
     dataset_name: str,
     caption: str,
@@ -274,10 +323,12 @@ def _startup_download_models() -> None:
 
 
 if __name__ == "__main__":
-    # 1) Download models at startup (blocking by design)
+    # 1) Ensure musubi-tuner is cloned before anything else
+    _startup_clone_musubi_tuner()
+
+    # 2) Download models at startup (blocking by design)
     _startup_download_models()
 
-    # 2) Launch Gradio app
+    # 3) Launch Gradio app
     ui = build_ui()
     ui.launch(server_name="0.0.0.0", server_port=int(os.environ.get("PORT", 7860)))
-