# ============================================================
# AIGC Telegram Bot - Docker Compose
# ============================================================

version: '3.8'

services:
  aigc-bot:
    build: .
    container_name: aigc-telegram-bot
    restart: unless-stopped
    env_file:
      - .env
    volumes:
      # Persist models and outputs
      - ./data/models:/app/data/models
      - ./data/output:/app/data/output
      - ./assets/mdxnet_models:/app/assets/mdxnet_models
      # Mount the original AIGC pipeline source
      - ./AIGC-Telegram/src:/app/AIGC-Telegram/src
    environment:
      - PYTHONUNBUFFERED=1
    # GPU support (uncomment if available)
    # deploy:
    #   resources:
    #     reservations:
    #       devices:
    #         - capabilities: [gpu]