import os

def add_dir(txt_path, output_path, parent_dir):
    with open(txt_path, "r") as file:
        lines = file.readlines()

    updated_lines = []
    for line in lines:
        # 拆分每行的数据
        columns = line.strip().split("\t")  # 假设列之间用制表符分隔，你可以根据实际情况修改分隔符
        if columns:  # 检查是否有数据
            # 更新第一列
            columns[0] = os.path.join(parent_dir, columns[0])
            # 重新组合每行数据
            updated_line = "\t".join(columns) + "\n"  # 同样，使用相同的分隔符重新组合数据
            updated_lines.append(updated_line)

    with open(output_path, "w") as file:
        file.writelines(updated_lines)

if __name__ == "__main__":
    txt_path = "/media/DataDrive/yiling/annotation/XD_train_annotation_file.txt"
    output_path = "/media/DataDrive/yiling/annotation/XD_train_annotation.txt"

    add_dir(txt_path, output_path, 'train')