BIDI_EXTRACTION_AI/BiddingKG/dl/interface/modelFactory.py

'''
Created on 2019年5月16日

@author: User
'''

import os
import sys
sys.path.append(os.path.abspath("../.."))
from keras import models
from keras import layers
# from keras_contrib.layers import CRF
from keras.preprocessing.sequence import pad_sequences
from keras import optimizers,losses,metrics
from BiddingKG.dl.common.Utils import *
import tensorflow as tf
import json

    
class Model_role_classify():
    def __init__(self,lazyLoad=getLazyLoad()):
        #self.model_role_file = os.path.abspath("../role/models/model_role.model.hdf5")
        self.model_role_file = os.path.dirname(__file__)+"/../role/log/new_biLSTM-ep012-loss0.028-val_loss0.040-f10.954.h5"
        self.model_role = None
        self.graph = tf.get_default_graph()
        if not lazyLoad:
            self.getModel()
        
        
    def getModel(self):
        if self.model_role is None:
            self.model_role = models.load_model(self.model_role_file,custom_objects={'precision':precision,'recall':recall,'f1_score':f1_score})
        return self.model_role
    
    def encode(self,tokens,begin_index,end_index,**kwargs):
        return embedding(spanWindow(tokens=tokens,begin_index=begin_index,end_index=end_index,size=10),shape=(2,10,128))
    
    def predict(self,x):
        x = np.transpose(np.array(x),(1,0,2,3))
        with self.graph.as_default():
            return self.getModel().predict([x[0],x[1]])
    
