version: "3.9"

services:
  backend:
    build:
      context: ./backend
      dockerfile: Dockerfile
    container_name: adfd-backend
    restart: unless-stopped
    ports:
      - "8000:8000"
    environment:
      - PYTHONUNBUFFERED=1
      - HF_HOME=/app/.cache/huggingface
      - TRANSFORMERS_CACHE=/app/.cache/huggingface
      - DEMO_MODE=${DEMO_MODE:-false}
      - ENABLE_HEAVY_MODELS=${ENABLE_HEAVY_MODELS:-true}
      - ALLOWED_ORIGINS=http://localhost:5173,http://localhost:3000,http://frontend:3000
    volumes:
      - hf_cache:/app/.cache/huggingface
      - ./backend/checkpoints:/app/checkpoints
      - ./data/sample_audios:/app/data/sample_audios:ro
    # Uncomment for NVIDIA GPU support:
    # deploy:
    #   resources:
    #     reservations:
    #       devices:
    #         - driver: nvidia
    #           count: 1
    #           capabilities: [gpu]

  frontend:
    build:
      context: ./frontend
      dockerfile: Dockerfile
    container_name: adfd-frontend
    restart: unless-stopped
    ports:
      - "3000:3000"
    environment:
      - VITE_API_BASE_URL=http://localhost:8000
      - VITE_WS_BASE_URL=ws://localhost:8000
    depends_on:
      - backend

volumes:
  hf_cache:
    driver: local