import sys

cnet_models = ["canny", "normal", "depth", "openpose", "hed", "scribble", "mlsd", "seg"]
processors = ["ref", "diffusers"]
image_types = ['bird', 'human', 'room', 'vermeer']

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

def print_header():
    print("|seed|", end="")
    for proc in processors:
        print(f"{proc} with prompt|", end="")
    for proc in processors:
        print(f"{proc} guess_mode|", end="")
    for proc in ['diffusers']:
        print(f"{proc} guess_mode with prompt|", end="")
    print("")

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

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

for cnet in cnet_models:
    print(f"## ControlNet {cnet}")
    print("")
    print_header()

    for image_type in image_types:    
        for seed in range(4):
            if cnet=='openpose' and image_type=='human' and seed == 3:
                continue

            print(f"|{seed}|", end="")
            for proc in processors:
                url = get_url(path=f"{proc}/output_{image_type}_{cnet}_{seed:01d}.png")
                print(f'<a href="{url}"><img width="128" src="{url}"/></a>|', end="")

            for proc in processors:
                url = get_url(path=f"{proc}/output_{image_type}_{cnet}_{seed:01d}_gm.png")
                print(f'<a href="{url}"><img width="128" src="{url}"/></a>|', end="")

            for proc in ['diffusers']:
                url = get_url(path=f"{proc}/output_{image_type}_{cnet}_{seed:01d}_gm_wp.png")
                print(f'<a href="{url}"><img width="128" src="{url}"/></a>|', end="")
            print("")