Upload start.sh with huggingface_hub

start.sh

@@ -7,14 +7,41 @@ echo "CUDA version: $(nvcc --version 2>/dev/null || echo 'NVCC not found')"
 echo "GPU information: $(nvidia-smi || echo 'NVIDIA-SMI not found')"
 echo "==========================="
 
+# Create necessary directories if they don't exist
+mkdir -p /app/ComfyUI/input
+mkdir -p /app/ComfyUI/output
+mkdir -p /app/workflows
+
 # Start ComfyUI in the background
 cd /app/ComfyUI
+echo "Starting ComfyUI server..."
 python3 main.py --listen 0.0.0.0 --port 8188 &
 COMFY_PID=$!
 
-# Wait for ComfyUI to start
-echo "Starting ComfyUI server..."
-sleep 10
+# Robust check to ensure ComfyUI is actually running
+echo "Waiting for ComfyUI server to initialize..."
+MAX_RETRIES=30
+RETRY_COUNT=0
+COMFY_READY=false
+
+while [ $RETRY_COUNT -lt $MAX_RETRIES ]; do
+    echo "Checking if ComfyUI is ready (attempt $((RETRY_COUNT+1))/$MAX_RETRIES)..."
+    if curl -s "http://localhost:8188/system_stats" > /dev/null; then
+        COMFY_READY=true
+        echo "ComfyUI server is up and running!"
+        break
+    fi
+    RETRY_COUNT=$((RETRY_COUNT+1))
+    sleep 5
+done
+
+if [ "$COMFY_READY" = false ]; then
+    echo "WARNING: ComfyUI server didn't respond within the expected time."
+    echo "The application may not function correctly."
+fi
+
+# Print message about network configuration
+echo "ComfyUI should be accessible at: http://0.0.0.0:8188"
 
 # Start the web interface
 cd /app
@@ -22,4 +49,4 @@ echo "Starting Gradio interface..."
 python3 app.py
 
 # If the Gradio app crashes, kill ComfyUI too
-kill $COMFY_PID
+kill $COMFY_PID