#!/usr/bin/env python3

import argparse
import io

#
# command line arguments
#
arg_parser = argparse.ArgumentParser(
    """Creates groundtruth files from text2image generated box files"""
)

# Text ground truth
arg_parser.add_argument(
    '-t',
    '--txt',
    nargs='?',
    metavar='TXT',
    help='Line text (GT)',
    required=True,
)

# Text box file
arg_parser.add_argument(
    '-b',
    '--box',
    nargs='?',
    metavar='BOX',
    help='text2image generated box file (BOX)',
    required=True,
)

args = arg_parser.parse_args()

#
# main
# uses US (ASCII unit separator, U+001F) for substitution to  get the space delimiters
#

gtstring = io.StringIO()
gtfile = open(args.txt, 'w', encoding='utf-8')
with io.open(args.box, 'r', encoding='utf-8') as boxfile:
    print(
        ''.join(
            line.replace('  ', '\u001f ').split(' ', 1)[0]
            for line in boxfile
            if line
        ),
        file=gtstring,
    )
gt = gtstring.getvalue().replace('\u001f', ' ').replace('\t', '\n')
print(gt, file=gtfile)