Write the labels to separate files

[DesBw]

I was trying to use the labels as ground-truth texts in tesseract. Currently, trdg writes the labels into a single lable.txt file. I understand that the part of the script that writes them to single file is the following.

if args.name_format == 2:
      # Create file with filename-to-label connections
      with open(
          os.path.join(args.output_dir, "labels.txt"), "w", encoding="utf8"
      ) as f:
          for i in range(string_count):
              file_name = str(i) + "." + args.extension
              label = strings[i]
              if args.space_width == 0:
                  label = label.replace(" ", "")
              f.write("{} {}\n".format(file_name, label))

Can someone with sufficient knowledge of python can help me to modify it so that the labels will be written as separate files?

What I want is:

0.gt.txt 1.gt.txt Each of those files would contain their respective labels inside.