class Model_role_classify_word():
    def __init__(self,lazyLoad=getLazyLoad(),config=None):
        if USE_PAI_EAS:
            lazyLoad = True
        #self.model_role_file = os.path.abspath("../role/log/ep071-loss0.107-val_loss0.122-f10.956.h5")
        # self.model_role_file = os.path.dirname(__file__)+"/../role/models/ep038-loss0.140-val_loss0.149-f10.947.h5"
        #self.model_role_file = os.path.abspath("../role/log/textcnn_ep017-loss0.088-val_loss0.125-f10.955.h5")
        self.model_role = None
        
        self.sess_role = tf.Session(graph=tf.Graph(),config=config)
        if not lazyLoad:
            self.getModel()
        
        
    def getModel(self):
        if self.model_role is None:
          with self.sess_role.as_default() as sess:
            with self.sess_role.graph.as_default():
              meta_graph_def = tf.saved_model.loader.load(sess=self.sess_role, tags=["serve"], export_dir=os.path.dirname(__file__)+"/role_savedmodel")
              signature_key = tf.saved_model.signature_constants.DEFAULT_SERVING_SIGNATURE_DEF_KEY
              signature_def = meta_graph_def.signature_def
              
              input0 = self.sess_role.graph.get_tensor_by_name(signature_def[signature_key].inputs["input0"].name)
              input1 = self.sess_role.graph.get_tensor_by_name(signature_def[signature_key].inputs["input1"].name)
              # input2 = self.sess_role.graph.get_tensor_by_name(signature_def[signature_key].inputs["input2"].name)
              output = self.sess_role.graph.get_tensor_by_name(signature_def[signature_key].outputs["outputs"].name)
              self.model_role = [[input0,input1],output]  #,input2
        return self.model_role
    '''
    def load_weights(self):
        model = self.getModel()
        model.load_weights(self.model_role_file)
    '''
    
    def encode(self,tokens,begin_index,end_index,entity_text,**kwargs):
        _span = spanWindow(tokens=tokens,begin_index=begin_index,end_index=end_index,size=20,center_include=False,word_flag=True,text=entity_text) #size=12 center_include=True
        # print(_span)
        _encode_span = encodeInput(_span, word_len=20, word_flag=True,userFool=False) #  word_len=20
        # print(_encode_span)
        return _encode_span

    def fix_digit_eng(self, text):
        '''
        处理数字及英文编号等
        :param text:
        :return:
        '''
        text = re.sub('第[一二三1-3]([条项章轮次场期批]|中学|医院|分公司|附属)|第三方(服务机构)?|(中标|成交|中选)(候选|结果)?(单位|人)?公[示告]', 'xxx', text) # 修复 642200681 第二轮推荐中选供应商：第一中选候选人： 预测错为第二名
        text = re.sub('第01(中标|成交)?候选人', '第一中标候选人', text)
        text = re.sub('(标[段的包项]?|品目)【?[一二三1-3]】?', '标段', text)
        text = re.sub('(第?[一二三1-3]|\d+)(标段?|[分子]?(标?包|标))', ' ', text) # 修复 503121188 供应商名称：2分标：广西恒广建筑工程有限公司 预测错第二候选; 701962050 供应商名称：02包 武汉欣威制冷空调工程有限公司
        text = re.sub('[a-zA-Z][a-zA-Z0-9=&_—-]{3,}', 'abc', text)
        text = re.sub('[【（\[][0-9]{2,}[\]）】]|\d+([：:.-]\d+)+', 'd', text)
        text = re.sub('[一二三四五六七八九十]{2,}|[四五六七八九十]+', 'd', text)
        text = re.sub('\d{2,}(\.\d+)?|\d\.\d+|[04-9]|/\d+', 'd', text) # 693713326 全自动/50/3，供应商名称：阿荣旗人民检察院，预测错第三候选
        text = re.sub('序[号號]：\d+', '序号：d', text)
        text = re.sub('第?[一二三四五六七八九十\d]+，?[次年包]|[一二三四五六七八九十\d]+、|([^\w]|^)序：?\d+', '  d', text) # ，序：1，单位名称：
        text = re.sub('(中标|成交|中选|入围)(工程|项目)', '工程', text)  # 修复易错分为中标人
        text = re.sub('约定|(盖章|签名)：?', '  ', text) # 修复 233233636 错分为中标人 国有产权网上竞价有关约定 辽阳市公共资源交易中心 ，标  修复 273505905 乡镇签名：盖章： 次村产权交易服务中心 预测为中标
        text = re.sub('中介机构', '投标机构', text) # 251058999 错分为中标人 序号：2，中介机构名称：
        text = re.sub('(采购|招标)人名称、地址和联系方式：|采购方，指|(采购|招标)承办单位：', '采购人：', text) # 275065998  修复 224703143 采购的中标人；采购方，指 预测为中标 750041352 招标承办单位：芜湖安得智联科技有限公司湖南分公司
        if re.search('(最终)?排名：', text) and re.search('(最终)?排名：第?[123一二三]', text)==None:
            text = re.sub('(最终)?排名：', '    ', text)
        text = re.sub('交易单位', '发布单位', text)
        text = re.sub('[，：]各种数据：', '：', text) # 20240620优化 478331984 山东省交通运输厅站源提取不到 各种数据：中标单位，各种数据：济南金曰公路工程有限公司，
        text = re.sub('电子签章', '', text) # 20240924 修复 529923459 电子签名：投标人名称（电子签章：西君兰信息科技有限公司，2024年9月7日 预测为中标
        text = re.sub('采购方式', 'xxxx', text) # 修复 499096797 招标人预测错误
        text = re.sub('中标人\d名称', '中标人名称', text) # 修复 499096797 中标人预测错误
        text = re.sub('\|候选人|比选申请人', '、候选人', text) # 修复 626660259 排名：1|候选人：库尔勒海南广电工程有限责任公司
        text = re.sub('(中标|成交|中选|中拍|排名：1，)(出版社|回收商|医院|(招投?标)?代理(机构|公司))', '成交供应商', text) # 修复 642076501 成交出版社：科学出版社成都有限责任公司， 657862378 确定中标医院为：驻马店市第二人民医院。
        text = re.sub('（(品牌|签名|盖章)）?：|（主）|[【（]?(联合体)?牵头(方|人|单位)[）】]?|[【（]?联合体[）】]?', ' ', text) # 修复 642283019 成交人（品牌： 预测为5 ；修复 702782357 第一中标候选人（主）广东省建设工程质量安全检测总站有限公司
        text = re.sub('竞价（比选）单位', '竞价单位', text) # 修复 598259105 预测为招标
        text = re.sub('第一备选人', '第二候选人', text) # 修复 693681470排名错误
        text = re.sub('第二备选人', '第三候选人', text)
        text = re.sub('公示日期|代理机构备选库', 'xxxx', text) # 修复 679154280 通辽市住房和城乡建设局，公示日期2025年9月11日 错分中标  674197943 珠海市金湾区红旗镇卫生院招标代理机构备选库公开遴选项目 错分代理
        text = re.sub('比选申请人|应答人|参与机构名称|投标公司信息|竞争性磋商公司', '投标人', text) # 修复 700724372 排名：1，比选申请人预测5； 700052214 候选人排名：1，应答人名称：海南厨玛特智能厨房设备有限公司，应答报价（元）：39800；701894552 参加项目竞争性磋商公司有：预测错招标
        text = re.sub('(，|^)[一二三123][、.，]', '，d、', text) # 修复 688267598 二、供应商名称：预测错第二候选；2.供应商名称：复旦大学
        text = re.sub(r'候选人(\d+)名称', r'第\1候选人', text) # 修复 695298070 二、候选内容，候选人1名称：
        text = re.sub(r'(编号|名称)：(\d+)，', r'\1：d，', text) # 修复 692125069 包组名称：3，供应商名称：中国人民健康保险股份有限公司辽宁分公司
        text = re.sub('顺延至?第[二三23]名', ' ', text)
        text = re.sub('(中标|成交)人：[^，：]{2,12}：', '入选供应商：', text) # 修复 697781181 中标人：次氯酸钠：
        text = re.sub('(?<!排名|排序|名次)第[一二三1-3]单位', 'xxx', text) # 修复 689290626 推荐排名第一单位	安徽康鼎建设有限公司 错误替换，其他的要替换
        text = re.sub('中原|出函机构', 'xx', text) # 修复 750901787 中原林场、中原镇企业均为 中原镇人民政府 下属单位，预测错中标 出函机构： 浦发银行
        # print('实体前文：', text)
        return text.replace('(', '（').replace(')', '）').replace('單', '单').replace('稱','称').replace('標', '标').replace('採購', '采购').replace('機構', '机构').replace('應', '应')

    def encode_word(self, sentence_text, begin_index, end_index, size=20, **kwargs):
        '''
        上下文数字化，使用字偏移
        :param sentence_text: 句子文本
        :param begin_index: 实体字开始位置
        :param end_index: 实体字结束位置
        :param size: 字偏移量
        :param kwargs:
        :return:
        '''
        _span = get_context(sentence_text, begin_index, end_index,size=size, center_include=False)  # size=12 center_include=True
        # print(_span)
        _span = [self.fix_digit_eng(text) for text in _span]
        if len(_span) > 1 and _span[0].find('；') > -1: # 修复 638733137 ，排名第一；中标候选人（上海元山建设有限公司） 分句问题导致预测错误
            _span[0] = _span[0][_span[0].find('；'):]
        _encode_span = encodeInput(_span, word_len=30, word_flag=True, userFool=False)  # word_len=20
        # print(_encode_span)
        return _encode_span
    
    def predict(self,x):
        x = np.transpose(np.array(x),(1,0,2))
        model_role = self.getModel()
        assert len(x)==len(model_role[0])
        feed_dict = {}
        for _x,_t in zip(x,model_role[0]):
          feed_dict[_t] = _x
        list_result = limitRun(self.sess_role,[model_role[1]],feed_dict)[0]
        return list_result
        #return self.sess_role.run(model_role[1],feed_dict=feed_dict)
        
    
