diff --git a/markov.py b/markov.py index c08133c..7db2f3a 100755 --- a/markov.py +++ b/markov.py @@ -19,7 +19,7 @@ def sanitizeText(text): if text[0] == "@" or text[1] == "@": if split[1][0] not in string.ascii_lowercase: return sanitizeText(" ".join(split[1:])) - if split[-1][0] == "@" and text[-1] in string.ascii_lowercase: + if split[-1][0] == "@" and text[-1] in (string.ascii_letters + "_"): return sanitizeText(" ".join(split[:-1])) if text[:4] == "RT @": return sanitizeText(text.partition(":")[2])