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 = "picto.safetensors"
    LORA_FILENAME = "retroart.safetensors"
    LORA_STRENGTH = 1.0 # Fixed strength for fusion
    
    # Trigger Words for the LoRA
    STYLE_TRIGGER = "HD pixel art artwork and high quality illustration in retroart style of "
    
    # Default Negative Prompt
    DEFAULT_NEGATIVE_PROMPT = "Ugly, artifacts, blurry, disformed, photo-realistic, photo, photography, realistic, low-quality, text."

    # InstantID Assets
    INSTANTID_REPO = "InstantX/InstantID"
    
    # ControlNet Repos
    CN_ZOE_REPO = "diffusers/controlnet-zoE-depth-sdxl-1.0"
    CN_LINEART_REPO = "ShermanG/ControlNet-Standard-Lineart-for-SDXL"
    
    # Preprocessor (Annotator) Repo
    ANNOTATOR_REPO = "lllyasviel/Annotators"
    
    # Captioning Model
    CAPTIONER_REPO = "Salesforce/blip-image-captioning-base"
    
    # InsightFace Model (HF Hub mirror)
    ANTELOPEV2_REPO = "DIAMONIK7777/antelopev2"
    ANTELOPEV2_ROOT = "." # Parent folder
    ANTELOPEV2_NAME = "antelopev2"
    
    # Gradio Parameters
    CGF_SCALE = 1.2
    STEPS_NUMBER = 10
    IMG_STRENGTH = 0.65
    FACE_STRENGTH = 0.75
    DEPTH_STRENGTH = 0.75
    EDGE_STRENGTH = 0.75
    CLIP_SKIP = 1