aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rwxr-xr-xratspub.py3
-rwxr-xr-xserver.py9
2 files changed, 8 insertions, 4 deletions
diff --git a/ratspub.py b/ratspub.py
index c82c9b8..eca554c 100755
--- a/ratspub.py
+++ b/ratspub.py
@@ -30,6 +30,7 @@ def getSentences(query, gene):
def gene_category(gene, cat_d, query, cat):
#e.g. BDNF, addiction_d, undic(addiction_d) "addiction"
q="\"(" + query.replace("|", " OR ") + ") AND " + gene + "\""
+ print (">>>>>>>>>>" + q)
sents=getSentences(q, gene)
out=str()
for sent in sents.split("\n"):
@@ -75,7 +76,7 @@ addiction=undic(addiction_d)
drug_d = {"alcohol":"alcohol|alcoholism|alcoholic|alcoholics",
"nicotine":"smoking|nicotine|tobacco|smoker|smokers",
"cocaine":"cocaine",
- "opioid":"opioid|opioids|fentanyl|oxycodone|oxycontin|heroin|morphine|methadone|buprenorphine|vicodin|hydrocodone|hycodan|kadian|percoset|hydromorphone|naloxone|codeine|suboxone|tramadol|kratom",
+ "opioid":"opioid|opioids|fentanyl|oxycodone|oxycontin|heroin|morphine|methadone|buprenorphine|vicodin|hydrocodone|hycodan|kadian|percoset|hydromorphone|naloxone|codeine|suboxone|tramadol|kratom|ultram",
"amphetamine":"methamphetamine|amphetamine|METH|AMPH",
"cannabinoid":"endocannabinoid|cannabinoids|cannabis|endocannabinoids|marijuana|cannabidiol|cannabinoid|tetrahydrocannabinol|thc|thc 9|Oleoylethanolamide|palmitoylethanolamide|acylethanolamides"
}
diff --git a/server.py b/server.py
index b71e6bb..90b2e51 100755
--- a/server.py
+++ b/server.py
@@ -25,7 +25,7 @@ def progress():
genes=genes.replace(",", " ")
genes=genes.replace(";", " ")
genes=genes.split()
- if len(genes)>=6:
+ if len(genes)>=60:
message="<span class='text-danger'>Up to five terms can be searched at a time</span>"
return render_template('index.html', message=message)
elif len(genes)==0:
@@ -50,13 +50,15 @@ def search():
edges=str()
nodes=default_nodes
progress=0
+ searchCnt=0
for gene in genes:
gene=gene.replace("-"," ")
nodes+="{ data: { id: '" + gene + "', nodecolor:'#E74C3C', fontweight:700, url:'/gene_gene?gene="+gene+"'} },\n"
# report progress immediately
progress+=percent
yield "data:"+str(progress)+"\n\n"
- addiction=undic(addiction_d)
+ #addiction terms must present with at least one drug
+ addiction=undic(addiction_d) +") AND ("+undic(drug_d)
sent0=gene_category(gene, addiction_d, addiction, "addiction")
e0=generate_edges(sent0, tf_name)
#
@@ -78,7 +80,8 @@ def search():
edges+=e0+e1+e2+e3
sentences+=sent0+sent1+sent2+sent3
#save data before the last yield
- if (progress>99):
+ searchCnt+=1
+ if (searchCnt==len(genes)):
progress=100
sntdata.write(sentences)
sntdata.close()