services:
  train:
    build: .
    image: cerebellum:latest
    runtime: nvidia
    environment:
      - NVIDIA_VISIBLE_DEVICES=all
      - NVIDIA_DRIVER_CAPABILITIES=compute,utility
      - HF_TOKEN=${HF_TOKEN}
    volumes:
      # Mount entire project — no files baked into image, code changes instant
      - .:/workspace
    command: >
      python -u scripts/train_a11y.py
      --compile
    tty: true
    stdin_open: true

  androidworld:
    build: .
    image: cerebellum:latest
    runtime: nvidia
    privileged: true
    devices:
      - /dev/kvm:/dev/kvm
    ports:
      - "5000:5000"
      - "5554:5554"
      - "5555:5555"
      - "8554:8554"
    environment:
      - NVIDIA_VISIBLE_DEVICES=all
      - NVIDIA_DRIVER_CAPABILITIES=compute,utility,graphics
      - HF_TOKEN=${HF_TOKEN}
      - ANDROID_WORLD_EMULATOR_SETUP=0
      - EMULATOR_TIMEOUT=300
      - HW_ACCEL_OVERRIDE=-accel on
    volumes:
      - .:/workspace
    working_dir: /opt/android_world
    command: >
      bash -lc "./docker_setup/start_emu_headless.sh &&
      adb root &&
      python3 -m server.android_server"
    tty: true
    stdin_open: true

  shell:
    build: .
    image: cerebellum:latest
    runtime: nvidia
    privileged: true
    devices:
      - /dev/kvm:/dev/kvm
    environment:
      - NVIDIA_VISIBLE_DEVICES=all
      - NVIDIA_DRIVER_CAPABILITIES=compute,utility,graphics
      - HF_TOKEN=${HF_TOKEN}
    volumes:
      - .:/workspace
    command: bash
    tty: true
    stdin_open: true