import requests
import json
from Config import *

# Dify API 配置
url = DIFY_URL + "/chat-messages"
headers = {
    'Authorization': 'Bearer ' + DIFY_API_KEY,
    'Content-Type': 'application/json'
}

# 请求数据
payload = json.dumps({
    "inputs": {},
    "query": "你是谁？",
    "response_mode": "streaming",  # 启用流式输出
    "conversation_id": "",
    "user": "abc-123"
})

# 发送请求并处理流式响应
try:
    with requests.post(url, headers=headers, data=payload, stream=True) as response:
        response.raise_for_status()  # 检查请求是否成功

        showed_task_id = False
        # 逐行处理流式响应
        for line in response.iter_lines():
            if line:  # 过滤掉空行
                decoded_line = line.decode('utf-8')  # 解码为 UTF-8
                if decoded_line.startswith("data: "):  # 处理以 "data: " 开头的行
                    json_str = decoded_line[len("data: "):]  # 去掉 "data: " 前缀
                    if json_str == "[DONE]":  # 流式响应结束
                        break
                    try:
                        # 解析 JSON 数据
                        json_data = json.loads(json_str)
                        # 首次显示task_id
                        if not showed_task_id:
                            print('task_id=' + json_data['task_id'])
                            print('message_id=' + json_data['message_id'] )
                            print('conversation_id=' + json_data['conversation_id'] + '\n')
                            showed_task_id = True
                        # 提取并输出流式内容
                        if "answer" in json_data:
                            print(json_data["answer"], end="", flush=True)  # 实时输出
                    except json.JSONDecodeError:
                        pass  # 忽略解析错误

except requests.exceptions.RequestException as e:
    print("请求失败:", e)