import os


CIVITAI_API_KEY = os.environ.get("CIVITAI_API_KEY")
HF_TOKEN = os.environ.get("HF_TOKEN")
HF_READ_TOKEN = os.environ.get('HF_READ_TOKEN') # only use for private repo


num_loras = 3
num_cns = 2


models_dev = [
    "camenduru/FLUX.1-dev-diffusers",
    "sayakpaul/FLUX.1-merged",
    "ostris/OpenFLUX.1",
    "multimodalart/FLUX.1-dev2pro-full",
    "John6666/flux1-dev-minus-v1-fp8-flux",
    "John6666/hyper-flux1-dev-fp8-flux",
    "jimmycarter/LibreFLUX",
    "jimmycarter/LibreFlux-SimpleTuner",
    "shuttleai/shuttle-3-diffusion",
    "shuttleai/shuttle-4-diffusion",
    "advokat/AnimePro-FLUX",
    "John6666/blue-pencil-flux1-v021-fp8-flux",
    "Raelina/Raemu-Flux",
    "John6666/raemu-flux-v10-fp8-flux",
    "John6666/copycat-flux-test-fp8-v11-fp8-flux",
    "John6666/wai-ani-flux-v10forfp8-fp8-flux",
    "John6666/flux-dev8-anime-nsfw-fp8-flux",
    "John6666/nepotism-fuxdevschnell-v3aio-fp8-flux",
    "John6666/sumeshi-flux1s-v002e-fp8-flux",
    "John6666/fca-style-v33-x10-8step-fp8-flux",
    "John6666/lyh-anime-v10f1-fp8-flux",
    "John6666/lyh-dalle-anime-v12dalle-fp8-flux",
    "John6666/lyh-anime-flux-v2a1-fp8-flux",
    "John6666/glimmerkin-flux-cute-v10-fp8-flux",
    "John6666/niji-style-flux-devfp8-fp8-flux",
    "John6666/niji56-style-v3-fp8-flux",
    "John6666/xe-anime-flux-v04-fp8-flux",
    "John6666/xe-figure-flux-01-fp8-flux",
    "John6666/xe-pixel-flux-01-fp8-flux",
    "John6666/xe-guoman-flux-02-fp8-flux",
    "terminusresearch/flux-booru-v0.2",
    "John6666/carnival-unchained-v10-fp8-flux",
    "John6666/real-flux-10b-schnell-fp8-flux",
    "John6666/fluxunchained-artfulnsfw-fut516xfp8e4m3fnv11-fp8-flux",
    "John6666/fastflux-unchained-t5f16-fp8-flux",
    "John6666/iniverse-mix-sfwnsfw-f1drealnsfwguofengv2-fp8-flux",
    "John6666/nsfw-master-flux-lora-merged-with-flux1-dev-fp16-v10-fp8-flux",
    "John6666/the-araminta-flux1a1-fp8-flux",
    "John6666/acorn-is-spinning-flux-v11-fp8-flux",
    "John6666/acorn-is-spinning-flux-aisfluxdedistilled-fp8-flux",
    "John6666/stoiqo-afrodite-fluxxl-f1dalpha-fp8-flux",
    "John6666/real-horny-pro-fp8-flux",
    "John6666/centerfold-flux-v20fp8e5m2-fp8-flux",
    "John6666/jib-mix-flux-v5itsalive-fp8-flux",
    "John6666/sapianf-nude-men-women-for-flux-v20fp16-fp8-flux",
    "John6666/uncensored-females-flux-fluxdevufv7fp16-fp8-flux",
    "John6666/flux-asian-realistic-v10-fp8-flux",
    "John6666/fluxasiandoll-v10-fp8-flux",
    "John6666/xe-asian-flux-01-fp8-flux",
    "John6666/fluxescore-dev-v10fp16-fp8-flux",
    "John6666/2758-flux-asian-utopian-v30fp8noclip-fp8-flux",
    "trongg/FLUX_dev2pro_nsfw_context_lora",
    "https://huggingface.co/StableDiffusionVN/SDVN11-Ghibli-Flux/blob/main/SDVN11-Ghibli-Flux_fp8-hyper.safetensors",
    "https://huggingface.co/datasets/John6666/flux1-backup-202410/blob/main/iniverseMixXLSFWNSFW_f1dFP16V10.safetensors",
    "https://huggingface.co/city96/FLUX.1-dev-gguf/blob/main/flux1-dev-Q2_K.gguf",
    # "",
]

models_schnell = [
    "black-forest-labs/FLUX.1-schnell",
]

models_fill = ["fuliucansheng/FLUX.1-Fill-dev-diffusers"]

models_dedistill = []

models_canny = ["sayakpaul/FLUX.1-Canny-dev-nf4"]

models_depth = ["sayakpaul/FLUX.1-Depth-dev-nf4"]

models = models_dev + models_schnell + models_fill

model_trigger = {
    "Raelina/Raemu-Flux": "anime",
    "John6666/raemu-flux-v10-fp8-flux": "anime",
    "John6666/fca-style-v33-x10-8step-fp8-flux": "fca_style",
}

single_file_base_models = {
    "dev": "camenduru/FLUX.1-dev-diffusers",
    "schnell": "black-forest-labs/FLUX.1-schnell",
    "fill": "fuliucansheng/FLUX.1-Fill-dev-diffusers",
}

# List all Models for specified user
HF_MODEL_USER_LIKES = [] # sorted by number of likes
HF_MODEL_USER_EX = [] # sorted by a special rule

# - **Download Models**
DOWNLOAD_MODEL_LIST = [
]

# - **Download VAEs**
DOWNLOAD_VAE_LIST = [
]

# - **Download LoRAs**
DOWNLOAD_LORA_LIST = [
]

DIFFUSERS_FORMAT_LORAS = []

DIRECTORY_MODELS = 'models'
os.makedirs(DIRECTORY_MODELS, exist_ok=True)
DIRECTORY_LORAS = 'loras'
os.makedirs(DIRECTORY_LORAS, exist_ok=True)
DIRECTORY_VAES = 'vaes'
os.makedirs(DIRECTORY_VAES, exist_ok=True)


HF_LORA_PRIVATE_REPOS1 = []
HF_LORA_PRIVATE_REPOS2 = [] # to be sorted as 1 repo
HF_LORA_PRIVATE_REPOS = HF_LORA_PRIVATE_REPOS1 + HF_LORA_PRIVATE_REPOS2
HF_LORA_ESSENTIAL_PRIVATE_REPO = '' # to be downloaded on run app
HF_VAE_PRIVATE_REPO = ''