import requests


def dialog_line_parse(url, text):
    """
    将数据输入模型进行分析并输出结果
    :param url: 模型url
    :param text: 进入模型的数据
    :return: 模型返回结果
    """

    response = requests.post(
        url,
        json=text,
        timeout=100000
    )
    if response.status_code == 200:
        return response.json()
    else:
        # logger.error(
        #     "【{}】 Failed to get a proper response from remote "
        #     "server. Status Code: {}. Response: {}"
        #     "".format(url, response.status_code, response.text)
        # )
        print("【{}】 Failed to get a proper response from remote "
              "server. Status Code: {}. Response: {}"
              "".format(url, response.status_code, response.text))
        print(text)
        return {}


with open("data/drop_weight_data.txt", encoding="utf-8") as f:
    text_list = [i for i in f.read().split("\n")]
    for i in text_list[:-1]:
        text = dialog_line_parse("http://192.168.31.74:19000", {"texts": f"改写下面这句话，要求意思接近但是改动幅度比较大，字数只能多不能少:\n{i}"})
        print("原文",i)
        print("模型预测", text)