import torch

class Config:
    # Hardware
    DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
    DTYPE = torch.float16 if DEVICE == "cuda" else torch.float32

    # Base Model & LoRA (from primerz/pixagram)
    REPO_ID = "primerz/pixagram"
    CHECKPOINT_FILENAME = "horizon.safetensors"
    LORA_FILENAME = "retroart.safetensors"
    LORA_STRENGTH = 1.0
    
    # Trigger Words for the LoRA
    STYLE_TRIGGER = "(artwork:0.5), (illustration:0.5), p1x3l4rt, pixel art"
    
    # Default Negative Prompt
    DEFAULT_NEGATIVE_PROMPT = "Ugly, artifacts, blurry, disformed, photo-realistic, photo, photography, realistic, low-quality, pixel art, text."

    # Gradio Parameters
    CGF_SCALE = 1.2           
    STEPS_NUMBER = 8          # Reduced slightly for optimal LCM T2I performance
    
    # Aspect Ratios (width, height) - all fitting within ~1MP resolution
    ASPECT_RATIOS = {
        "16:9": (1344, 768),
        "4:3": (1152, 896),
        "1:1": (1024, 1024),
        "3:4": (896, 1152),
        "9:16": (768, 1344)
    }
    DEFAULT_ASPECT_RATIO = "1:1"