Textalk.py

from word2number import w2n
from recorder import calculate
import csv
words_numeric = set()
words_substituted = {}

# Load in data from txt files
with open('./data/numeric_words.csv', 'r') as file:
    for row in csv.reader(file):
        words_numeric.add(row[0])

with open('./data/substituted_words.csv', 'r') as file:
    for row in csv.reader(file):
        if len(row) == 2:
            words_substituted[row[0]] = row[1]

def parsestr(s):
    # for key in d.keys():
    #     if key in s:
    #         if type(d[key]) == str:
    #             s = s.replace(key, d[key])
    #         else:
    #             s = s.replace(key, d[key](s))
    # return s
    calculate(s)
    return parsestr_numeric(parsestr_substitute(s))

def parsestr_numeric(s):
    words = s.split()
    if not words: return ''
    i = 0
    while i < len(words) and words[i] in words_numeric:
        i = i + 1
    # print('PARSING:', ' '.join(words[:i]))
    if i == 0:
        return words[0] + ' ' + parsestr_numeric(' '.join(words[1:]))
    s = str(w2n.word_to_num(' '.join(words[:i]))) + ' ' + parsestr_numeric(' '.join(words[i:]))
    return s

def parsestr_substitute(s):
    words = s.split()
    for i in range(len(words)):
        if words[i] in words_substituted:
            words[i] = words_substituted[words[i]]
    return ' '.join(words)

# Returns a lst of indices containing <quantity and all>
def quantity(s):
    lst = []

    def helper(s, index_lst):
        if 'quantity' in s:
            # Remember to add index by 1 more for end index because second l in all will not be included \
            # in range() if not inclusive
            start_index, end_index = s.find('quantity'), s.find('all') + 2
            index_lst.append([start_index, end_index])
            return helper(s[end_index:], index_lst)
        return lst
    return helper(s, lst)


# Returns a list of the string segments containing <quantity and all>
def quantity_checker(s, lst_of_indices):
    str_lst = []

    def checker(s, lst):
        if lst:
            start_index, end_index = lst[0][0], lst[0][1]
            str_lst.append(s[start_index:end_index + 1])
            lst.pop(0)
            return checker(s[end_index:], lst)
        return str_lst
    return checker(s, lst_of_indices)

# Check before printing final latex string
def check(latex):
    for word in l_d.keys():
        if word in latex:
            latex = latex.replace(word, l_d[word])
    return latex

# print(parsestr('Your total is four thousand two hundred ninety one dollars sixty seven cents and one hundred seventeen unborn fetuses'))
# print(parsestr('one plus two minus three times four equals zero'))
# print(parsestr('I just did three hundred and sixty five epsilon delta proofs'))
print(parsestr('The integral from zero to x times 5 of x plus x to the power of two'))