init
This commit is contained in:
CakeCN
214
Html/runs/run_20241126-192000_3qsnbb/code/app.py
Normal file
214
Html/runs/run_20241126-192000_3qsnbb/code/app.py
Normal file
@@ -0,0 +1,214 @@
|
||||
from flask import Flask, session, request, jsonify, render_template, send_from_directory
|
||||
from flask_cors import CORS
|
||||
from flask_socketio import SocketIO, join_room, emit
|
||||
import markdown
|
||||
import os
|
||||
import uuid
|
||||
import shutil
|
||||
import sys
|
||||
import json
|
||||
parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))
|
||||
sys.path.insert(0, parent_dir)
|
||||
current_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '.'))
|
||||
sys.path.insert(0, current_dir)
|
||||
from AlgoriAgent.projects.algoriAgent.agent_manager import AgentManager
|
||||
|
||||
app = Flask(__name__)
|
||||
|
||||
app.secret_key = 'cakebaker'
|
||||
# 配置 CORS
|
||||
CORS(app, resources={r"/*": {"origins": "http://localhost:9888"}}, supports_credentials=True)
|
||||
|
||||
userid_recorder = {} # user_id&path -> session['user_id']
|
||||
|
||||
'''
|
||||
Backboard
|
||||
'''
|
||||
from backboardManager import BackBoardManager, Backboard
|
||||
|
||||
backboard_manager = BackBoardManager()
|
||||
|
||||
@app.route('/backboard')
|
||||
def backboard():
|
||||
|
||||
return jsonify({'status': 'success'})
|
||||
|
||||
|
||||
def realtime_response(config, realtime_action):
|
||||
user_id = config['user_id']
|
||||
folder_path = config['path']
|
||||
if backboard_manager.get_backboard(user_id) == None:
|
||||
backboard_manager.add_backboard(user_id, folder_path)
|
||||
bb = backboard_manager.get_backboard(user_id)
|
||||
assert type(bb) == Backboard
|
||||
|
||||
bb.add_history(realtime_action)
|
||||
|
||||
if realtime_action['type'] == 'workspaceFolders':
|
||||
pass
|
||||
# bb.file_tree = realtime_action['file_tree']
|
||||
|
||||
if realtime_action['type'] == 'activeFile':
|
||||
file_path = realtime_action['filePath']
|
||||
assert type(file_path) == str
|
||||
file_path = "".join(file_path.split(f'/{user_id}/{folder_path}/')[1:])
|
||||
bb.active_file_path = file_path
|
||||
with open(f"../{user_id}/{folder_path}/{file_path}") as f:
|
||||
bb.active_file_content = f.read()
|
||||
|
||||
if realtime_action['type'] == 'paste':
|
||||
file_path = realtime_action['filePath']
|
||||
file_path = "".join(file_path.split(f'/{user_id}/{folder_path}/')[1:])
|
||||
bb.pasted_file_path = file_path
|
||||
bb.pasted_content = realtime_action['content']
|
||||
|
||||
|
||||
print("config"+str(config))
|
||||
print("realtime_action"+str(realtime_action))
|
||||
pass
|
||||
|
||||
|
||||
'''
|
||||
Agent and Chat
|
||||
'''
|
||||
|
||||
socketio = SocketIO(app, cors_allowed_origins="*")
|
||||
|
||||
agent_manager = AgentManager()
|
||||
|
||||
@socketio.on('connect')
|
||||
def handle_connect():
|
||||
user = request.args.get('username')
|
||||
folder_name = request.args.get('folder')
|
||||
user_id = userid_recorder[user+'&'+folder_name]
|
||||
session['user_id'] = user_id
|
||||
print(f'User connected with session user_id: {user_id}')
|
||||
# 从markdown、markdown_prompts、score_prompts中各读取数据folder.md并进行new_agent
|
||||
with open(f'books/markdown/{folder_name}.md','r') as fmd,\
|
||||
open(f'books/markdown_prompts/{folder_name}.md','r')as fmdp,\
|
||||
open(f'books/score_prompts/{folder_name}.md','r') as fsp:
|
||||
markdown = fmd.read()
|
||||
markdown_prompts = fmdp.read()
|
||||
score_prompts = fsp.read()
|
||||
print('-----------------------')
|
||||
print(user_id)
|
||||
user_id, agent = agent_manager.new_agent(markdown, markdown_prompts, score_prompts, id=user_id)
|
||||
join_room(user_id) # 将该用户加入以user_id为名的room
|
||||
|
||||
|
||||
# 当接收到前端的消息时，触发此事件
|
||||
@socketio.on('send_message')
|
||||
def handle_message(data):
|
||||
print(f"Message from client: {data}")
|
||||
id = session.get('user_id')
|
||||
# 构造回复消息
|
||||
res = agent_manager.invoke(id, data)
|
||||
reply = f"AI: {res.content['speak']}"
|
||||
# 将回复发送回前端
|
||||
emit('receive_message', reply)
|
||||
|
||||
|
||||
'''
|
||||
Markdown to HTML
|
||||
'''
|
||||
# 配置文件路径
|
||||
MARKDOWN_DIR = 'books/markdown' # 存放 markdown 文件的文件夹
|
||||
STATIC_DIR = 'static' # 存放生成的 HTML 和图片资源的文件夹
|
||||
IMAGE_DIR = 'image' # 图片资源相对于 markdown 的位置
|
||||
|
||||
|
||||
|
||||
@app.route('/<filename>-markdown', methods=['GET'])
|
||||
def convert_md(filename):
|
||||
md_file_path = os.path.join(MARKDOWN_DIR, f'{filename}.md')
|
||||
|
||||
if not os.path.exists(md_file_path):
|
||||
return jsonify({'error': 'Markdown file not found'}), 404
|
||||
|
||||
# 读取 markdown 文件内容
|
||||
with open(md_file_path, 'r', encoding='utf-8') as file:
|
||||
md_content = file.read()
|
||||
|
||||
# 将 markdown 转换为 HTML
|
||||
html_content = markdown.markdown(md_content)
|
||||
# 插入 CSS 样式，限制图片最大宽度为 100%，高度自动调整
|
||||
html_with_styles = f"""
|
||||
<html>
|
||||
<head>
|
||||
<style>
|
||||
img {{
|
||||
max-width: 100%;
|
||||
height: auto;
|
||||
}}
|
||||
</style>
|
||||
<link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/github-markdown-css/5.2.0/github-markdown.min.css">
|
||||
</head>
|
||||
<body>
|
||||
{html_content}
|
||||
</body>
|
||||
</html>
|
||||
"""
|
||||
# 保存 HTML 文件到 static 文件夹
|
||||
html_output_path = os.path.join(STATIC_DIR, f'{filename}.html')
|
||||
with open(html_output_path, 'w', encoding='utf-8') as html_file:
|
||||
html_file.write(html_with_styles)
|
||||
|
||||
# 处理图片资源，将 image/xxx 文件夹拷贝到 static/image/xxx
|
||||
image_source_dir = os.path.join(MARKDOWN_DIR, IMAGE_DIR, filename)
|
||||
image_target_dir = os.path.join(STATIC_DIR, IMAGE_DIR, filename)
|
||||
print(f"Copying image resources from {image_source_dir} to {image_target_dir}")
|
||||
if os.path.exists(image_source_dir):
|
||||
# 确保 static/image/xxx 目录存在
|
||||
if not os.path.exists(image_target_dir):
|
||||
os.makedirs(image_target_dir)
|
||||
|
||||
# 拷贝图片资源
|
||||
for image_file in os.listdir(image_source_dir):
|
||||
full_image_file_path = os.path.join(image_source_dir, image_file)
|
||||
if os.path.isfile(full_image_file_path):
|
||||
shutil.copy(full_image_file_path, image_target_dir)
|
||||
|
||||
return jsonify({'html_url': f'/static/{filename}.html'})
|
||||
|
||||
# 静态文件（HTML 和图片资源）的访问
|
||||
@app.route('/static/<path:filename>')
|
||||
def serve_static(filename):
|
||||
return send_from_directory(STATIC_DIR, filename)
|
||||
|
||||
|
||||
'''
|
||||
Vscode
|
||||
'''
|
||||
@app.route('/desktop/<user_id>/<path>')
|
||||
def hello_world(user_id, path):
|
||||
if 'user_id' not in session:
|
||||
session['user_id'] = 'user_' + str(uuid.uuid4())
|
||||
userid_recorder[user_id+'&'+path] = session['user_id']
|
||||
|
||||
# 在上层 目录下创建一个名为 user_id_path 的文件夹
|
||||
path_dir = os.path.join(parent_dir, user_id, path)
|
||||
os.makedirs(path_dir, exist_ok=True)
|
||||
# 在此文件夹内部创建一个.config文件，并写入 user_id=user_id\n path=path
|
||||
config_path = os.path.join(path_dir, '.config')
|
||||
with open(config_path, 'w') as f:
|
||||
f.write(f"user_id={user_id}\npath={path}")
|
||||
return render_template('index.html', user_id=user_id, path=path)
|
||||
|
||||
|
||||
@app.route('/vscode_data', methods=['POST'])
|
||||
def vscode_data():
|
||||
data = request.json
|
||||
config = data['config']
|
||||
realtime_response(config,data)
|
||||
print(f"Received data from VSCode: {data}")
|
||||
return jsonify({"status": "success", "received": data})
|
||||
|
||||
@app.route('/get_session')
|
||||
def get_session():
|
||||
user_session = session.get('user_id', 'default_session')
|
||||
return jsonify({"session": user_session})
|
||||
|
||||
if __name__ == '__main__':
|
||||
|
||||
socketio.run(app, host='0.0.0.0', port=5500)
|
||||
|
||||
@@ -0,0 +1,76 @@
|
||||
|
||||
|
||||
|
||||
import time
|
||||
|
||||
|
||||
class BackBoardManager:
|
||||
def __init__(self):
|
||||
self.backboards = {} # user_id: Backboard
|
||||
|
||||
def add_backboard(self, user_id, folder):
|
||||
self.backboards[user_id] = Backboard(user_id, folder)
|
||||
|
||||
def get_backboard(self, user_id):
|
||||
if user_id not in self.backboards:
|
||||
return None
|
||||
return self.backboards[user_id]
|
||||
|
||||
|
||||
|
||||
|
||||
class Backboard:
|
||||
def __init__(self, user_id, folder):
|
||||
self.user_id = user_id
|
||||
self.folder = folder
|
||||
self.create_time = time.time()
|
||||
self.enter_chapter_time = time.time()
|
||||
self.history = []
|
||||
self.file_tree = "[]"
|
||||
self.active_file_path = ""
|
||||
self.active_file_content = ""
|
||||
self.pasted_file_path = ""
|
||||
self.pasted_content = ""
|
||||
|
||||
|
||||
def next_chapter(self):
|
||||
self.enter_chapter_time = time.time()
|
||||
|
||||
def get_info_prompt(self):
|
||||
|
||||
five_history = ""
|
||||
for h in self.history[-5:]:
|
||||
five_history += h['type'] + "\n" + h.get('filePath', '') + "\n\n"
|
||||
# return f"###Global Info:###\n\
|
||||
# User is {self.user_id}\n\
|
||||
# User's total study time is {self.get_deltatime_mmss(self.create_time, time.time())}\n\
|
||||
# User's current chapter study time is {self.get_deltatime_mmss(self.enter_chapter_time, time.time())}\n\
|
||||
# \n\n\
|
||||
# Activated file path: {self.active_file_path}\n\
|
||||
# Activated file content (current editing 10 lines): \n\
|
||||
# ```\n\
|
||||
# {self.active_file_content[-10:]}\n\
|
||||
# ```\n\
|
||||
# \n\n\
|
||||
# Last five action:\n\
|
||||
# {"\n".join(self.history[-5:])}\n\nFile tree: \n{self.file_tree}\n"
|
||||
|
||||
|
||||
def add_history(self, realtime_action):
|
||||
if(realtime_action['type']=='config'):return
|
||||
if len(self.history)!=0:
|
||||
if (self.history[-1]['type']=='edit'):
|
||||
if(self.history[-1]['filepath'] == realtime_action['filepath']):
|
||||
self.history[-1]['content'] = realtime_action['content']
|
||||
return
|
||||
self.history.append(realtime_action)
|
||||
|
||||
|
||||
def get_deltatime_mmss(self, start_time, end_time):
|
||||
elapsed_time = end_time - start_time
|
||||
hours = int(elapsed_time // 3600)
|
||||
minutes = int((elapsed_time % 3600) // 60)
|
||||
seconds = int(elapsed_time % 60)
|
||||
return f"{hours:02d}:{minutes:02d}:{seconds:02d}"
|
||||
|
||||
|
||||
Reference in New Issue
Block a user