util.py

import os
import sys

def reverse(dict):
    rev_dict = {}
    for k, v in dict.items():
        rev_dict[v] = k
    return rev_dict

def to_str(type, idxs, cont, r_vdict, r_adict, r_exp_vdict):
    if type == 'a':
        return r_adict[idxs]
    elif type == 'q':
        words = []
        for idx in idxs:
                words.append(r_vdict[idx])

        start = 0
        for i, indicator in enumerate(cont):
            if indicator == 1:
                start = i
                break
        start = max(0, start - 1)
        words = words[start:]
    elif type == 'exp':
        words = []
        for idx in idxs:
            if idx == 0:
                break
            words.append(r_exp_vdict[idx])

    return ' '.join(words)

def batch_to_str(type, batch_idx, batch_cont, r_vdict, r_adict, r_exp_vdict):

    converted = []
    for idxs, cont in zip(batch_idx, batch_cont):
        converted.append(to_str(type, idxs, cont, r_vdict, r_adict, r_exp_vdict))
    return converted