In [ ]:
from practnlptools.tools import Annotator
annotator=Annotator()
annotator.getAnnotations("they pour it into a bowl")['srl'][0]
def create_instructions(phrase):
annotated = annotator.getAnnotations(phrase)['srl']
annotated_steps = []
if len(annotated) > 0:
for i in xrange(len(annotated)):
annotated_step = dict()
annotated_step['action'] = annotated[i]['V']
if set(['A1','A2']).issubset(annotated[i].keys()):
annotated_step['object'] = annotated[i]['A1']
annotated_step['target'] = annotated[i]['A2']
elif set(['A2']).issubset(annotated[i].keys()):
annotated_step['target'] = annotated[i]['A2']
elif set(['A1']).issubset(annotated[i].keys()):
annotated_step['object'] = annotated[i]['A1']
else:
pass
annotated_steps.append(annotated_step)
return annotated_steps
create_instructions(s)