Update app.py

app.py

@@ -248,23 +248,27 @@ if submit and len(x) != 0:
   edcrecep = []
   hormrecep = []
   hormcan = []
-  
+
   for e in edccanbis:
-    if e[-1]== ["B-EDC","B-CANCER"]:
+    if e[-1]== ["B-EDC","B-CANCER"] and e[2] in ["CAUSE","INCREASE_RISK","SPEED_UP","NO_ASSOCIATION"]:
       edccann += [[e[0],e[1],e[2]]]
+      edccan += [[e[3],e[0]+" ["+e[-1][0][2:]+"]", e[1]+" ["+e[-1][1][2:]+"]",e[2]]]
       
-      
-    elif e[-1]== ["B-EDC","B-HORMONE"]:
+    elif e[-1]== ["B-EDC","B-HORMONE"] and e[2] in ["ALTER", "INCREASE", "DECREASE", "MIMIC", "INCREASE_ACTIVITY","DECREASE_ACTIVITY"]:
       edchorm += [[e[0],e[1],e[2]]]
+      edccan += [[e[3],e[0]+" ["+e[-1][0][2:]+"]", e[1]+" ["+e[-1][1][2:]+"]",e[2]]]
       
-    elif e[-1]== ["B-EDC","B-RECEPTOR"]:
+    elif e[-1]== ["B-EDC","B-RECEPTOR"] and e[2] in ["BLOCK", "ACTIVATE", "INTERFER"] :
       edcrecep += [[e[0],e[1],e[2]]]
+      edccan += [[e[3],e[0]+" ["+e[-1][0][2:]+"]", e[1]+" ["+e[-1][1][2:]+"]",e[2]]]
      
-    elif e[-1]== ["B-HORMONE","B-RECEPTOR"]:
+    elif e[-1]== ["B-HORMONE","B-RECEPTOR"] and e[2] in ["BIND"] :
       hormrecep += [[e[0],e[1],e[2]]]
+      edccan += [[e[3],e[0]+" ["+e[-1][0][2:]+"]", e[1]+" ["+e[-1][1][2:]+"]",e[2]]]
       
-    elif e[-1]== ["B-HORMONE","B-CANCER"]:
+    elif e[-1]== ["B-HORMONE","B-CANCER"] and e[2] in ["DEVELOP"]:
       hormcan += [[e[0],e[1],e[2]]]
+      edccan += [[e[3],e[0]+" ["+e[-1][0][2:]+"]", e[1]+" ["+e[-1][1][2:]+"]",e[2]]]
   
   
   edcrecepdf  = pd.DataFrame(edcrecep, columns=["EDC", "RECEPTOR", "RELATION"])
@@ -281,8 +285,7 @@ if submit and len(x) != 0:
   
   
   
-  for e in edccanbis:
-    edccan += [[e[3],e[0]+" ["+e[-1][0][2:]+"]", e[1]+" ["+e[-1][1][2:]+"]",e[2]]]
+
    
   edccandf = pd.DataFrame(edccan, columns= ["Sentence", "Entity 1", "Entity 2", "Relation"] )