|
|
""" |
|
|
formats output text file from llampia |
|
|
into markdown tables |
|
|
""" |
|
|
|
|
|
import os |
|
|
|
|
|
def format_edu(turn_str): |
|
|
no = turn_str.split('<')[0].strip() |
|
|
text = turn_str.split('>')[1].strip() |
|
|
speaker = 'BUIL' |
|
|
if 'Arch' in turn_str.split('>')[0]: |
|
|
speaker = 'ARCH' |
|
|
edu_str = ' ' + '**' + no + ' ' + speaker + '**' + ' ' + text |
|
|
return edu_str |
|
|
|
|
|
current_folder=os.getcwd() |
|
|
|
|
|
output_path = '/path/to/test-output-file.txt' |
|
|
table_path = current_folder + '/test_predictions.md' |
|
|
|
|
|
with open(output_path, 'r') as txt: |
|
|
text = txt.read().split('\n') |
|
|
|
|
|
group = 0 |
|
|
pairs = [] |
|
|
pair = [] |
|
|
for t in text: |
|
|
if t.startswith('New Turn'): |
|
|
group = 1 |
|
|
pair.append(t) |
|
|
elif t.startswith(' ### DS:'): |
|
|
pair.append(t) |
|
|
pairs.append(pair) |
|
|
pair = [] |
|
|
group = 0 |
|
|
elif group == 1: |
|
|
pair.append(t) |
|
|
|
|
|
tables = [] |
|
|
rows = [] |
|
|
dial_str = '' |
|
|
struct_str = '' |
|
|
for pair in pairs: |
|
|
|
|
|
if pair[0].startswith('New Turn: 1 <'): |
|
|
if len(rows) > 0: |
|
|
tables.append(rows) |
|
|
rows = [] |
|
|
dial_str = '**0 BUIL** Mission has started.' |
|
|
dial_str += format_edu(pair[0].lstrip('New Turn: ')) |
|
|
for p in pair[1:]: |
|
|
if p.startswith(' ###'): |
|
|
struct_str = p.lstrip(' ### DS:') |
|
|
rows.append([dial_str, struct_str]) |
|
|
|
|
|
|
|
|
else: |
|
|
dial_str += format_edu(p) |
|
|
elif pair[0].startswith('New Turn: '): |
|
|
dial_str = format_edu(pair[0].lstrip('New Turn: ')) |
|
|
|
|
|
for p in pair[1:]: |
|
|
if p.startswith(' ###'): |
|
|
struct_str = p.lstrip(' ### DS:') |
|
|
|
|
|
|
|
|
rows.append([dial_str, struct_str]) |
|
|
|
|
|
|
|
|
else: |
|
|
dial_str += format_edu(p) |
|
|
|
|
|
all_md_tables = [] |
|
|
|
|
|
for table in tables: |
|
|
table_rows = ['| Dialogue | Structure |', '| ----- | ----- |'] |
|
|
for tr in table: |
|
|
st = '| ' + tr[0] + ' | ' + tr[1] + ' |' |
|
|
table_rows.append(st) |
|
|
all_md_tables.extend(table_rows) |
|
|
all_md_tables.extend(' ') |
|
|
|
|
|
md_tables_string = '\n'.join(all_md_tables) |
|
|
|
|
|
|
|
|
f = open(table_path, 'w') |
|
|
for r in all_md_tables: |
|
|
print(r) |
|
|
print(r, file=f) |
|
|
print("markdown printed") |
|
|
|
|
|
|
