# RunPod Template Configuration
# Use this as reference when creating a RunPod template

name: AudioCraft Studio
description: AI-powered music and sound generation using Meta's AudioCraft

# Container settings
container:
  image: your-dockerhub-username/audiocraft-studio:latest

  # Or build from GitHub
  # dockerfile: Dockerfile
  # context: https://github.com/your-username/audiocraft-ui.git

# GPU requirements
gpu:
  type: RTX 4090  # Recommended: RTX 4090, RTX 3090, A100
  count: 1
  minVram: 24  # GB

# Resource limits
resources:
  cpu: 8
  memory: 32  # GB
  disk: 100   # GB (for model cache and outputs)

# Port mappings
ports:
  - name: Gradio UI
    internal: 7860
    external: 7860
    protocol: http
  - name: REST API
    internal: 8000
    external: 8000
    protocol: http

# Volume mounts
volumes:
  - name: outputs
    mountPath: /workspace/outputs
    size: 50  # GB
  - name: models
    mountPath: /workspace/models
    size: 30  # GB (model cache)
  - name: data
    mountPath: /workspace/data
    size: 10  # GB

# Environment variables
env:
  - name: AUDIOCRAFT_HOST
    value: "0.0.0.0"
  - name: AUDIOCRAFT_GRADIO_PORT
    value: "7860"
  - name: AUDIOCRAFT_API_PORT
    value: "8000"
  - name: AUDIOCRAFT_COMFYUI_RESERVE_GB
    value: "10"  # Reserve VRAM for ComfyUI if running alongside
  - name: AUDIOCRAFT_MAX_LOADED_MODELS
    value: "2"
  - name: AUDIOCRAFT_IDLE_UNLOAD_MINUTES
    value: "15"
  - name: HF_HOME
    value: "/workspace/models/huggingface"

# Startup command
command: ["python", "main.py"]

# Health check
healthCheck:
  path: /
  port: 7860
  initialDelaySeconds: 120
  periodSeconds: 30
  timeoutSeconds: 10
  failureThreshold: 3