#!/bin/bash
#
# ComfyUI Startup Script
# Starts ComfyUI server on port 8188
#

WORKSPACE_DIR="${WORKSPACE_DIR:-/workspace}"
COMFYUI_DIR="${WORKSPACE_DIR}/ComfyUI"
HF_CACHE="${WORKSPACE_DIR}/huggingface_cache"

# Set environment variables
export PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True
export HF_HOME="${HF_CACHE}"
export TQDM_DISABLE=1

# Navigate to ComfyUI directory
cd "${COMFYUI_DIR}" || exit 1

# Activate virtual environment
if [ -d "venv" ]; then
  echo "Activating ComfyUI virtual environment..."
  source venv/bin/activate
else
  echo "WARNING: venv not found, using system Python"
fi

echo "Starting ComfyUI on port 8188..."
echo "Access at: http://localhost:8188"
echo "Using HuggingFace cache: ${HF_CACHE}"

# Start ComfyUI with GPU support
python main.py \
  --listen 0.0.0.0 \
  --port 8188 \
  --enable-cors-header \
  --preview-method auto

echo "ComfyUI stopped"