devtools/scripts/docker-manage.sh

#!/bin/bash
# Agent Q3 — Docker Management Script
# Usage: ./docker-manage.sh [command]

set -e

REPO="maddegen/agent-q3"
HQ_IMAGE="$REPO-hq:latest"
EVO_IMAGE="$REPO-evo:latest"

login() {
  echo "Logging into HuggingFace Docker registry..."
  echo "$HF_TOKEN" | docker login registry.hf.space --username madDegen --password-stdin
}

build_hq() {
  echo "Building Agent Q3 [HQ] image..."
  docker build -f hq/Dockerfile -t $HQ_IMAGE .
  echo "Built: $HQ_IMAGE"
}

build_evo() {
  echo "Building Agent Q3 [Evo] image..."
  docker build -f evo/Dockerfile -t $EVO_IMAGE .
  echo "Built: $EVO_IMAGE"
}

push_hq() {
  login
  docker push $HQ_IMAGE
  echo "Pushed: $HQ_IMAGE"
}

push_evo() {
  login
  docker push $EVO_IMAGE
  echo "Pushed: $EVO_IMAGE"
}

run_hq() {
  echo "Starting Agent Q3 [HQ] locally..."
  docker run -d \
    --name agent-q3-hq \
    -p 8000:8000 \
    -p 11434:11434 \
    -e HF_TOKEN=$HF_TOKEN \
    -e OPENROUTER_API_KEY=$OPENROUTER_API_KEY \
    -v ollama_data:/root/.ollama \
    $HQ_IMAGE
  echo "HQ running at http://localhost:8000"
}

run_evo() {
  echo "Starting Agent Q3 [Evo] locally..."
  docker run -d \
    --name agent-q3-evo \
    -p 8001:8000 \
    -e HF_TOKEN=$HF_TOKEN \
    -v evo_data:/data \
    $EVO_IMAGE
  echo "Evo running at http://localhost:8001"
}

stop_all() {
  echo "Stopping all Agent Q3 containers..."
  docker stop agent-q3-hq agent-q3-evo 2>/dev/null || true
  docker rm agent-q3-hq agent-q3-evo 2>/dev/null || true
}

logs_hq() {
  docker logs -f agent-q3-hq
}

logs_evo() {
  docker logs -f agent-q3-evo
}

status() {
  echo "=== Agent Q3 Container Status ==="
  docker ps -a --filter "name=agent-q3" --format "table {{.Names}}\t{{.Status}}\t{{.Ports}}"
}

delete_hf_repos() {
  echo "Deleting duplicate HF repos (requires huggingface-cli login)..."
  huggingface-cli repo delete madDegen/agent-q3-hq --type model --yes
  huggingface-cli repo delete madDegen/agent-q3-core --type model --yes
  huggingface-cli repo delete madDegen/agent-q3-loras --type model --yes
  huggingface-cli repo delete madDegen/agent-q3-hq-data --type dataset --yes
  huggingface-cli repo delete madDegen/agent-q3-trainingevo --type dataset --yes
  huggingface-cli repo delete madDegen/agent-q3-evo --type space --yes
  huggingface-cli repo delete madDegen/agentq3-ui --type space --yes
  huggingface-cli repo delete madDegen/agentq3 --type space --yes
  echo "Done. 8 duplicate repos deleted."
}

case "$1" in
  login)         login ;;
  build-hq)      build_hq ;;
  build-evo)     build_evo ;;
  build)         build_hq && build_evo ;;
  push-hq)       push_hq ;;
  push-evo)      push_evo ;;
  push)          push_hq && push_evo ;;
  run-hq)        run_hq ;;
  run-evo)       run_evo ;;
  run)           run_hq && run_evo ;;
  stop)          stop_all ;;
  logs-hq)       logs_hq ;;
  logs-evo)      logs_evo ;;
  status)        status ;;
  delete-hf)     delete_hf_repos ;;
  *)
    echo "Agent Q3 Docker Manager"
    echo ""
    echo "Usage: $0 [command]"
    echo ""
    echo "Commands:"
    echo "  login        Login to HF Docker registry"
    echo "  build        Build both HQ + Evo images"
    echo "  build-hq     Build HQ image only"
    echo "  build-evo    Build Evo image only"
    echo "  push         Push both images to HF registry"
    echo "  run          Run both containers locally"
    echo "  run-hq       Run HQ container (port 8000)"
    echo "  run-evo      Run Evo container (port 8001)"
    echo "  stop         Stop all Agent Q3 containers"
    echo "  logs-hq      Tail HQ logs"
    echo "  logs-evo     Tail Evo logs"
    echo "  status       Show container status"
    echo "  delete-hf    Delete 8 duplicate HF repos via CLI"
    ;;
esac