speechemotiondetection / replace_explitives.py
miracle01's picture
Upload 21 files
d51980e
import regex as re
import nltk
def load_words_from_file(file_path):
with open(file_path, "r", encoding="utf-8") as f:
words = [line.strip() for line in f.readlines()]
return words
def sub_explitives(textfile, selection):
replacetext = "person"
# Load target words from text files
b_word_list = load_words_from_file("b_word.txt")
n_word_list = load_words_from_file("n_word.txt")
expletives_list = load_words_from_file("expletives.txt")
# text = word_tokenize(textfile)
# print(text)
# sentences = sent_tokenize(textfile)
if selection == "B-Word":
target_word = b_word_list
elif selection == "N-Word":
target_word = n_word_list
elif selection == "All Explitives":
target_word = expletives_list
else:
target_word = []
print("selection:", selection, "target_word:", target_word)
lines = textfile.split('\n')
if target_word:
print("target word was found, ", target_word)
print(textfile)
for i, line in enumerate(lines):
for word in target_word:
pattern = r"\b" + re.escape(word) + r"\b"
# textfile = re.sub(target_word, replacetext, textfile, flags=re.IGNORECASE)
lines[i] = re.sub(pattern, replacetext, lines[i], flags=re.IGNORECASE)
textfile = '\n'.join(lines)
return textfile