import os

def get_filenames_from_txt(txt_path):
    filenames = set()
    with open(txt_path, "r") as f:
        for line in f:
            file, ext = os.path.splitext(line.strip())
            filenames.add(file)
    return filenames

def find_difference(txt1_path, txt2_path):
    filenames1 = get_filenames_from_txt(txt1_path)
    filenames2 = get_filenames_from_txt(txt2_path)
    difference1 = filenames1.difference(filenames2)
    difference2 = filenames2.difference(filenames1)
    return difference1, difference2

def write_to_txt(filename, difference):
    with open(filename, "w") as f:
        for item in difference:
            f.write("%s\n" % item)

if __name__ == "__main__":
    txt1_path = "/media/DataDrive/yiling/annotation/XD_test_annotation.txt"
    txt2_path = "/media/DataDrive/yiling/annotation/XD_feature_list.txt"
    difference1, difference2 = find_difference(txt1_path, txt2_path)

    output_txt1 = "/media/DataDrive/yiling/annotation/difference1.txt"
    output_txt2 = "/media/DataDrive/yiling/annotation/difference2.txt"

    write_to_txt(output_txt1, difference1)
    write_to_txt(output_txt2, difference2)

    print("txt1中不同的部分已存储到:", output_txt1)
    print("txt2中不同的部分已存储到:", output_txt2)