syllableizer/syllableizer.py at master · jacobtye/syllableizer · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
#!/usr/bin/python
import inflect
import sys
import os


class Syllableizer:
    '''
    Class which takes in a string and then will return a list of the string tokenized into its phonetic syllables
    '''

    def __init__(self, dictPath):
        self.p = inflect.engine()
        self.d = {}
        self.words_list = []

        with open(dictPath) as f:
            for line in f:
                (key, val) = line.split("\t", 1)
                val = val.replace("\n", "")
                self.d[key.lower()] = val

    def isfloat(self, value):
        try:
            float(value)
            return True
        except ValueError:
            return False

    def split_words(self, usr_input):
        words = usr_input
        # separate punctiation to separate all words for tokenizing
        words = words.replace(".", " .PERIOD ")
        words = words.replace("!", " !EXCLAMATION-POINT ")
        words = words.replace("?", " ?QUESTION-MARK ")
        words = words.replace(",", " ,COMMA ")
        words = words.replace("-", " -HYPHEN ")
        words = words.replace(":", " :COLON ")
        words = words.replace(";", " ;SEMI-COLON ")
        words = words.replace("\"", " \"QUOTE ")
        words = words.replace(")", " )CLOSE-PARENTHESES ")
        words = words.replace("(", " (OPEN-PARENTHESES ")
        words = words.replace("\\", " \\ ")
        words = words.replace("/", " /SLASH ")
        words = words.replace("%", " %PERCENT ")
        words = words.replace("[", " [ ")
        words = words.replace("]", " ] ")
        words = words.split()
        for a in words:
            try:
                if a.isdigit():
                    # print ("true", a)
                    w = self.p.number_to_words(a)
                    # print w
                    self.split_words(w)

                elif self.isfloat(a):
                    (w, dec) = a.split(".", 1)
                    w = self.p.number_to_words(a)
                    # print w
                    self.split_words(w)
                else:
                    # print a
                    self.words_list.append(self.d[a.lower()])
            except:
                print "Errors, see error.txt"
                f = open("error.txt", "a+")
                f.write("Failed Word: %s\n" % a)
                f.close()

    def getList(self):
        returnList = self.words_list
        self.words_list = []
        return returnList


def main(args):
    parser = Syllableizer('cmudict-modified.txt')
    usr_input = "none"
    while (usr_input != "quit"):
        usr_input = raw_input('enter word to break apart\n')
        parser.split_words(usr_input)
        print parser.getList()


if __name__ == '__main__':
    main(sys.argv)