Update start.sh

start.sh

@@ -4,48 +4,10 @@ set -Eeuo pipefail
 COMFY="${COMFYUI_PATH:-/app/ComfyUI}"
 
 # ------------------------------------------------------------
-# 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
+# Runtime directories
 # ------------------------------------------------------------
+# This is intentionally done before ComfyUI starts.
+# ComfyUI 0.35.x creates /app/ComfyUI/user itself.
 mkdir -p \
   "$COMFY/user" \
   "$COMFY/input" \
@@ -57,8 +19,23 @@ mkdir -p \
   "$COMFY/models/vae"
 
 # ------------------------------------------------------------
-# 3. Persistent storage (Hugging Face Space /data)
+# 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
 # ------------------------------------------------------------
+# 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
@@ -75,8 +52,10 @@ mkdir -p \
   "$MODEL_ROOT/vae"
 
 # ------------------------------------------------------------
-# 4. Model directory links
+# 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"
@@ -107,7 +86,7 @@ link_model_dir text_encoders
 link_model_dir vae
 
 echo "============================================================"
-echo "ComfyUI Krea2 - Hugging Face CPU Space v2 (Git/Pip Enabled)"
+echo "ComfyUI Krea2 - Hugging Face CPU"
 echo "============================================================"
 
 python - <<'PY'
@@ -121,8 +100,9 @@ print("RAM limit is controlled by the Hugging Face Space hardware.")
 PY
 
 # ------------------------------------------------------------
-# 5. Model configuration & Downloader
+# Model configuration
 # ------------------------------------------------------------
+# 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}"
 
@@ -140,6 +120,9 @@ 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"
@@ -176,6 +159,9 @@ 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"
@@ -186,7 +172,7 @@ else
 fi
 
 # ------------------------------------------------------------
-# 6. CPU optimization & Start
+# CPU optimization
 # ------------------------------------------------------------
 export OMP_NUM_THREADS="${OMP_NUM_THREADS:-2}"
 export MKL_NUM_THREADS="${MKL_NUM_THREADS:-2}"