import requests
import json

def callServer(prompt, model="deepseek-ai/DeepSeek-V3"):
    url = "https://api.siliconflow.cn/v1/chat/completions"
    headers = {
        "Authorization": f"Bearer sk-szevaazcpwllljvxlyddutjsjzeexsptgtkfmqesswqoldvq",
        "Content-Type": "application/json"
    }
    payload = {
        "model": model,
        "messages": [{"role": "user", "content": prompt}],
        #注意，这里的tempreture要调低，保证结果稳定性
        "temperature": 0.1,
        "max_tokens": 6000
    }

    success = True
    msg = "OK"
    think = ""
    result = ""
    try:
        response = requests.post(url, json=payload, headers=headers)
        response.raise_for_status()
        result = response.json()['choices'][0]['message']['content']
    except Exception as e:
        success = False
        msg = "API调用异常:" + str(e)
    return success, msg, result

def loadRawDataSet(fileName):
    dataList = []
    with open(fileName, "r", encoding="utf-8") as file:
        while True:
            line = file.readline()
            if not line:  # 读取到文件末尾时退出
                break
            line = line.strip()
            if len(line)>0:
                dataList.append(line.split("\t",1))
    return dataList

def writeData(dataList,fileName):
    with open(fileName, 'w', encoding='utf-8') as f:
        for data in dataList:
            f.write(data + "\n")



def tagValid(dataList,prompt):
        correctList = []
        for data in dataList[:1]:
            prompt = prompt.format(data[0],data[1])
            success = False
            while not success:
                success, msg, result = callServer(prompt)
                print(result)
                if success and len(result) == 1 and result in {"是","否"}:
                    correctList += list(result)
                else:
                    success = False
        writeData(correctList, "valid.txt")

if __name__ == "__main__":
    dataList = loadRawDataSet("total.txt")

    prompt="""下列文本是不是{}分类的？文本内容是：{}"""

    tagValid(dataList,prompt)