audio_root = {
    'preferencestrict': 'dataset/audio',
    'preferencestrictreverse': 'dataset/audio',
}

video_root = {
    'preferencestrict': 'dataset/video',
    'preferencestrictreverse': 'dataset/video',
}

label_csv = {
    'preferencestrict': 'dataset/preference_threestrict.csv',
    'preferencestrictreverse': 'dataset/preference_threestrict_reverse.csv',
}

label_withname_csv = {
    'preferencestrict': 'dataset/preference_threestrict_with_modelnames.csv',
    
}

#############################
# 仅使用于 reverse 测试的情况下
reverse_mapping = {
    'a1': 'a2',
    'a2': 'a1',
    'same': 'same',
}
#############################

## model -> model path
model2path = {
    'gemini_15_pro': 'models/gemini-1.5-pro-latest',
    'gemini_15_flash': 'models/gemini-1.5-flash-latest',
    'gemini_25_flash': 'models/gemini-2.5-flash-preview-05-20',
    'gemini_20_flash': 'models/gemini-2.0-flash',
    'gemini_20_flash_lite': 'models/gemini-2.0-flash-lite',

    'gpt_41': 'gpt-4.1',
    'gpt_41_mini': 'gpt-4.1-mini',
    'gpt_4o': 'gpt-4o',

    'qwen3_8b': "./Qwen3-8B",
    'qwen3_14b': "./Qwen3-14B",
    'qwen3_32b': "./Qwen3-32B", # 单卡A100无法运行，对于机器要求比较高
    'qwen25': "Qwen2-main/Qwen2.5-7B-Instruct",

    'qwen25vl_3b': "Qwen-main/Qwen2.5-VL-3B-Instruct",
    'qwen25vl_7b': "Qwen-main/Qwen2.5-VL-7B-Instruct",
    'qwen25vl_32b': "./Qwen2.5-VL-32B-Instruct",

    'qwen25omni_3b': "./Qwen2.5-Omni-3B",
    'qwen25omni_7b': "Qwen-main/Qwen2.5-Omni-7B",

    'chatunivi_7b':    'Chat-UniVi-main/weights/Chat-UniVi',
    'chatunivi_7b_15': './Chat-UniVi-7B-v1.5',

    'qwen2audio': './Qwen2-Audio-7B-Instruct',

    'llavanextvideo_7b':     './LLaVA-NeXT-Video-7B-hf',
    'llavanextvideo_7b_dpo': './LLaVA-NeXT-Video-7B-DPO-hf',

    'vita_15': './VITA-1.5',

    'vila_15_7b': './VILA1.5-7b',
    'vila_15_13b': './VILA1.5-13b',

    'pllava_7b':  './pllava-7b',
    'pllava_13b': './pllava-13b',
}

## model -> default input type
model2input = {
    'gemini_15_pro': 'audiovideo',
    'gemini_15_flash': 'audiovideo',
    'gemini_25_flash': 'audiovideo',
    'gemini_20_flash': 'audiovideo',
    'gemini_20_flash_lite': 'audiovideo',

    'gpt_41': 'video',
    'gpt_41_mini': 'video',
    'gpt_4o': 'video',

    'videollava': 'video',
    'videochatgpt': 'video',
    'chatunivi_7b': 'video',
    'chatunivi_7b_15': 'video',
    'mplugowl': 'video',
    'otter': 'video',
    'llamavid': 'video',
    'videochat': 'video',
    'videochat2': 'video',

    'qwenaudio': 'audio',
    'salmonn': 'audio',

    'qwen25vl_3b': 'video',
    'qwen25vl_7b': 'video',
    'qwen25vl_32b': 'video',

    'qwen2audio': 'audio',

    'llavanextvideo_7b':     'video',
    'llavanextvideo_7b_dpo': 'video',

    'vita_15': 'video',

    'vila_15_7b': 'video',
    'vila_15_13b': 'video',

    'pllava_7b':  'video',
    'pllava_13b': 'video',
}
