fix_numbering.py

import sys
import pyconll

corpus = pyconll.load_from_file(sys.argv[1])
with open(sys.argv[2],'w') as fixed:
    for sentence in corpus:
        print(sentence.id)
        if sentence.meta_present('newdoc'):
            print(sentence.meta_value("newdoc"))
            fixed.write('# newdoc = %s\n' % sentence.meta_value('newdoc'))
        if sentence.meta_present('comment'):
            fixed.write('# comment = %s\n' % sentence.meta_value('comment'))
        fixed.write('# sent_id = %s\n' % sentence.id)
        if sentence.meta_present('speaker'):
            fixed.write('# speaker = %s\n' % sentence.meta_value('speaker'))
        fixed.write('# text = %s\n' % sentence.text.replace('_', ' '))
        offset = 0
        skip = False
        mapping = {}
        new_tokens = {}
        i = 0
        for token in sentence:
            if '-' not in token.id: # fix this
                i+=1
                if int(token.id) != i:
                    mapping[token.id] = str(i)
                    token.id = str(i)
            else:
                head = int(token.id.split('-')[0])
                tail = int(token.id.split('-')[1])
                token.id = f"{str(i + 1)}-{str(i + 1 + tail - head)}"
        for token in sentence:
            if token.head in mapping:
                token.head = mapping[token.head]
            fixed.write("%s\n" % token.conll())
        fixed.write('\n')