Issue 350

gunthercox#350
vkosuri · Oct 30, 2016 · 0e6aaf3 · 0e6aaf3
1 parent 5287520
commit 0e6aaf3
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/.gitignore b/.gitignore
@@ -7,6 +7,7 @@ venv
 *.pyc
 *.swp
 *.egg-info
+*.egg/*
 
 # Database files
 *.sqlite3

diff --git a/chatterbot/utils/word_net.py b/chatterbot/utils/word_net.py
@@ -11,8 +11,15 @@ class Wordnet():
     def __init__(self):
         from nltk.data import find
         from nltk import download
+        import os
 
         try:
+            if os.name == 'nt':
+                nltk.data.path.append(os.path.join(os.getenv('APPDATA',
+                                                            'nltk_data')))
+            else:
+                nltk.data.path.append(os.path.join(os.path.expanduser('~'), 
+                                                            'nltk_data'))
             find('wordnet.zip')
         except LookupError:
             download('wordnet')