#!/bin/bash
set -e

# --- Initialize or update ComfyUI repo ---
if [ -d "ComfyUI" ]; then
  cd ComfyUI
  git pull
else
  git clone https://github.com/comfyanonymous/ComfyUI.git
  cd ComfyUI
fi

# --- Python virtual environment ---
if [ ! -d "venv" ]; then
  python3 -m venv venv
fi
source venv/bin/activate

# --- Core requirements for ComfyUI ---
pip install --upgrade pip
pip install -r requirements.txt

# --- Install/Update ComfyUI Manager ---
cd custom_nodes
if [ ! -d "ComfyUI-Manager" ]; then
  git clone https://github.com/ltdrdata/ComfyUI-Manager.git
fi
cd ComfyUI-Manager
git pull
if [ -f requirements.txt ]; then
  pip install -r requirements.txt
fi
cd ../

# --- Install/Update ComfyUI-GGUF (GGUF support) ---
if [ ! -d "ComfyUI-GGUF" ]; then
  git clone https://github.com/city96/ComfyUI-GGUF
fi
cd ComfyUI-GGUF
git pull
if [ -f requirements.txt ]; then
  pip install -r requirements.txt
fi
cd ..
pip install --upgrade gguf
cd ../

echo "Setup complete."