Upload start.sh with huggingface_hub

start.sh

@@ -16,6 +16,11 @@ mkdir -p /app/workflows
 mkdir -p /tmp/comfyui_user
 chmod 777 /tmp/comfyui_user
 
+# Install ComfyUI requirements to make sure frontend package is installed
+echo "Installing ComfyUI requirements..."
+cd /app/ComfyUI
+pip install -r requirements.txt
+
 # List the contents of key directories to debug
 echo "Checking for workflow files:"
 ls -la /app/workflows/ || echo "Workflow directory empty or not found"
@@ -23,19 +28,17 @@ ls -la /app/workflows/ || echo "Workflow directory empty or not found"
 echo "Checking for input images:"
 ls -la /app/ComfyUI/input/ || echo "Input directory empty or not found"
 
-# Download pose4.jpg image if it's missing or is an LFS pointer
-if [ ! -s "/app/ComfyUI/input/pose4.jpg" ] || grep -q "version https://git-lfs.github.com/spec/" /app/ComfyUI/input/pose4.jpg; then
-    echo "Downloading pose4.jpg image..."
-    # Directly download from Hugging Face repo
-    curl -L -o /app/ComfyUI/input/pose4.jpg "https://huggingface.co/spaces/Defter77/Avatars/resolve/main/ComfyUI/input/pose4.jpg"
-fi
+# Download pose4.jpg image
+echo "Downloading pose4.jpg image..."
+curl -L -o /app/ComfyUI/input/pose4.jpg "https://huggingface.co/spaces/Defter77/Avatars/resolve/main/ComfyUI/input/pose4.jpg" || echo "Failed to download pose4.jpg"
 
-# Download paita2.jpg image if it's missing or is an LFS pointer 
-if [ ! -s "/app/ComfyUI/input/paita2.jpg" ] || grep -q "version https://git-lfs.github.com/spec/" /app/ComfyUI/input/paita2.jpg; then
-    echo "Downloading paita2.jpg image..."
-    # Directly download from Hugging Face repo
-    curl -L -o /app/ComfyUI/input/paita2.jpg "https://huggingface.co/spaces/Defter77/Avatars/resolve/main/ComfyUI/input/paita2.jpg"
-fi
+# Download paita2.jpg image
+echo "Downloading paita2.jpg image..."
+curl -L -o /app/ComfyUI/input/paita2.jpg "https://huggingface.co/spaces/Defter77/Avatars/resolve/main/ComfyUI/input/paita2.jpg" || echo "Failed to download paita2.jpg"
+
+# Verify images were downloaded
+echo "Verifying downloaded images:"
+ls -la /app/ComfyUI/input/
 
 # Start ComfyUI in the background with custom user directory
 cd /app/ComfyUI
@@ -47,6 +50,14 @@ COMFY_PID=$!
 echo "Waiting for ComfyUI server to initialize..."
 sleep 30  # Give it more time to start
 
+# Check if ComfyUI is running
+if curl -s "http://localhost:8188/system_stats" > /dev/null; then
+    echo "ComfyUI server is up and running!"
+else
+    echo "WARNING: ComfyUI server might not be running correctly."
+    echo "Proceeding anyway..."
+fi
+
 # Start the web interface
 cd /app
 echo "Starting Gradio interface..."