version: '3.8'

services:
  ai-image-gen:
    build: .
    ports:
      - "7860:7860"
    volumes:
      - ./models:/app/models
    environment:
      - PUBLIC_API_KEY=${PUBLIC_API_KEY:-demo-key-123456}
      - HF_TOKEN=${HF_TOKEN:-}
      - SPACE_URL=${SPACE_URL:-http://localhost:7860}
    restart: unless-stopped