class Model_money_classify():
    def __init__(self,lazyLoad=getLazyLoad(),config=None):
        if USE_PAI_EAS:
            lazyLoad = True
        self.model_money_file = os.path.dirname(__file__)+"/../money/models/model_money_word.h5"
        self.model_money = None
        self.sess_money = tf.Session(graph=tf.Graph(),config=config)
        if not lazyLoad:
            self.getModel()
        
        
    def getModel(self):
      if self.model_money is None:
        with self.sess_money.as_default() as sess:
          with sess.graph.as_default():
            meta_graph_def = tf.saved_model.loader.load(sess,tags=["serve"],export_dir=os.path.dirname(__file__)+"/money_savedmodel")
            # meta_graph_def = tf.saved_model.loader.load(sess,tags=["serve"],export_dir=os.path.dirname(__file__)+"/money_savedmodel_bilstmonly")
            signature_key = tf.saved_model.signature_constants.DEFAULT_SERVING_SIGNATURE_DEF_KEY
            signature_def = meta_graph_def.signature_def
            
            input0 = sess.graph.get_tensor_by_name(signature_def[signature_key].inputs["input0"].name)
            input1 = sess.graph.get_tensor_by_name(signature_def[signature_key].inputs["input1"].name)
            input2 = sess.graph.get_tensor_by_name(signature_def[signature_key].inputs["input2"].name)
            
            output = sess.graph.get_tensor_by_name(signature_def[signature_key].outputs["outputs"].name)
            
            self.model_money = [[input0,input1,input2],output]
      return self.model_money
      '''
      if self.model_money is None:
          self.model_money = models.load_model(self.model_money_file,custom_objects={'precision':precision,'recall':recall,'f1_score':f1_score})
      return self.model_money
      '''
    
    '''
    def load_weights(self):
        model = self.getModel()
        model.load_weights(self.model_money_file)
    '''
    
    def encode(self,tokens,begin_index,end_index,**kwargs):
        _span = spanWindow(tokens=tokens, begin_index=begin_index, end_index=end_index, size=10, center_include=True, word_flag=True)
        _span[0] = _span[0].replace('拦标价格人民币含税金额', '拦标价') # 20250918 修复 674891060 （二）拦标价格人民币含税金额1877.8081万元 预测错 1
        # print(_span)
        return encodeInput(_span, word_len=30, word_flag=True,userFool=False)
        return embedding_word(_span,shape=(3,100,60))
    
    def predict(self,x):
        # print("shape",np.shape(x))
        x = np.transpose(np.array(x),(1,0,2))
         
        model_money = self.getModel()
        assert len(x)==len(model_money[0])
        feed_dict = {}
        for _x,_t in zip(x,model_money[0]):
          feed_dict[_t] = _x
        list_result = limitRun(self.sess_money,[model_money[1]],feed_dict)[0]
        #return self.sess_money.run(model_money[1],feed_dict=feed_dict)
        return list_result
        '''
        with self.graph.as_default():
            return self.getModel().predict([x[0],x[1],x[2]])
        '''

