chatbot update

Makefile

@@ -39,6 +39,9 @@ wordtagger: tesseract
 talktochatbot: tesseract
 	cat output/plain.txt | python3 src/textbotconversation.py
 
+n+7: tesseract
+	cat output/plain.txt | python3 src/n_7.py > output/blah.txt
+
 visualization: $(images) $(tmpfile) #requires mplayer
 	@echo $(tmpfile)
 	for i in $(images); do \

src/n_7.py

@@ -0,0 +1,37 @@
+import sys
+from sys import stdin, stdout
+
+
+def seven(text):
+    fpath = open('91K nouns.txt')
+    nouns = fpath.readlines()
+    separated = text.split()    #use nltk tokenize instead
+    #print(separated)
+    new_separated = []
+    for word in separated:
+        word = word.lower() + '\n'
+        if word in nouns:
+            position = nouns.index(word)
+            new_word = nouns[position + 7]
+            #print(" replacing", new_position)
+            new_separated.append(new_word.strip())
+        else:
+            #print("notinlist")
+            #print("adding to new_separated ", word)
+            new_separated.append(word.strip())
+    #print(new_separated)
+    return ' '.join(new_separated)
+
+text =  stdin.read()
+output = seven(text)
+print(output)
+
+#sentence = input('What is your sentence? ')
+#print(seven(sentence))
+
+
+# pytest requires that you name your tests with test_<your-name>
+# run with the 'pytest' command in your terminal
+#def test_seven():
+#    assert seven('Baboons') == 'babushkas'
+#    assert seven('Baboons,') == 'babushkas'