Update start.sh

start.sh

@@ -6,6 +6,8 @@ COMFY="${COMFYUI_PATH:-/app/ComfyUI}"
 # ------------------------------------------------------------
 # Runtime directories
 # ------------------------------------------------------------
+# This is intentionally done before ComfyUI starts.
+# ComfyUI 0.35.x creates /app/ComfyUI/user itself.
 mkdir -p \
   "$COMFY/user" \
   "$COMFY/input" \
@@ -13,33 +15,69 @@ mkdir -p \
   "$COMFY/temp" \
   "$COMFY/workflows" \
   "$COMFY/models/unet" \
+  "$COMFY/models/checkpoints" \
+  "$COMFY/models/audio_encoders" \
   "$COMFY/models/text_encoders" \
   "$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
 # ------------------------------------------------------------
+# 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
   mkdir -p "$PERSIST_ROOT/models"
-  mkdir -p "$PERSIST_ROOT/output"
-  mkdir -p "$PERSIST_ROOT/input"
-  mkdir -p "$PERSIST_ROOT/temp"
-  mkdir -p "$PERSIST_ROOT/workflows"
 else
   PERSIST_ROOT="$COMFY"
 fi
 
 MODEL_ROOT="$PERSIST_ROOT/models"
-OUTPUT_ROOT="$PERSIST_ROOT/output"
-INPUT_ROOT="$PERSIST_ROOT/input"
-TEMP_ROOT="$PERSIST_ROOT/temp"
-WORKFLOW_ROOT="$PERSIST_ROOT/workflows"
+
+# ------------------------------------------------------------
+# Model configuration
+# ------------------------------------------------------------
+# Define models to download.
+# Format: "repo_id|filename|dest_subdir"
+# dest_subdir should be one of: unet, text_encoders, vae, loras, controlnet, workflows, etc.
+# 
+# You can override this list by setting the MODELS_TO_DOWNLOAD environment variable.
+# Example: export MODELS_TO_DOWNLOAD="repo1|file1|unet;repo2|file2|vae"
+
+if [ -n "${MODELS_TO_DOWNLOAD:-}" ]; then
+  IFS=';' read -r -a MODEL_LIST <<< "$MODELS_TO_DOWNLOAD"
+else
+  MODEL_LIST=(
+    "realrebelai/KREA-2_GGUFs|TURBO/Krea-2-Turbo-Q3_K_S.gguf|unet"
+    "Comfy-Org/Qwen3-VL|text_encoders/qwen3vl_4b_fp8_scaled.safetensors|text_encoders"
+    "Comfy-Org/Qwen-Image_ComfyUI|split_files/vae/qwen_image_vae.safetensors|vae"
+    "realrebelai/KREA-2_GGUFs|Rebels KREA-2-TURBO.json|workflows"
+    "Comfy-Org/YuE2|checkpoints/yue2_3b_int8_convrot.safetensors|checkpoints"
+    "Comfy-Org/YuE2|audio_encoders/sheetsage2_bf16.safetensors|audio_encoders"    
+    # เพิ่มโมเดลอื่นๆ ได้ที่นี่ เช่น:
+    # "some-repo/some-model|path/to/model.safetensors|loras"audio_encoders
+  )
+fi
 
 # ------------------------------------------------------------
 # 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"
@@ -65,67 +103,12 @@ link_model_dir() {
   fi
 }
 
-# ------------------------------------------------------------
-# Output/Input/Temp/Workflow directory links
-# ------------------------------------------------------------
-# Link output, input, temp, workflows to persistent storage
-# This ensures files survive restarts and are shared across sessions
-link_runtime_dir() {
-  local name="$1"
-  local target="$COMFY/$name"
-  local source="$PERSIST_ROOT/$name"
-
-  mkdir -p "$source"
-
-  if [ "$source" = "$target" ]; then
-    return
-  fi
-
-  if [ -L "$target" ]; then
-    return
-  fi
-
-  if [ -d "$target" ]; then
-    if [ -z "$(find "$target" -mindepth 1 -maxdepth 1 -print -quit 2>/dev/null)" ]; then
-      rmdir "$target"
-      ln -s "$source" "$target"
-      echo "[INFO] Linked $target -> $source"
-    else
-      echo "[INFO] Keeping non-empty directory: $target"
-    fi
-  fi
-}
-
-# ------------------------------------------------------------
-# Model configuration
-# ------------------------------------------------------------
-if [ -n "${MODELS_TO_DOWNLOAD:-}" ]; then
-  IFS=';' read -r -a MODEL_LIST <<< "$MODELS_TO_DOWNLOAD"
-else
-  MODEL_LIST=(
-    "realrebelai/KREA-2_GGUFs|TURBO/Krea-2-Turbo-Q3_K_S.gguf|unet"
-    "Comfy-Org/Qwen3-VL|text_encoders/qwen3vl_4b_fp8_scaled.safetensors|text_encoders"
-    "Comfy-Org/Qwen-Image_ComfyUI|split_files/vae/qwen_image_vae.safetensors|vae"
-    "realrebelai/KREA-2_GGUFs|Rebels KREA-2-TURBO.json|workflows"
-    "Comfy-Org/YuE2|checkpoints/yue2_3b_int8_convrot.safetensors|checkpoints"
-    "Comfy-Org/YuE2|audio_encoders/sheetsage2_bf16.safetensors|audio_encoders"    
-    # เพิ่มโมเดลอื่นๆ ได้ที่นี่ เช่น:
-    # "some-repo/some-model|path/to/model.safetensors|loras"audio_encoders
-  )
-fi
-
-# Dynamically link model directories
+# Dynamically link model directories based on MODEL_LIST
 UNIQUE_SUBDIRS=$(printf "%s\n" "${MODEL_LIST[@]}" | awk -F'|' '$3 != "" {print $3}' | grep -v "^workflows$" | sort -u)
 for subdir in $UNIQUE_SUBDIRS; do
   link_model_dir "$subdir"
 done
 
-# Link runtime directories to persistent storage
-link_runtime_dir "output"
-link_runtime_dir "input"
-link_runtime_dir "temp"
-link_runtime_dir "workflows"
-
 echo "============================================================"
 echo "ComfyUI Krea2 - Hugging Face CPU"
 echo "============================================================"
@@ -216,8 +199,6 @@ cd "$COMFY"
 echo "============================================================"
 echo "Starting ComfyUI"
 echo "============================================================"
-echo "Output directory: $COMFY/output (linked to $OUTPUT_ROOT)"
-echo "============================================================"
 
 exec python -u main.py \
   --cpu \