import json
import time

from aip import AipNlp

APP_ID = '23011695'
API_KEY = 'KCWNoMvOADpTsCSkYU6Hlhcv'
SECRET_KEY = 'CQjASIyFwPCWTw4OFHjBP4VHrzboybwa'
client = AipNlp(APP_ID, API_KEY, SECRET_KEY)

dataList = []
with open("file/juzi.txt", "r") as f:
    text = f.read()
    for i, juzi in enumerate(text.split("\n")):
        for j in range(3):
            if juzi == "":
                continue
            options = {}
            options["mode"] = 0
            res = client.depParser(juzi, options)
            print(res)
            res["id"] = i
            if res.get("error_code") is not None:
                time.sleep(5)
                continue
            dataList.append(res)
            break
        else:
            raise Exception("处理失败")
dataStr = json.dumps(dataList, ensure_ascii=False, indent=3)
with open("../model/nlp_model.json", "w") as f:
    f.write(dataStr)