Update Dockerfile

Dockerfile

@@ -2,21 +2,74 @@ FROM python:3.11-slim
 
 WORKDIR /app
 
-# 1. ติดตั้ง System Dependencies ที่จำเป็นสำหรับ ComfyUI
-RUN apt-get update && apt-get install -y \
-    git \
-    wget \
-    libgl1 \
-    libglib2.0-0 \
-    && rm -rf /var/lib/apt/lists/*
+# 1. ติดตั้ง System Dependencies
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    git git-lfs curl ffmpeg libgl1 libglib2.0-0 wget \
+    && rm -rf /var/lib/apt/lists/* \
+    && git lfs install
 
-# 2. โหลด Source Code ของ ComfyUI
-RUN git clone https://github.com/comfyanonymous/ComfyUI.git /app/ComfyUI
+# 2. ติดตั้ง PyTorch สำหรับ CPU เท่านั้น (ประหยัดพื้นที่และ RAM)
+RUN pip install --no-cache-dir \
+    --extra-index-url https://download.pytorch.org/whl/cpu \
+    "torch==2.6.0" "torchvision==0.21.0" "torchaudio==2.6.0"
 
-WORKDIR /app/ComfyUI
+# 3. Clone ComfyUI และติดตั้ง Requirements หลัก
+RUN git clone --depth 1 https://github.com/comfyanonymous/ComfyUI.git /app/ComfyUI \
+    && pip install --no-cache-dir -r /app/ComfyUI/requirements.txt
 
-# 3. ติดตั้ง Python Dependencies (ขั้นตอนที่ขาดไป ทำให้เกิด Error alembic/sqlalchemy)
-RUN pip install --no-cache-dir -r requirements.txt
+# 4. ติดตั้ง Custom Nodes ที่จำเป็น
+RUN cd /app/ComfyUI/custom_nodes && \
+    git clone --depth 1 https://github.com/ltdrdata/ComfyUI-Manager.git && \
+    git clone --depth 1 https://github.com/city96/ComfyUI-GGUF.git && \
+    git clone --depth 1 https://github.com/crystian/ComfyUI-Crystools && \
+    git clone --depth 1 https://github.com/rgthree/rgthree-comfy.git && \
+    git clone --depth 1 https://github.com/ty0x2333/ComfyUI-Dev-Utils.git && \
+    git clone --depth 1 https://github.com/XLabs-AI/x-flux-comfyui && \
+    git clone --depth 1 https://github.com/willmiao/ComfyUI-Lora-Manager && \
+    git clone --depth 1 https://github.com/kijai/ComfyUI-KJNodes.git && \
+    git clone --depth 1 https://github.com/lquesada/ComfyUI-Inpaint-CropAndStitch && \
+    git clone --depth 1 https://github.com/gseth/ControlAltAI-Nodes && \
+    git clone --depth 1 https://github.com/ssitu/ComfyUI_UltimateSDUpscale --recursive && \
+    git clone --depth 1 https://github.com/nicolab28/ComfyUI-ClipProj && \
+    git clone --depth 1 https://github.com/ethanfel/ComfyUI-Krea2TextEncoder && \
+    git clone --depth 1 https://github.com/LAOGOU-666/Comfyui-Memory_Cleanup && \
+    git clone --depth 1 https://github.com/KLL535/ComfyUI_SimpleButcher
 
-# 4. รัน ComfyUI ในโหมดประหยัดทรัพยากร (CPU + Low VRAM)
-CMD ["python", "main.py", "--cpu", "--listen", "0.0.0.0", "--port", "7860"]
+# ติดตั้ง requirements ของ custom nodes บางตัวที่จำเป็น
+RUN cd /app/ComfyUI/custom_nodes/ComfyUI-GGUF && pip install --no-cache-dir -r requirements.txt || true
+RUN cd /app/ComfyUI/custom_nodes/ComfyUI-Crystools && pip install --no-cache-dir -r requirements.txt || true
+RUN cd /app/ComfyUI/custom_nodes/rgthree-comfy && pip install --no-cache-dir -r requirements.txt || true
+RUN cd /app/ComfyUI/custom_nodes/ComfyUI-KJNodes && pip install --no-cache-dir -r requirements.txt || true
+
+# 5. ติดตั้ง Python Packages เพิ่มเติม (เพิ่ม color-matcher สำหรับ Node ColorMatch)
+RUN pip install --no-cache-dir \
+    huggingface_hub psutil websocket-client opencv-python-headless \
+    einops transformers safetensors color-matcher
+
+# 6. สร้างโครงสร้างโฟลเดอร์
+RUN mkdir -p /app/ComfyUI/models/diffusion_models \
+    /app/ComfyUI/models/text_encoders \
+    /app/ComfyUI/models/vae \
+    /app/ComfyUI/models/loras \
+    /app/ComfyUI/models/upscale_models \
+    /app/ComfyUI/output
+
+# 7. ดาวน์โหลดโมเดลที่ Workflow นี้ต้องการ (ใช้ wget แทน ADD เพื่อป้องกัน RAM พุ่ง)
+RUN wget -c https://huggingface.co/lilcheaty/Krea2-INT8-ConvRot/resolve/main/Krea2-Turbo-int8-ConvRot.safetensors -P /app/ComfyUI/models/diffusion_models/ && \
+    wget -c https://huggingface.co/DreamFast/qwen3-4b-heretic/resolve/main/comfyui/qwen3-4b-heretic_nvfp4.safetensors -P /app/ComfyUI/models/text_encoders/ && \
+    wget -c https://huggingface.co/Comfy-Org/Krea-2/resolve/main/text_encoders/qwen3vl_4b_fp8_scaled.safetensors -P /app/ComfyUI/models/text_encoders/ && \
+    wget -c https://huggingface.co/Comfy-Org/z_image_turbo/resolve/main/split_files/vae/ae.safetensors -P /app/ComfyUI/models/vae/ && \
+    wget -c https://huggingface.co/Comfy-Org/flux2-dev/resolve/main/split_files/vae/flux2-vae.safetensors -P /app/ComfyUI/models/vae/ && \
+    wget -c https://huggingface.co/Comfy-Org/Krea-2/resolve/main/vae/qwen_image_vae.safetensors -P /app/ComfyUI/models/vae/ && \
+    wget -c https://huggingface.co/ABDALLALSWAITI/Upscalers/resolve/main/photo/4xNomosWebPhoto_RealPLKSR.pth -P /app/ComfyUI/models/upscale_models/
+
+# ดาวน์โหลดโมเดลเพิ่มเติมที่ Workflow เรียกใช้ (หาก URL เปลี่ยนแปลง กรุณาตรวจสอบจาก Hugging Face)
+RUN wget -c https://huggingface.co/Comfy-Org/flux2-klein-4B/resolve/main/split_files/diffusion_models/flux-2-klein-4b-nvfp4.safetensors -P /app/ComfyUI/models/diffusion_models/ || true
+RUN wget -c https://huggingface.co/lvladikov/Krea2-Turbo-Distill-4step-LoRA/resolve/main/krea2_turbo_4step_rank_64_lora_comfyui.safetensors -P /app/ComfyUI/models/loras/ || true
+
+# 8. Health Check
+HEALTHCHECK --interval=30s --timeout=30s --start-period=10s --retries=3 \
+    CMD curl -f http://localhost:7860/health || exit 1
+
+# 9. คำสั่งเริ่มต้น (รัน ComfyUI บน CPU พร้อมบังคับจัดการหน่วยความจำ)
+CMD ["python", "-u", "/app/ComfyUI/main.py", "--cpu", "--disable-smart-memory", "--listen", "0.0.0.0", "--port", "7860", "--output-directory", "/app/ComfyUI/output"]