zhouhaibin
4 months ago
8 changed files with 963 additions and 609 deletions
@ -1,206 +1,286 @@ |
|||
from flask import Flask, request, jsonify, Response |
|||
# from flask import Flask, request, jsonify, Response |
|||
import os |
|||
from checkPlaceName import checkPlaceName |
|||
from checkRepeatText import checkRepeatText |
|||
from checkCompanyName import checkCompanyName |
|||
from checkDocumentError import checkDocumentError |
|||
from checkTitleName import checkTitleName |
|||
from flask_cors import CORS |
|||
# from flask_cors import CORS |
|||
import qwen_agenttext |
|||
from myLogger import outLog |
|||
import time |
|||
app = Flask(__name__) |
|||
cros = CORS(app) |
|||
# app = Flask(__name__) |
|||
# cros = CORS(app) |
|||
import uvicorn |
|||
from fastapi import FastAPI, Request, File, UploadFile, HTTPException |
|||
from fastapi.responses import JSONResponse |
|||
from fastapi.middleware.cors import CORSMiddleware |
|||
from sse_starlette.sse import EventSourceResponse |
|||
import asyncio |
|||
|
|||
app = FastAPI() |
|||
# 允许所有来源的跨域请求 |
|||
app.add_middleware( |
|||
CORSMiddleware, |
|||
allow_origins=["*"], |
|||
allow_credentials=True, |
|||
allow_methods=["*"], |
|||
allow_headers=["*"] |
|||
) |
|||
|
|||
UPLOAD_FOLDER = 'uploads' |
|||
if not os.path.exists(UPLOAD_FOLDER): |
|||
os.makedirs(UPLOAD_FOLDER) |
|||
|
|||
|
|||
@app.route('/upload', methods=['POST']) |
|||
def upload_file(): |
|||
if 'file' not in request.files: |
|||
return jsonify({"error": "No file part"}), 400 |
|||
file = request.files['file'] |
|||
if file.filename == '': |
|||
return jsonify({"error": "No selected file"}), 400 |
|||
if file: |
|||
filename = file.filename |
|||
file.save(os.path.join(UPLOAD_FOLDER, filename)) |
|||
return jsonify({"message": "File uploaded successfully"}), 200 |
|||
|
|||
|
|||
@app.route('/stream', methods=["GET", "POST"]) |
|||
def stream_numbers(): |
|||
context = request.args.get('context') |
|||
# 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" # 每次生成一个数字就发送 |
|||
# time.sleep(0.5) # 为了演示，加入短暂延迟 |
|||
# json_data = json.dumps({"number": "done"}) |
|||
# yield f"id: {1}\n" |
|||
# yield f"event: time-update\n" |
|||
# yield f"data: {json_data}\n\n" # 发送完成信号 |
|||
|
|||
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", |
|||
# @app.route('/upload', methods=['POST']) |
|||
# def upload_file(): |
|||
# if 'file' not in request.files: |
|||
# return jsonify({"error": "No file part"}), 400 |
|||
# file = request.files['file'] |
|||
# if file.filename == '': |
|||
# return jsonify({"error": "No selected file"}), 400 |
|||
# if file: |
|||
# filename = file.filename |
|||
# file.save(os.path.join(UPLOAD_FOLDER, filename)) |
|||
# return jsonify({"message": "File uploaded successfully"}), 200 |
|||
@app.post("/sse/upload") |
|||
async def upload_file(file: UploadFile = File(...)): |
|||
if not file.filename: |
|||
raise HTTPException(status_code=400, detail="No selected file") |
|||
|
|||
# 保存文件 |
|||
try: |
|||
file_location = os.path.join(UPLOAD_FOLDER, file.filename) |
|||
with open(file_location, "wb") as f: |
|||
content = await file.read() |
|||
f.write(content) |
|||
return JSONResponse(content={"message": "文件上传成功"}, status_code=200) |
|||
except Exception as e: |
|||
raise HTTPException(status_code=500, detail="文件上传失败，错误信息：" + str(e)) |
|||
|
|||
|
|||
@app.get("/sse") |
|||
async def root(request: Request): |
|||
async def event_generator(request: Request): |
|||
res_str = "七夕情人节即将来临，我们为您准备了精美的鲜花和美味的蛋糕" |
|||
for i in res_str: |
|||
if await request.is_disconnected(): |
|||
print("连接已中断") |
|||
break |
|||
yield { |
|||
"event": "message", |
|||
"id": "7", |
|||
"data": f"{i}" |
|||
} |
|||
return Response(qwen_agenttext.getxinx(context), headers=headers) |
|||
|
|||
await asyncio.sleep(0.1) |
|||
|
|||
g = event_generator(request) |
|||
return EventSourceResponse(g) |
|||
|
|||
|
|||
@app.route('/sse/checkRepeatText', methods=['GET']) |
|||
def checkRepeatTextWeb(): |
|||
filename = request.args.get('filename') |
|||
userId = request.args.get("userId") |
|||
# def stream_numbers(): |
|||
# context = request.args.get('context') |
|||
# # 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" # 每次生成一个数字就发送 |
|||
# # time.sleep(0.5) # 为了演示，加入短暂延迟 |
|||
# # json_data = json.dumps({"number": "done"}) |
|||
# # yield f"id: {1}\n" |
|||
# # yield f"event: time-update\n" |
|||
# # yield f"data: {json_data}\n\n" # 发送完成信号 |
|||
|
|||
def generate_checkRepeatText(filename,userId): |
|||
# 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(qwen_agenttext.getxinx(context), headers=headers) |
|||
|
|||
@app.get("/sse/checkRepeatText") |
|||
async def checkRepeatTextWeb(filename, userId, request: Request): |
|||
async def generate_checkRepeatText(filename, userId, request: Request): |
|||
global outLog |
|||
id = 0 |
|||
for i in checkRepeatText(filename,userId): |
|||
yield f"id: {id + 1}\n" |
|||
yield f"event: checkRepeatText\n" |
|||
yield f"data: {i}\n\n" # 发送完成信号 |
|||
# except Exception as e: |
|||
|
|||
# yield f"id: {id+1}\n" |
|||
# yield f"event: checkRepeatText\n" |
|||
# yield f"data: **程序出现异常**\n\n" # 发送完成信号 |
|||
|
|||
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", |
|||
for i in checkRepeatText(filename, userId, outLog): |
|||
id += 1 |
|||
if await request.is_disconnected(): |
|||
yield { |
|||
"id": f"{id}", |
|||
"event": "checkRepeatText", |
|||
"data": "checkRepeatText连接已中断" |
|||
} |
|||
return Response(generate_checkRepeatText(filename,userId), headers=headers) |
|||
break |
|||
yield { |
|||
"id": f"{id}", |
|||
"event": "checkRepeatText", |
|||
"data": i |
|||
} |
|||
|
|||
g = generate_checkRepeatText(filename, userId, request) |
|||
return EventSourceResponse(g) |
|||
|
|||
|
|||
@app.route('/sse/checkPlaceName', methods=['GET']) |
|||
def checkPlaceNameWebSse(): |
|||
filename = request.args.get('filename') |
|||
userId = request.args.get("userId") |
|||
def generate_checkPlaceName(filename,userId): |
|||
@app.get('/sse/checkPlaceName') |
|||
def checkPlaceNameWebSse(filename, userId, request: Request): |
|||
async def generate_checkPlaceName(filename, userId, request: Request): |
|||
id = 0 |
|||
for i in checkPlaceName(filename,userId): |
|||
yield f"id: {id + 1}\n" |
|||
yield f"event: checkPlaceName\n" |
|||
yield f"data: {i}\n\n" # 发送完成信号 |
|||
|
|||
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", |
|||
global outLog |
|||
for i in checkPlaceName(filename, userId, outLog): |
|||
id += 1 |
|||
if await request.is_disconnected(): |
|||
yield { |
|||
"id": f"{id}", |
|||
"event": "checkPlaceName", |
|||
"data": "checkPlaceName连接已中断" |
|||
} |
|||
return Response(generate_checkPlaceName(filename,userId), headers=headers) |
|||
break |
|||
yield { |
|||
"id": f"{id}", |
|||
"event": "checkPlaceName", |
|||
"data": i |
|||
} |
|||
|
|||
g = generate_checkPlaceName(filename, userId, request) |
|||
return EventSourceResponse(g) |
|||
|
|||
|
|||
@app.route('/sse/checkCompanyName', methods=['GET']) |
|||
def checkCompanyNameWebSse(): |
|||
filename = request.args.get('filename') |
|||
userId = request.args.get("userId") |
|||
def generate_checkCompanyName(filename,userId): |
|||
@app.get('/sse/checkCompanyName') |
|||
def checkCompanyNameWebSse(filename, userId, request: Request): |
|||
async def generate_checkCompanyName(filename, userId, request: Request): |
|||
id = 0 |
|||
for i in checkCompanyName(filename,userId): |
|||
yield f"id: {id + 1}\n" |
|||
yield f"event: checkCompanyName\n" |
|||
yield f"data: {i}\n\n" # 发送完成信号 |
|||
|
|||
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", |
|||
global outLog |
|||
for i in checkCompanyName(filename, userId, outLog): |
|||
id += 1 |
|||
if await request.is_disconnected(): |
|||
yield { |
|||
"id": f"{id}", |
|||
"event": "checkCompanyName", |
|||
"data": "checkCompanyName连接已中断" |
|||
} |
|||
return Response(generate_checkCompanyName(filename,userId), headers=headers) |
|||
break |
|||
yield { |
|||
"id": f"{id}", |
|||
"event": "checkCompanyName", |
|||
"data": i |
|||
} |
|||
|
|||
g = generate_checkCompanyName(filename, userId, request) |
|||
return EventSourceResponse(g) |
|||
|
|||
|
|||
@app.route('/sse/checkDocumentErrorWeb', methods=['GET']) |
|||
def checkDocumentErrorWebSse(): |
|||
filename = request.args.get('filename') |
|||
userId = request.args.get("userId") |
|||
def generate_checkDocumentError(filename,userId): |
|||
@app.get('/sse/checkDocumentErrorWeb') |
|||
def checkDocumentErrorWebSse(filename, userId, request: Request): |
|||
async def generate_checkDocumentError(filename, userId, request: Request): |
|||
id = 0 |
|||
for i in checkDocumentError(filename,userId): |
|||
yield f"id: {id + 1}\n" |
|||
yield f"event: checkDocumentError\n" |
|||
yield f"data: {i}\n\n" # 发送完成信号 |
|||
|
|||
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", |
|||
global outLog |
|||
for i in checkDocumentError(filename, userId, outLog): |
|||
id += 1 |
|||
if await request.is_disconnected(): |
|||
yield { |
|||
"id": f"{id}", |
|||
"event": "checkDocumentError", |
|||
"data": "checkDocumentError连接已中断" |
|||
} |
|||
break |
|||
yield { |
|||
"id": f"{id}", |
|||
"event": "checkDocumentError", |
|||
"data": i |
|||
} |
|||
return Response(generate_checkDocumentError(filename,userId), headers=headers) |
|||
|
|||
g = generate_checkDocumentError(filename, userId, request) |
|||
return EventSourceResponse(g) |
|||
|
|||
@app.route('/sse/checkTitleName', methods=['GET']) |
|||
def checkTitleNameWebSse(): |
|||
filename = request.args.get('filename') |
|||
userId = request.args.get("userId") |
|||
def generate_checkTitleName(filename,userId): |
|||
|
|||
@app.get('/sse/checkTitleName') |
|||
def checkTitleNameWebSse(filename, userId, request: Request): |
|||
async def generate_checkTitleName(filename, userId, request: Request): |
|||
id = 0 |
|||
for i in checkTitleName(filename,userId): |
|||
yield f"id: {id + 1}\n" |
|||
yield f"event: checkTitleName\n" |
|||
yield f"data: {i}\n\n" # 发送完成信号 |
|||
|
|||
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", |
|||
global outLog |
|||
for i in checkTitleName(filename, userId, outLog): |
|||
id += 1 |
|||
if await request.is_disconnected(): |
|||
yield { |
|||
"id": f"{id}", |
|||
"event": "checkTitleName", |
|||
"data": "checkTitleName连接已中断" |
|||
} |
|||
break |
|||
yield { |
|||
"id": f"{id}", |
|||
"event": "checkTitleName", |
|||
"data": i |
|||
} |
|||
return Response(generate_checkTitleName(filename,userId), headers=headers) |
|||
|
|||
@app.route('/sse/getLog', methods=['GET']) |
|||
def getlog(): |
|||
userId = request.args.get("userId") |
|||
def generate_getLog(userId): |
|||
time.sleep(1) |
|||
g = generate_checkTitleName(filename, userId, request) |
|||
return EventSourceResponse(g) |
|||
|
|||
|
|||
@app.get("/sse/getLog") |
|||
# @app.route('/sse/getLog', methods=['GET']) |
|||
async def getlog(userId, request: Request): |
|||
# userId = request.args.get("userId") |
|||
async def generate_getLog(userId): |
|||
id = 0 |
|||
global outLog |
|||
await asyncio.sleep(5) |
|||
while True: |
|||
if outLog.is_done(userId): |
|||
isbreak = outLog.is_done(userId) |
|||
if isbreak: |
|||
break # 完成了 |
|||
text = outLog.get_queueData(userId) |
|||
if await request.is_disconnected(): |
|||
yield { |
|||
"id": f"{id}", |
|||
"event": "checkTitleName", |
|||
"data": "checkTitleName连接已中断" |
|||
} |
|||
break |
|||
q = outLog.get_queueData(userId) |
|||
if q: |
|||
if text: |
|||
id += 1 |
|||
text = q.pop(0) |
|||
yield f"id: {id}\n" |
|||
yield f"event: getlog\n" |
|||
yield f"data: {text}\n\n" # 发送完成信号 |
|||
yield f"id: {id}\n" |
|||
yield f"event: getlog\n" |
|||
yield f"data: 任务结束！！！！！\n\n" # 发送完成信号 |
|||
outLog.del_queue(userId) |
|||
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", |
|||
yield { |
|||
"id": id, |
|||
"event": "getlog", |
|||
"data": text |
|||
} |
|||
# yield f"id: {id}\n" |
|||
# yield f"event: getlog\n" |
|||
# yield f"data: {text}\n\n" # 发送完成信号 |
|||
# yield f"id: {id}\n" |
|||
# yield f"event: getlog\n" |
|||
# yield f"data: 任务结束！！！！！\n\n" # 发送完成信号 |
|||
yield { |
|||
"id": id, |
|||
"event": "getlog", |
|||
"data": "任务结束！！！！" |
|||
} |
|||
return Response(generate_getLog(userId), headers=headers) |
|||
outLog.del_queue(userId) |
|||
|
|||
# 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", |
|||
# } |
|||
g = generate_getLog(userId) |
|||
return EventSourceResponse(g) |
|||
# return Response(generate_getLog(userId), headers=headers) |
|||
|
|||
|
|||
if __name__ == '__main__': |
|||
app.run(host="0.0.0.0", port=80) |
|||
# app.run(host="0.0.0.0", port=80,threaded=True) |
|||
# uvicorn.run(app='main:app', host="0.0.0.0", port=80,workers=1) |
|||
app.run() |
|||
|
|||
Loading…
Reference in new issue