sep the keywords out

3 files changed, 10 insertions, 3 deletions

diff --git a/ratspub.png b/ratspub.png
new file mode 100644
index 0000000..345e40a
--- /dev/null
+++ b/ratspub.png
diff --git a/ratspub.py b/ratspub.py
index 64e750a..0cc5d8a 100755
--- a/ratspub.py
+++ b/ratspub.py
@@ -2,11 +2,10 @@
 from nltk.tokenize import sent_tokenize
 import os
 import re
-import ratspub_keywords
+from ratspub_keywords import *
 
 global function_d, brain_d, drug_d, addiction_d, brain_query_term, pubmed_path
 
-pubmed_path="/media/hao/2d554499-6c5b-462d-85f3-5c49b25f4ac8/PubMed/Archive"
 
 ## turn dictionary (synonyms) to regular expression
 def undic(dic):
@@ -79,3 +78,11 @@ n1=generate_nodes(addiction_d, 'addiction')
 n2=generate_nodes(drug_d, 'drug')
 n3=generate_nodes(brain_d, 'brain')
 default_nodes=n0+n1+n2+n3
+
+
+host= os.popen('hostname').read().strip()
+if host=="x1":
+    pubmed_path="/run/media/hao/PubMed/Archive/"
+elif host=="hchen3":
+    pubmed_path="/media/hao/2d554499-6c5b-462d-85f3-5c49b25f4ac8/PubMed/Archive"
+
diff --git a/ratspub_keywords.py b/ratspub_keywords.py
index 6d63174..baeb5a0 100644
--- a/ratspub_keywords.py
+++ b/ratspub_keywords.py
@@ -1,7 +1,7 @@
 addiction_d = {"reward":"reward|hedonic|incentive|intracranial self stimulation|ICSS|reinforcement|reinforcing|conditioned place preference|CPP|self administration|self administered|drug reinforced|operant|instrumental response",
         "aversion":"aversion|aversive|CTA|conditioned taste aversion",
         "withdrawal":"withdrawal",
-        "relapse":"relapse|reinstatement|craving|drug seeking|seeking",
+        "relapse":"relapse|reinstatement|craving|drug seeking|seeking|incubation",
         "sensitization":"sensitization",
         "addiction":"addiction|addictive|drug abuse|punishment|compulsive|escalation",
         "dependence":"dependence",