Update start.sh

start.sh

@@ -4,10 +4,48 @@ set -Eeuo pipefail
 COMFY="${COMFYUI_PATH:-/app/ComfyUI}"
 
 # ------------------------------------------------------------
-# Runtime directories
+# 1. Bootstrap & Update ComfyUI (Git & Pip)
+# ------------------------------------------------------------
+# ตรวจสอบว่ามีโฟลเดอร์ ComfyUI หรือไม่ ถ้าไม่มีให้ Clone ใหม่
+if [ ! -d "$COMFY" ]; then
+  echo "[INFO] ComfyUI not found. Cloning from GitHub..."
+  git clone https://github.com/comfyanonymous/ComfyUI.git "$COMFY"
+fi
+
+# อัปเดต ComfyUI และ Dependencies หากตั้งค่า COMFYUI_UPDATE=1
+if [ "${COMFYUI_UPDATE:-0}" = "1" ]; then
+  echo "[INFO] Updating ComfyUI core via git pull..."
+  cd "$COMFY"
+  git pull origin master
+  
+  echo "[INFO] Updating ComfyUI dependencies via pip..."
+  python -m pip install --no-cache-dir --upgrade -r requirements.txt
+fi
+
+# อัปเดต Custom Nodes ทั้งหมดที่มีไฟล์ .git และ requirements.txt
+if [ "${COMFYUI_UPDATE:-0}" = "1" ] && [ -d "$COMFY/custom_nodes" ]; then
+  echo "[INFO] Updating custom nodes via git & pip..."
+  find "$COMFY/custom_nodes" -mindepth 1 -maxdepth 1 -type d | while read -r node_dir; do
+    if [ -d "$node_dir/.git" ]; then
+      echo "  -> Updating $(basename "$node_dir")"
+      git -C "$node_dir" pull --rebase || echo "  [WARN] Failed to pull $(basename "$node_dir")"
+      
+      if [ -f "$node_dir/requirements.txt" ]; then
+        python -m pip install --no-cache-dir -r "$node_dir/requirements.txt" || echo "  [WARN] Failed to install requirements for $(basename "$node_dir")"
+      fi
+    fi
+  done
+fi
+
+# ติดตั้ง Python Packages เพิ่มเติมแบบไดนามิก (ถ้ามีการระบุในตัวแปรสภาพแวดล้อม)
+if [ -n "${EXTRA_PIP_PACKAGES:-}" ]; then
+  echo "[INFO] Installing extra pip packages: $EXTRA_PIP_PACKAGES"
+  python -m pip install --no-cache-dir $EXTRA_PIP_PACKAGES
+fi
+
+# ------------------------------------------------------------
+# 2. Runtime directories
 # ------------------------------------------------------------
-# This is intentionally done before ComfyUI starts.
-# ComfyUI 0.35.x creates /app/ComfyUI/user itself.
 mkdir -p \
   "$COMFY/user" \
   "$COMFY/input" \
@@ -19,23 +57,8 @@ mkdir -p \
   "$COMFY/models/vae"
 
 # ------------------------------------------------------------
-# ComfyUI Manager Security Setup
-# ------------------------------------------------------------
-MANAGER_CONFIG_DIR="$COMFY/user/default/ComfyUI-Manager"
-# mkdir -p "$MANAGER_CONFIG_DIR"
-# cat << 'EOF' > "$MANAGER_CONFIG_DIR/config.ini"
-# [ComfyUI-Manager]
-# security_level = weak
-# privacy_mode = false
-# EOF
-# echo "[INFO] ComfyUI Manager security level set to weak."
-
-# ------------------------------------------------------------
-# Persistent storage
+# 3. Persistent storage (Hugging Face Space /data)
 # ------------------------------------------------------------
-# If HF Storage is mounted at /data, models survive Space restarts.
-# Without persistent storage, downloaded models live in the container
-# filesystem and may have to be downloaded again after a rebuild/restart.
 PERSIST_ROOT="/data/comfyui"
 
 if [ -d "/data" ] && [ -w "/data" ]; then
@@ -52,10 +75,8 @@ mkdir -p \
   "$MODEL_ROOT/vae"
 
 # ------------------------------------------------------------
-# Model directory links
+# 4. Model directory links
 # ------------------------------------------------------------
-# Only replace an empty ComfyUI model directory.
-# This avoids deleting any models supplied in the Space image.
 link_model_dir() {
   local name="$1"
   local target="$COMFY/models/$name"
@@ -86,7 +107,7 @@ link_model_dir text_encoders
 link_model_dir vae
 
 echo "============================================================"
-echo "ComfyUI Krea2 - Hugging Face CPU Space v2"
+echo "ComfyUI Krea2 - Hugging Face CPU Space v2 (Git/Pip Enabled)"
 echo "============================================================"
 
 python - <<'PY'
@@ -100,9 +121,8 @@ print("RAM limit is controlled by the Hugging Face Space hardware.")
 PY
 
 # ------------------------------------------------------------
-# Model configuration
+# 5. Model configuration & Downloader
 # ------------------------------------------------------------
-# Q3_K_S is selected for the ~15 GB RAM CPU Space.
 GGUF_REPO="${KREA2_GGUF_REPO:-realrebelai/KREA-2_GGUFs}"
 GGUF_FILE="${KREA2_GGUF_FILE:-TURBO/Krea-2-Turbo-Q3_K_S.gguf}"
 
@@ -120,9 +140,6 @@ QWEN_PATH="$COMFY/models/text_encoders/$(basename "$QWEN_FILE")"
 VAE_PATH="$COMFY/models/vae/$(basename "$VAE_FILE")"
 WORKFLOW_PATH="$COMFY/workflows/$(basename "$WORKFLOW_FILE")"
 
-# ------------------------------------------------------------
-# Hugging Face downloader
-# ------------------------------------------------------------
 download_hf() {
   local repo="$1"
   local file="$2"
@@ -159,9 +176,6 @@ print("[OK] downloaded:", dest)
 PY
 }
 
-# ------------------------------------------------------------
-# Download models
-# ------------------------------------------------------------
 if [ "${KREA2_DOWNLOAD_MODELS:-1}" = "1" ]; then
   download_hf "$GGUF_REPO" "$GGUF_FILE" "$GGUF_PATH"
   download_hf "$QWEN_REPO" "$QWEN_FILE" "$QWEN_PATH"
@@ -172,7 +186,7 @@ else
 fi
 
 # ------------------------------------------------------------
-# CPU optimization
+# 6. CPU optimization & Start
 # ------------------------------------------------------------
 export OMP_NUM_THREADS="${OMP_NUM_THREADS:-2}"
 export MKL_NUM_THREADS="${MKL_NUM_THREADS:-2}"
@@ -190,4 +204,4 @@ exec python -u main.py \
   --cpu \
   --listen 0.0.0.0 \
   --port "${PORT:-7860}" \
-  --output-directory "$COMFY/output"
+  --output-directory "$COMFY/output"