from itertools import groupby
from BiddingKG.dl.relation_extraction.model import get_words_matrix
class Model_relation_extraction():
    def __init__(self,lazyLoad=getLazyLoad()):
        if USE_PAI_EAS:
            lazyLoad = True
        self.subject_model_file = os.path.dirname(__file__)+"/../relation_extraction/models2/subject_model"
        self.object_model_file = os.path.dirname(__file__)+"/../relation_extraction/models2/object_model"
        self.model_subject = None
        self.model_object = None
        self.sess_subject = tf.Session(graph=tf.Graph())
        self.sess_object = tf.Session(graph=tf.Graph())
        if not lazyLoad:
            self.getModel1()
            self.getModel2()
        self.entity_type_dict = {
            'org': '<company/org>',
            'company': '<company/org>',
            'location': '<location>',
            'phone': '<phone>',
            'person': '<contact_person>'
        }
        self.id2predicate = {
            0: "rel_person",  # 公司——联系人
            1: "rel_phone",  # 联系人——电话
            2: "rel_address"  # 公司——地址
        }
        self.words_size = 128

    # subject_model
    def getModel1(self):
        if self.model_subject is None:
            with self.sess_subject.as_default() as sess:
                with sess.graph.as_default():
                    meta_graph_def = tf.saved_model.loader.load(sess,tags=["serve"],export_dir=self.subject_model_file)
                    signature_key = tf.saved_model.signature_constants.DEFAULT_SERVING_SIGNATURE_DEF_KEY
                    signature_def = meta_graph_def.signature_def
                    input0 = sess.graph.get_tensor_by_name(signature_def[signature_key].inputs["input0"].name)
                    input1 = sess.graph.get_tensor_by_name(signature_def[signature_key].inputs["input1"].name)
                    output = sess.graph.get_tensor_by_name(signature_def[signature_key].outputs["outputs"].name)
                    self.model_subject = [[input0,input1],output]
        return self.model_subject
    # object_model
    def getModel2(self):
        if self.model_object is None:
            with self.sess_object.as_default() as sess:
                  with sess.graph.as_default():
                        meta_graph_def = tf.saved_model.loader.load(sess,tags=["serve"],export_dir=self.object_model_file)
                        signature_key = tf.saved_model.signature_constants.DEFAULT_SERVING_SIGNATURE_DEF_KEY
                        signature_def = meta_graph_def.signature_def
                        input0 = sess.graph.get_tensor_by_name(signature_def[signature_key].inputs["input0"].name)
                        input1 = sess.graph.get_tensor_by_name(signature_def[signature_key].inputs["input1"].name)
                        input2 = sess.graph.get_tensor_by_name(signature_def[signature_key].inputs["input2"].name)
                        output = sess.graph.get_tensor_by_name(signature_def[signature_key].outputs["outputs"].name)
                        self.model_object = [[input0,input1,input2],output]
        return self.model_object

    def encode(self,entity_list,list_sentence):
        list_sentence = sorted(list_sentence, key=lambda x: x.sentence_index)
        entity_list = sorted(entity_list, key=lambda x: (x.sentence_index, x.begin_index))
        pre_data = []
        text_data = []
        last_sentence_index = -1
        for key, group in groupby(entity_list, key=lambda x: x.sentence_index):
            if key - last_sentence_index > 1:
                for i in range(last_sentence_index + 1, key):
                    pre_data.extend(list_sentence[i].tokens)
                    text_data.extend([0] * len(list_sentence[i].tokens))
            group = list(group)
            for i in range(len(group)):
                ent = group[i]
                _tokens = list_sentence[key].tokens
                if i == len(group) - 1:
                    if i == 0:
                        pre_data.extend(_tokens[:ent.begin_index])
                        text_data.extend([0] * len(_tokens[:ent.begin_index]))
                        pre_data.append(self.entity_type_dict[ent.entity_type])
                        text_data.append(ent)
                        pre_data.extend(_tokens[ent.end_index + 1:])
                        text_data.extend([0] * len(_tokens[ent.end_index + 1:]))
                        break
                    else:
                        pre_data.append(self.entity_type_dict[ent.entity_type])
                        text_data.append(ent)
                        pre_data.extend(_tokens[ent.end_index + 1:])
                        text_data.extend([0] * len(_tokens[ent.end_index + 1:]))
                        break
                if i == 0:
                    pre_data.extend(_tokens[:ent.begin_index])
                    text_data.extend([0] * len(_tokens[:ent.begin_index]))
                    pre_data.append(self.entity_type_dict[ent.entity_type])
                    text_data.append(ent)
                    pre_data.extend(_tokens[ent.end_index + 1:group[i + 1].begin_index])
                    text_data.extend([0] * len(_tokens[ent.end_index + 1:group[i + 1].begin_index]))
                else:
                    pre_data.append(self.entity_type_dict[ent.entity_type])
                    text_data.append(ent)
                    pre_data.extend(_tokens[ent.end_index + 1:group[i + 1].begin_index])
                    text_data.extend([0] * len(_tokens[ent.end_index + 1:group[i + 1].begin_index]))
            last_sentence_index = key
        return text_data, pre_data

    def check_data(self, words):
        # 检查数据是否包含可预测的subject和object
        # 没有需要预测的链接属性，直接return
        company_relation = 0
        person_relation = 0
        if '<company/org>' in words:
            company_relation += 1
        if '<contact_person>' in words:
            person_relation += 1
            if company_relation:
                company_relation += 1
        # 暂时不考虑地址location实体
        # if '<location>' in words and company_relation:
        #     company_relation += 1
        if '<phone>' in words and company_relation:
            person_relation += 1
        if company_relation < 2 and person_relation < 2:
            return False
        return True
    def predict_by_api(self,text_in,words,sentence_vetor):
        status_code = 0
        # save([words,sentence_vetor.tolist()],"C:/Users/Administrator/Desktop/test_data.pk")
        try:
            requests_result = requests.post(API_URL + "/predict_relation", json={"sentence_vetor": sentence_vetor.tolist(), "words": words},
                                            verify=True)
            status_code = requests_result.status_code
            triple_index_list = json.loads(requests_result.text)['triple_list']
            # print("triple_list:",json.loads(requests_result.text)['triple_list'])
            print("cost_time:",json.loads(requests_result.text)['cost_time'])
            triple_list = [(text_in[triple[0]], triple[1], text_in[triple[2]]) for triple in triple_index_list]
            return triple_list,status_code
        except Exception as e:
            print(e)
            return [],status_code

    def predict(self,text_in, words, rate=0.5):
        _t2 = np.zeros((len(words), self.words_size))
        for i in range(len(words)):
            _t2[i] = np.array(get_words_matrix(words[i]))

        # a = time.time()
        # triple_list, status_code = self.predict_by_api(text_in, words,_t2)
        # print('time',time.time()-a)
        # print("status_code",status_code)
        # if status_code==200:
        #     return triple_list
        # else:
        # 使用模型预测
        triple_list = []
        # print("tokens:",words)
        # _t2 = [self.words2id.get(c, 1) for c in words]

        _t2 = np.array([_t2])
        _t3 = [1 for _ in words]
        _t3 = np.array([_t3])
        # _k1 = self.model_subject.predict([_t2, _t3])
        _k1 = limitRun(self.sess_subject,[self.model_subject[1]],feed_dict={self.model_subject[0][0]:_t2,
                                                                            self.model_subject[0][1]:_t3})[0]
        _k1 = _k1[0, :, 0]
        _k1 = np.where(_k1 > rate)[0]
        # print('k1',_k1)
        _subjects = []
        for i in _k1:
            _subject = text_in[i]
            _subjects.append((_subject, i, i))
        if _subjects:
            _t2 = np.repeat(_t2, len(_subjects), 0)
            _t3 = np.repeat(_t3, len(_subjects), 0)
            _k1, _ = np.array([_s[1:] for _s in _subjects]).T.reshape((2, -1, 1))
            # _o1 = self.model_object.predict([_t2, _t3, _k1])
            _o1 = limitRun(self.sess_object, [self.model_object[1]], feed_dict={self.model_object[0][0]: _t2,
                                                                                  self.model_object[0][1]: _t3,
                                                                                  self.model_object[0][2]: _k1})[0]
            for i, _subject in enumerate(_subjects):
                _oo1 = np.where(_o1[i] > 0.5)
                # print('_oo1', _oo1)
                for _ooo1, _c1 in zip(*_oo1):
                    _object = text_in[_ooo1]
                    _predicate = self.id2predicate[_c1]
                    triple_list.append((_subject[0], _predicate, _object))
            # print([(t[0].entity_text,t[1],t[2].entity_text) for t in triple_list])
            return triple_list
        else:
            return []

    
