import sys
from event_extract import event_extract
from event_model import EventModel
from event_entity import args
from split_sentence import Cut
import tensorflow as tf
# s="aaaaaaa}{aaaaaaaa"
# s=s.replace('}{', '},{')
# print(s)
if __name__ == '__main__':
saver_path = sys.path[0] + "\data\saver\checkpoint_rnn.data"
model = EventModel(args)
with tf.Session() as sess:
saver = tf.train.Saver(tf.global_variables())
saver.restore(sess, saver_path)
# s="4月17日上午10时59分，晋江市灵源街道好日子超市门口，一辆平板载货车与一辆货车发生碰撞， 致使货车司机被困。"
# new_event,event_type=event_extract(model, sess, s)
# print(new_event)
# print(event_type)
# ----------------------------------------------------- begin --------------------------------------------------
while True:
sentence = sys.stdin.readline()
if sentence == 'exit\n' or sentence == 'exit':
print(sentence)
