import sys

cnet_models = ["canny", "normal", "depth", "openpose", "hed", "scribble", "mlsd", "seg"]
unet_models = ["sd15", "openjourney", "Nitro-Diffusion", "anything-v3.0", "vintedois-diffusion-v0-1"]

print(f"# ControlNet Generation Comparison")
print("")

def print_header():
    print("|seed|", end="")
    for ot in unet_models:
        print(f"{ot}|", end="")
    print("")

    print("|", end="")
    for _ in range(len(unet_models) + 1):
        print("---|", end="")
    print("")

def get_url(path):
    # url = "./all_controlnet_models/"
    url = 'https://huggingface.co/takuma104/controlnet_dev/resolve/main/model_compare_full/all_controlnet_models/'
    return url + path

for cnet in cnet_models:
    print(f"## ControlNet {cnet}")
    print("")
    print_header()
    for seed in range(16):
        print(f"|{seed}|", end="")
        for unet in unet_models:
            url = get_url(path=f"{cnet}/{unet}/output_{seed:02d}.png")
            print(f'<a href="{url}"><img width="128" src="{url}"/></a>|', end="")
        print("")