class Model_person_classify():
    def __init__(self,lazyLoad=getLazyLoad(),config=None):
        if USE_PAI_EAS:
            lazyLoad = True
        self.model_person_file = os.path.dirname(__file__)+"/../person/models/model_person.model.hdf5"
        self.model_person = None
        self.sess_person = tf.Session(graph=tf.Graph(),config=config)
        if not lazyLoad:
            self.getModel()
        
        
    def getModel(self):
      if self.model_person is None:
        with self.sess_person.as_default() as sess:
          with sess.graph.as_default():
            # meta_graph_def = tf.saved_model.loader.load(sess,tags=["serve"],export_dir=os.path.dirname(__file__)+"/person_savedmodel_new")
            meta_graph_def = tf.saved_model.loader.load(sess,tags=["serve"],export_dir=os.path.dirname(__file__)+"/person_savedmodel_new_znj")
            signature_key = tf.saved_model.signature_constants.DEFAULT_SERVING_SIGNATURE_DEF_KEY
            signature_def = meta_graph_def.signature_def
            
            input0 = sess.graph.get_tensor_by_name(signature_def[signature_key].inputs["input0"].name)
            input1 = sess.graph.get_tensor_by_name(signature_def[signature_key].inputs["input1"].name)
            
            output = sess.graph.get_tensor_by_name(signature_def[signature_key].outputs["outputs"].name)
            
            self.model_person = [[input0,input1],output]
      return self.model_person
            
      
      '''
        if self.model_person is None:
            self.model_person = models.load_model(self.model_person_file,custom_objects={'precision':precision,'recall':recall,'f1_score':f1_score})
        return self.model_person
      '''
    '''
    def load_weights(self):
        model = self.getModel()
        model.load_weights(self.model_person_file)
    '''
    
    def encode(self,tokens,begin_index,end_index,**kwargs):
        # return embedding(spanWindow(tokens=tokens,begin_index=begin_index,end_index=end_index,size=10),shape=(2,10,128))
        return embedding(spanWindow(tokens=tokens,begin_index=begin_index,end_index=end_index,size=20),shape=(2,20,128))

    def predict(self,x):
        x = np.transpose(np.array(x),(1,0,2,3))
        
        model_person = self.getModel()
        assert len(x)==len(model_person[0])
        feed_dict = {}
        for _x,_t in zip(x,model_person[0]):
          feed_dict[_t] = _x

        list_result = limitRun(self.sess_person,[model_person[1]],feed_dict)[0]
        return list_result
        #return self.sess_person.run(model_person[1],feed_dict=feed_dict)
        '''
        with self.graph.as_default():
            return self.getModel().predict([x[0],x[1]])
        '''
    
