import os
folder_json = "/Users/tuvn18/Desktop/tuvn18/dev/KIAI/dev/trace/40_page_70_110925"

for name in os.listdir(folder_json):
    if name.endswith('.png'):
        new_name = name.split('-')[0] + '.png'
        os.rename(os.path.join(folder_json, name), os.path.join(folder_json, new_name))
    elif name.endswith('.json'):
        new_name = name.split('-')[1] + '.json'
        os.rename(os.path.join(folder_json, name), os.path.join(folder_json, new_name))