import os

WORKSPACE = "/workspace"
CACHE_DIR = "/workspace/.cache/huggingface"
OUTPUTS_DIR = "/workspace/outputs"
MODELS_DIR = "/workspace/models"

GPU_ALLOCATION = {
    "orchestrator": 7,
    "video_primary": 0,
    "video_backup": 1,
    "face_restore": 2,
    "frame_interpolate": 3,
    "video_upscale": 4,
    "image_generator": 5,
    "audio_generator": 6,
    "quality_evaluator": 6,
}

MODELS = {
    "wan_t2v": "Wan-AI/Wan2.6-T2V-14B",
    "wan_i2v": "Wan-AI/Wan2.6-I2V-14B-720P",
    "ltx": "Lightricks/LTX-2.3",
    "flux": "black-forest-labs/FLUX.1-dev",
    "llama": "meta-llama/Llama-3.3-70B-Instruct",
    "qwen": "Qwen/Qwen2.5-72B-Instruct",
}

DEFAULTS = {
    "fps": 24,
    "cfg_scale": 7.0,
    "num_inference_steps": 30,
    "motion_strength": 5,
    "seed": None,
}

R2_CONFIG = {
    "bucket": os.getenv("R2_BUCKET_NAME", "video-ai-swarm"),
    "public_domain": os.getenv("R2_PUBLIC_DOMAIN", ""),
}