version: "3.9"

services:
  flagsft:
    build: .
    image: flagsft:latest
    environment:
      - OPENAI_API_KEY=${OPENAI_API_KEY:-}
      - FLAGSFT_BASE_URL=${FLAGSFT_BASE_URL:-https://api.openai.com/v1}
      - FLAGSFT_GENERATOR_MODEL=${FLAGSFT_GENERATOR_MODEL:-gpt-4o-mini}
      - FLAGSFT_CRITIC_MODELS=${FLAGSFT_CRITIC_MODELS:-gpt-4o-mini,gpt-4o-mini,gpt-4o-mini}
      - FLAGSFT_VERIFIER_MODEL=${FLAGSFT_VERIFIER_MODEL:-gpt-4o-mini}
      - FLAGSFT_MOCK=${FLAGSFT_MOCK:-0}
    volumes:
      - ./sample_sources:/data:ro
      - ./runs:/runs
    command: ["annotate", "/data", "-o", "/runs/compose_demo"]

  flagsft-ui:
    build: .
    image: flagsft:latest
    entrypoint: ["streamlit", "run", "flagsft/ui/app.py", "--server.port=8501", "--server.address=0.0.0.0", "--"]
    command: ["--run-dir", "/runs/compose_demo"]
    ports:
      - "8501:8501"
    volumes:
      - ./runs:/runs:ro
    depends_on:
      - flagsft