class Model_form_line():
    def __init__(self,lazyLoad=getLazyLoad()):
        self.model_file = os.path.dirname(__file__)+"/../form/model/model_form.model - 副本.hdf5"
        self.model_form = None
        self.graph = tf.get_default_graph()
        if not lazyLoad:
            self.getModel()
        
        
    def getModel(self):
        if self.model_form is None:
            self.model_form = models.load_model(self.model_file,custom_objects={"precision":precision,"recall":recall,"f1_score":f1_score})
        return self.model_form
    
    def encode(self,data,shape=(100,60),expand=False,**kwargs):
        embedding = np.zeros(shape)
        word_model = getModel_word()
        for i in range(len(data)):
            if i>=shape[0]:
                break
            if data[i] in word_model.vocab:
                embedding[i] = word_model[data[i]]
        if expand:
            embedding = np.expand_dims(embedding,0)
        return embedding
    
    def predict(self,x):
        with self.graph.as_default():
            return self.getModel().predict(x)
    
class Model_form_item():
    def __init__(self,lazyLoad=getLazyLoad(),config=None):
        self.model_file = os.path.dirname(__file__)+"/../form/log/ep039-loss0.038-val_loss0.064-f10.9783.h5"
        self.model_form = None
        self.sess_form = tf.Session(graph=tf.Graph(),config=config)
        if not lazyLoad:
            self.getModel()


    def getModel(self):
      if self.model_form is None:
        with self.sess_form.as_default() as sess:
          with sess.graph.as_default():
            meta_graph_def = tf.saved_model.loader.load(sess,tags=["serve"],export_dir="%s/form_savedmodel"%(os.path.dirname(__file__)))
            signature_key = tf.saved_model.signature_constants.DEFAULT_SERVING_SIGNATURE_DEF_KEY
            signature_def = meta_graph_def.signature_def

            inputs = sess.graph.get_tensor_by_name(signature_def[signature_key].inputs["inputs"].name)
            output = sess.graph.get_tensor_by_name(signature_def[signature_key].outputs["outputs"].name)

            self.model_form = [[inputs],output]
      return self.model_form
      '''
        if self.model_form is None:
            with self.graph.as_defalt():
                self.model_form = models.load_model(self.model_file,custom_objects={"precision":precision,"recall":recall,"f1_score":f1_score})
        return self.model_form
      '''

    def encode(self,data,**kwargs):
        return encodeInput([data], word_len=50, word_flag=True,userFool=False)[0]
        return encodeInput_form(data)

    def predict(self,x):
        if USE_API:
            requests_result = requests.post(API_URL+"/predict_form_item",json={"inputs":x.tolist()}, verify=True)
            list_result = json.loads(requests_result.text)['result']
        else:
            model_form = self.getModel()

            list_result = limitRun(self.sess_form,[model_form[1]],feed_dict={model_form[0][0]:x})[0]
        return list_result
        # return self.sess_form.run(model_form[1],feed_dict={model_form[0][0]:x})

        '''
          with self.graph.as_default():
              return self.getModel().predict(x)
        '''

