import argparse
import os

import pandas as pd

# ['ALIGN:align-basecoyo700m', 'CLIPS:CLIPS-Large-14-224recap-datacomp1b', 'CLIPS:CLIPS-Large-14-336recap-datacomp1b', 'CLIPS:CLIPS-Huge-14-224recap-datacomp1b', 'DreamLIP:dreamlip-vitb16cc3m-long', 'DreamLIP:dreamlip-vitb16cc12m-long', 'DreamLIP:dreamlip-vitb16yfcc15m-long', 'DreamLIP:dreamlip-vitb16cc30m-long', 'FSC-CLIP:fsc-clip-ViT-B-32laioncoco-ft', 'FSC-CLIP:fsc-clip-ViT-B-16laioncoco-ft', 'FSC-CLIP:fsc-clip-ViT-L-14laioncoco-ft', 'Jina-CLIP:jina-clip-v1jinaai', 'Jina-CLIP:jina-clip-v2jinaai', 'LoTLIP:LoTLIP-ViT-B-32lotlip100m', 'LoTLIP:LoTLIP-ViT-B-16lotlip100m', 'LongCLIP:longclip-vitb32sharegpt4v-1m', 'LongCLIP:longclip-vitb16sharegpt4v-1m', 'LongCLIP:longclip-vitl14sharegpt4v-1m', 'LongCLIP:longclip-vitl14_336pxsharegpt4v-1m', 'OpenCLIP:roberta-ViT-B-32laion2b_s12b_b32k', 'OpenCLIP:coca_ViT-B-32laion2b_s13b_b90k', 'OpenCLIP:coca_ViT-L-14laion2b_s13b_b90k', 'OpenCLIP:ViT-H-14laion2b_s32b_b79k', 'OpenCLIP:ViT-L-14laion2b_s32b_b82k', 'OpenCLIP:ViT-B-32laion2b_s34b_b79k', 'OpenCLIP:ViT-B-16laion2b_s34b_b88k', 'OpenCLIP:ViT-g-14laion2b_s34b_b88k', 'OpenCLIP:ViT-B-16openai', 'OpenCLIP:ViT-B-32openai', 'OpenCLIP:ViT-L-14openai', 'OpenCLIP:ViT-L-14-336openai', 'OpenCLIP:ViT-B-16-SigLIPwebli', 'OpenCLIP:ViT-B-16-SigLIP-384webli', 'OpenCLIP:ViT-L-16-SigLIP-256webli', 'OpenCLIP:ViT-L-16-SigLIP-384webli', 'OpenCLIP:ViT-SO400M-14-SigLIPwebli', 'Recap-CLIP:ViT-L-16-HTxt-Recap-CLIPrecap-datacomp1b', 'StructuredCLIP:NegCLIP-ViT-B-32coco-ft', 'StructuredCLIP:CE-CLIP-ViT-B-32coco-ft', 'StructuredCLIP:DAC-LLM-ViT-B-32cc3m-ft', 'StructuredCLIP:DAC-SAM-ViT-B-32cc3m-ft']


def parse_args():
    parser = argparse.ArgumentParser()
    parser.add_argument("--csv-file", type=str)
    return parser.parse_args()


def main(args):
    df = pd.read_csv(args.csv_file)
    model_keys = []
    for _, row in df.iterrows():
        model_key = row["family"] + ":" + row["model"] + ":" + row["tag"]
        model_keys.append(model_key)
    import pdb
    pdb.set_trace()


if __name__ == "__main__":
    args = parse_args()
    main(args)