import json
import logging

import requests
from openpyxl import load_workbook

# 更详细地控制日志输出
logger = logging.getLogger('MathRag')
logger.setLevel(logging.INFO)
handler = logging.StreamHandler()
handler.setFormatter(logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s'))
logger.addHandler(handler)


def getAnswer(query_text):
    # 根据问题查询知识库
    url = "http://10.10.21.20:8100/api/chat"
    headers = {
        "Content-Type": "application/json"
    }
    data = {
        "query": query_text,
        "topic": "ChuZhongShuXue"
    }
    try:
        response = requests.post(url, headers=headers, data=json.dumps(data))
        response.raise_for_status()
        return response.json().get("data").get("reply")
    except requests.exceptions.RequestException as e:
        logger.error(f"请求发生异常: {str(e)}")
        return None


excel_file = r'D:\dsWork\dsProject\dsLightRag\Doc\T1、史校长资料\测试结果\100题测试结果.xlsx'

if __name__ == '__main__':
    # query_text = '什么是数学的性质命题'
    # print(getAnswer(query_text))

    wb = load_workbook(excel_file)
    ws = wb.active  # 获取第一个sheet
    # 读取C列第2行到101行的数据
    idx = 1
    for row in range(2, 102):  # 行号从2到101
        cell_value = ws[f'C{row}'].value
        logger.info("问题{}:{}".format(idx, cell_value))
        idx = idx + 1

    wb.close()