class Model_form_context():
    def __init__(self,lazyLoad=getLazyLoad(),config=None):
        self.model_form = None
        self.sess_form = tf.Session(graph=tf.Graph(),config=config)
        if not lazyLoad:
            self.getModel()


    def getModel(self):
        if self.model_form is None:
            with self.sess_form.as_default() as sess:
                with sess.graph.as_default():
                    meta_graph_def = tf.saved_model.loader.load(sess,tags=["serve"],export_dir="%s/form_context_savedmodel"%(os.path.dirname(__file__)))
                    signature_key = tf.saved_model.signature_constants.DEFAULT_SERVING_SIGNATURE_DEF_KEY
                    signature_def = meta_graph_def.signature_def

                    inputs = sess.graph.get_tensor_by_name(signature_def[signature_key].inputs["inputs"].name)
                    output = sess.graph.get_tensor_by_name(signature_def[signature_key].outputs["outputs"].name)

                    self.model_form = [[inputs],output]
        return self.model_form
        '''
          if self.model_form is None:
              with self.graph.as_defalt():
                  self.model_form = models.load_model(self.model_file,custom_objects={"precision":precision,"recall":recall,"f1_score":f1_score})
          return self.model_form
        '''

    def encode_table(self,inner_table,size=30):
        def encode_item(_table,i,j):
            _x = [_table[j-1][i-1],_table[j-1][i],_table[j-1][i+1],
                  _table[j][i-1],_table[j][i],_table[j][i+1],
                  _table[j+1][i-1],_table[j+1][i],_table[j+1][i+1]]
            e_x = [encodeInput_form(_temp[0],MAX_LEN=30) for _temp in _x]
            _label = _table[j][i][1]
            # print(_x)
            # print(_x[4],_label)
            return e_x,_label,_x

        def copytable(inner_table):
            table = []
            for line in inner_table:
                list_line = []
                for item in line:
                    list_line.append([item[0][:size],item[1]])
                table.append(list_line)
            return table

        table = copytable(inner_table)
        padding = ["#"*30,0]
        width = len(table[0])
        height = len(table)
        table.insert(0,[padding for i in range(width)])
        table.append([padding for i in range(width)])
        for item in table:
            item.insert(0,padding.copy())
            item.append(padding.copy())
        data_x = []
        data_y = []
        data_text = []
        data_position = []
        for _i in range(1,width+1):
            for _j in range(1,height+1):
                _x,_y,_text = encode_item(table,_i,_j)
                data_x.append(_x)
                _label = [0,0]
                _label[_y] = 1
                data_y.append(_label)
                data_text.append(_text)
                data_position.append([_i-1,_j-1])
                # input = table[_j][_i][0]
                # item_y = [0,0]
                # item_y[table[_j][_i][1]] = 1
                # data_x.append(encodeInput([input], word_len=50, word_flag=True,userFool=False)[0])
                # data_y.append(item_y)
        return data_x,data_y,data_text,data_position

    def encode(self,inner_table,**kwargs):
        data_x,_,_,data_position = self.encode_table(inner_table)
        return data_x,data_position

    def predict(self,x):

        model_form = self.getModel()

        list_result = limitRun(self.sess_form,[model_form[1]],feed_dict={model_form[0][0]:x})[0]
        return list_result

# class Model_form_item():
#     def __init__(self,lazyLoad=False):
#         self.model_file = os.path.dirname(__file__)+"/ep039-loss0.038-val_loss0.064-f10.9783.h5"
#         self.model_form = None
#
#         if not lazyLoad:
#             self.getModel()
#         self.graph = tf.get_default_graph()
#
#     def getModel(self):
#         if self.model_form is None:
#             self.model_form = models.load_model(self.model_file,custom_objects={"precision":precision,"recall":recall,"f1_score":f1_score})
#         return self.model_form
#
#     def encode(self,data,**kwargs):
#
#         return encodeInput_form(data)
#
#     def predict(self,x):
#         with self.graph.as_default():
#             return self.getModel().predict(x)