Upload start.sh with huggingface_hub

start.sh

@@ -14,9 +14,10 @@ mkdir -p /tmp/comfyui_output
 mkdir -p /tmp/comfyui_temp
 mkdir -p /tmp/comfyui_user
 mkdir -p /tmp/workflows
+mkdir -p /tmp/comfyui_models
 
 # Set proper permissions
-chmod -R 777 /tmp/comfyui_input /tmp/comfyui_output /tmp/comfyui_temp /tmp/comfyui_user /tmp/workflows
+chmod -R 777 /tmp/comfyui_input /tmp/comfyui_output /tmp/comfyui_temp /tmp/comfyui_user /tmp/workflows /tmp/comfyui_models
 
 # Download files from the dataset repository
 echo "Downloading images from dataset repository..."
@@ -31,14 +32,28 @@ echo "Verifying downloaded files:"
 ls -la /tmp/comfyui_input/
 ls -la /tmp/workflows/
 
+# Run model downloader
+echo "Downloading models..."
+cd /app
+python3 download_models.py
+
+# Set environment variables
+export PYTHONPATH="${PYTHONPATH}:/app/ComfyUI:/app/ComfyUI/custom_nodes"
+
 # Install additional Python packages
 echo "Installing required Python packages..."
+pip install -e /app/ComfyUI/custom_nodes/PuLID || echo "Failed to install PuLID, continuing anyway"
 pip install comfyui-frontend-package comfyui-workflow-templates || echo "Package installation failed, continuing anyway"
 
 # Start ComfyUI in the background with custom directories
 cd /app/ComfyUI
 echo "Starting ComfyUI server..."
-python3 main.py --listen 0.0.0.0 --port 8188 --input-directory /tmp/comfyui_input --output-directory /tmp/comfyui_output --temp-directory /tmp/comfyui_temp --user-dir /tmp/comfyui_user &
+python3 main.py --listen 0.0.0.0 --port 8188 \
+  --input-directory /tmp/comfyui_input \
+  --output-directory /tmp/comfyui_output \
+  --temp-directory /tmp/comfyui_temp \
+  --user-dir /tmp/comfyui_user \
+  --models-directory /tmp/comfyui_models &
 COMFY_PID=$!
 
 # Wait for ComfyUI to start
@@ -53,6 +68,10 @@ else
     echo "Proceeding anyway..."
 fi
 
+# List loaded custom nodes for debugging
+echo "Custom nodes check:"
+curl -s "http://localhost:8188/object_info" | grep -i "pulid" || echo "PuLID nodes not found"
+
 # Start the web interface with updated workflow path
 cd /app
 echo "Starting Gradio interface..."