import time
import json
import math
from flask import Flask,Response,request
from flask_sse import  sse
from flask_cors import CORS
import re
import qwen_agenttext
app = Flask(__name__)
cros = CORS(app)
# SSE 推送函数
import paddle;
paddle.device.get_available_device()


# SSE 推送路由


# @app.route('/register', methods=["GET"])
# def register():
    # 获取客户端标识符
    # client_id = str(uuid.uuid4())
    #
    # # 返回 SSE 响应
    # return jsonify({"client_id": client_id})


# SSE 推送路由


# @app.route('/sse', methods=['POST'])
# def stream():
#     # 获取客户端标识符
#     client_id = 1
#     print("client_id", client_id)
#
#     def aa():
#         # 循环发送 SSE 数据
#         for i in range(10):
#             data = 'Hello, %s!' % client_id + str(i)
#             print(data)
#             sse.publish(data, channel=client_id, type='message')
#             time.sleep(1)
#         sse.publish("end", channel=client_id, type='message')
#
#     # 返回 SSE 响应
#     response = Response(aa(), mimetype='text/event-stream')
#     response.headers.add('Cache-Control', 'no-cache')
#     response.headers.add('Connection', 'keep-alive')
#     response.headers.add('X-Accel-Buffering', 'no')
#     return response
#
#
#
# @app.route('/stream' ,methods=["GET", "POST"])
# def stream_numbers():
#     context= request.args.get('context')
#
#
#     headers = {
#         "Content-Type": "text/event-stream",
#         "Cache-Control": "no-cache",
#         "X-Accel-Buffering": "no",
#         "Access-Control-Allow-Origin": "*",
#         "Access-Control-Allow-Methods": "GET,POST",
#         "Access-Control-Allow-Headers": "x-requested-with,content-type",
#     }
#     return Response(generate_numbers(),headers=headers)
# def generate_numbers():
#     event_id=0
#     # for number in range(1, 10):
#     #     json_data = json.dumps({"number": number})
#     #     print(json_data)
#     #     event_id += 1
#     #     yield f"id: {event_id}\n"
#     #     yield f"event: time-update\n"
#     #     yield f"data: {json_data}\n\n"  # 每次生成一个数字就发送
#     json_data = json.dumps({"number": "done"})
#     yield f"id: {1}\n"
#     yield f"event: time-update\n"
#     yield f"data: 34568\n\n"  # 发送完成信号
# if __name__ == '__main__':
#
#
#     # 读取文件内容
#     with open("checkPlaceName.txt", "r", encoding='utf-8') as f:
#         gettext = f.read()
#     batchNum=20
#     sentences = re.split(r'[。\n]', gettext)
#     # 去掉空字符
#     sentences = [sentence.strip() for sentence in sentences if sentence.strip()]
#     # 计算总字符数
#     total_chars = len(sentences)
#
#     # 计算有多少份
#     num_chunks = math.ceil(total_chars / batchNum)
#
#     # 按batchNum字为一份进行处理
#     chunks = [sentences[i:i + batchNum] for i in range(0, total_chars, batchNum)]
#
#     # 打印每一份的内容
#     for i, chunk in enumerate(chunks):
#         print(f"Chunk {i + 1}:")
#         print(chunk)
#         print("-" * 40)
#
#     # 打印总份数
#     print(f"Total chunks: {num_chunks}")
#     app.run(debug=True,port=80)