# E:\AI_System\web_ui\server.py
import sys
import os
import logging
import json
import traceback # 添加traceback用于详细错误日志
from flask import Flask, jsonify, request, render_template, send_from_directory, Response
# 修复路径导入问题 - 确保项目根目录已添加到 Python 路径
base_dir = os.path.dirname
(os.path.dirname
(os.path.abspath
(__
file__)))
sys.path.insert
(0, base_dir)
# 添加必要的子目录到 Python 路径
sys.path.insert
(0, os.path.join
(base_dir, 'agent'))
sys.path.insert
(0, os.path.join
(base_dir, 'core'))
sys.path.insert
(0, os.path.join
(base_dir, 'utils'))
# 现在可以导入核心模块
try:
from core.config import system_config
from agent.autonomous_agent import AutonomousAgent
except ImportError as e:
# 添加更详细的导入错误处理
logging.error
(f"❌ 核心模块导入失败: {str
(e)}")
logging.error
("请检查以下路径是否存在:")
logging.error
(f"1. {os.path.join
(base_dir, 'core/config.py')}")
logging.error
(f"2. {os.path.join
(base_dir, 'agent/autonomous_agent.py')}")
# 设置默认值避免崩溃
class system_config:
LOG_LEVEL = logging.INFO
# 配置Flask应用
app = Flask
(__name__,
template_folder='templates',
static_folder='static')
# 配置日志 - 添加
文件日志记录
log_formatter = logging.Formatter
('%
(asctime)s - %
(name)s - %
(levelname)s - %
(message)s')
# 控制台日志
console_handler = logging.StreamHandler
()
console_handler.setFormatter
(log_formatter)
#
文件日志 - 保留30天日志
file_handler = logging.handlers.TimedRotating
FileHandler
(
'web_server.log', when='midnight', backupCount=30
)
file_handler.setFormatter
(log_formatter)
logger = logging.getLogger
('WebServer')
logger.setLevel
(system_config.LOG_LEVEL)
logger.addHandler
(console_handler)
logger.addHandler
(file_handler)
# 全局智能体实例
ai_agent = None
# 在应用启动时初始化智能体 - 添加重试机制
logger.info
("=" * 50)
logger.info
("正在初始化智能体核心模块...")
max_retries = 3
retry_delay = 2 # 秒
for attempt in range
(max_retries):
try:
ai_agent = AutonomousAgent
()
logger.info
("✅ 智能体核心模块初始化成功")
break
except Exception as e:
logger.error
(f"❌ 智能体初始化失败
(尝试 {attempt + 1}/{max_retries}): {str
(e)}")
if attempt < max_retries - 1:
logger.info
(f"⏳ {retry_delay}秒后重试...")
time.sleep
(retry_delay)
else:
logger.critical
("‼️ 智能体初始化彻底失败，部分功能将不可用")
# ========== 核心路由 - 完全保留原有功能 ==========
# 根路由 - 显示前端界面
@app.route
('/')
def index
():
"""首页路由 - 渲染前端界面"""
return render_template
('agent_interface.html')
# 静态
文件路由
@app.route
('/static/<path:
filename>')
def static_
files
(filename):
"""提供静态
文件"""
try:
return send_from_directory
(app.static_folder,
filename)
except Exception as e:
logger.error
(f"静态
文件服务失败: {
filename} - {str
(e)}")
return jsonify
({"error": "
文件未找到"}
), 404
# 系统状态检查 - 添加详细状态信息
@app.route
('/status')
def status
():
"""系统状态检查"""
if ai_agent:
try:
status_
data = ai_agent.get_status
()
# 添加服务器状态信息
status_
data.update
({
"server": {
"status": "running",
"uptime": time.time
() - start_time,
"version": "1.0.0"
}
})
return jsonify
(status_
data)
except Exception as e:
logger.error
(f"获取状态失败: {traceback.format_exc
()}")
return jsonify
({"error": "内部错误", "details": str
(e)}
), 500
return jsonify
({"status": "agent_not_initialized"}
), 503
# 聊天请求处理 - 优化流式响应
@app.route
('/chat', methods=['POST'])
def chat
():
"""处理聊天请求，支持流式响应"""
if ai_agent is None:
return jsonify
({"error": "Agent not initialized"}
), 503
try:
# 获取用户输入和请求参数
data = request.get_json
()
user_input =
data.get
('message', '')
user_id =
data.get
('user_id', 'default')
stream =
data.get
('stream', False) # 默认为非流式
# 记录请求信息
logger.info
(f"聊天请求: 用户={user_id}, 流式={stream}, 内容长度={len
(user_input)}")
# 非流式处理
if not stream:
response = ai_agent.process_input
(user_input, user_id)
return jsonify
({"response": response})
# 流式处理
(SSE) - 修复可能的生成器问题
def generate
():
try:
# 创建回调函数用于流式输出
def stream_callback
(chunk, is_final=False):
# 发送SSE格式的数据
event_
data = json.dumps
({'chunk': chunk, 'final': is_final})
yield f"
data: {event_
data}\n\n"
# 处理输入并获取响应生成器
response_gen = ai_agent.process_input_stream
(user_input, user_id, stream_callback)
# 流式传输响应
for chunk in response_gen:
yield chunk
except Exception as e:
logger.error
(f"流式响应生成失败: {traceback.format_exc
()}")
yield "
data: {\"error\": \"流式响应中断\"}\n\n"
yield "
data: {\"final\": true}\n\n"
# 返回流式响应
return Response
(generate
(), mimetype='text/event-stream')
except Exception as e:
logger.error
(f"聊天处理失败: {traceback.format_exc
()}")
return jsonify
({"error": "聊天处理失败", "details": str
(e)}
), 500
# 关机路由 - 添加详细日志
@app.route
('/shutdown', methods=['POST'])
def shutdown
():
"""处理关机请求"""
try:
data = request.get_json
()
delay =
data.get
('delay', 120) # 默认2分钟
logger.info
(f"收到关机请求，将在{delay}秒后关闭系统")
# 这里应该有实际的关机逻辑
# 例如: agent.shutdown
(delay=delay)
return jsonify
({
"message": f"系统将在{delay}秒后关闭",
"timestamp": time.time
(),
"shutdown_time": time.time
() + delay
})
except Exception as e:
logger.error
(f"关机请求失败: {traceback.format_exc
()}")
return jsonify
({"error": str
(e)}
), 500
# ========== 环境交互路由 - 完全保留原有功能 ==========
# 环境浏览器
@app.route
('/environment')
def environment_view
():
"""渲染环境浏览器界面"""
return render_template
('environment_view.html')
# 获取环境目录内容 - 添加路径验证
@app.route
('/api/
explore', methods=['GET'])
def
explore_directory
():
"""获取指定路径的目录内容"""
if not ai_agent:
return jsonify
({"error": "Agent not initialized"}
), 503
try:
path = request.args.get
('path', '')
logger.info
(f"探索目录: {path}")
# 如果路径为空，使用智能体的工作区根目录
if not path:
workspace_info = ai_agent.get_status
().get
('environment', {})
if workspace_info and 'workspace' in workspace_info:
path = workspace_info['workspace']
# 添加基本路径安全检查
if not path.startswith
(ai_agent.get_workspace_
root()):
logger.warning
(f"尝试访问受限路径: {path}")
return jsonify
({"error": "访问受限"}
), 403
# 调用智能体的环境探索方法
result = ai_agent.
explore_environment
(path)
return jsonify
(result)
except Exception as e:
logger.error
(f"目录探索失败: {traceback.format_exc
()}")
return jsonify
({"error": str
(e)}
), 500
# 获取家具数据 - 添加缓存支持
furniture_cache = {}
CACHE_DURATION = 3600 # 1小时
@app.route
('/api/furniture')
def get_furniture
():
"""获取指定房间的家具数据"""
try:
room = request.args.get
('room', 'workshop')
logger.info
(f"获取家具数据: 房间={room}")
# 检查缓存
current_time = time.time
()
if room in furniture_cache and current_time - furniture_cache[room]['timestamp'] < CACHE_DURATION:
logger.debug
(f"使用缓存家具数据: {room}")
return jsonify
(furniture_cache[room]['
data'])
# 这里简化实现，实际应从数据库获取
furniture_
data = {
"workshop": [
{"type": "desk", "position": {"x": 0, "y": -1.5, "z": -3}, "rotation": {"x": 0, "y": 0, "z": 0}},
{"type": "chair", "position": {"x": 0, "y": -1.5, "z": -1}, "rotation": {"x": 0, "y": 0, "z": 0}},
{"type": "bookshelf", "position": {"x": 3, "y": 0, "z": -3}, "rotation": {"x": 0, "y": 0, "z": 0}},
{"type": "computer", "position": {"x": 0.5, "y": -0.5, "z": -3.2}, "rotation": {"x": 0, "y": 0, "z": 0}}
],
"lounge": [
{"type": "sofa", "position": {"x": 0, "y": -1.5, "z": -2}, "rotation": {"x": 0, "y": 180, "z": 0}},
{"type": "coffee_table", "position": {"x": 0, "y": -1.5, "z": -3},
"rotation": {"x": 0, "y": 0, "z": 0}},
{"type": "tv", "position": {"x": 0, "y": 0.5, "z": -4}, "rotation": {"x": 0, "y": 0, "z": 0}}
],
"library": [
{"type": "bookshelf", "position": {"x": -3, "y": 0, "z": -3}, "rotation": {"x": 0, "y": 0, "z": 0}},
{"type": "bookshelf", "position": {"x": 0, "y": 0, "z": -3}, "rotation": {"x": 0, "y": 0, "z": 0}},
{"type": "bookshelf", "position": {"x": 3, "y": 0, "z": -3}, "rotation": {"x": 0, "y": 0, "z": 0}},
{"type": "reading_chair", "position": {"x": 0, "y": -1.5, "z": -1},
"rotation": {"x": 0, "y": 180, "z": 0}}
]
}
response_
data = {
"room": room,
"items": furniture_
data.get
(room, [])
}
# 更新缓存
furniture_cache[room] = {
'
data': response_
data,
'timestamp': current_time
}
return jsonify
(response_
data)
except Exception as e:
logger.error
(f"获取家具数据失败: {traceback.format_exc
()}")
return jsonify
({"error": str
(e)}
), 500
# 获取可用皮肤列表 - 添加版本信息
@app.route
('/api/skins')
def get_skins
():
"""获取可用皮肤列表"""
try:
# 这里简化实现，实际应从数据库获取
skins = [
{"id": "default", "name": "默认皮肤", "description": "简洁的蓝色主题", "version": "1.0"},
{"id": "midnight", "name": "午夜黑", "description": "深色主题，减少眼睛疲劳", "version": "1.2"},
{"id": "forest", "name": "森林绿", "description": "自然绿色主题，带来平静感", "version": "1.1"},
{"id": "tech", "name": "科技蓝", "description": "未来感十足的科技主题", "version": "2.0"},
{"id": "vintage", "name": "复古黄", "description": "复古纸张色调，怀旧风格", "version": "1.5"}
]
return jsonify
({
"skins": skins,
"last_updated": "2023-11-15",
"count": len
(skins)
})
except Exception as e:
logger.error
(f"获取皮肤列表失败: {traceback.format_exc
()}")
return jsonify
({"error": str
(e)}
), 500
# 更换皮肤 - 添加皮肤存在性检查
@app.route
('/api/change_skin', methods=['POST'])
def change_skin
():
"""更换智能体皮肤"""
if not ai_agent:
return jsonify
({"error": "Agent not initialized"}
), 503
try:
data = request.get_json
()
skin_name =
data.get
('skin')
if not skin_name:
return jsonify
({"error": "Missing skin parameter"}
), 400
# 检查皮肤是否存在
skins_response = get_skins
().get_json
()
skin_ids = [skin['id'] for skin in skins_response['skins']]
if skin_name not in skin_ids:
return jsonify
({"error": "无效的皮肤ID"}
), 400
# 调用智能体的换肤方法
result = ai_agent.execute_action
("change_skin", {"skin_name": skin_name})
if result.get
("success"):
logger.info
(f"皮肤更换成功: {skin_name}")
return jsonify
({"success": True, "message": f"皮肤已更换为 {skin_name}"})
else:
logger.warning
(f"换肤失败: {result.get
('message')}")
return jsonify
({"error": result.get
("message", "换肤失败")}
), 500
except Exception as e:
logger.error
(f"更换皮肤失败: {traceback.format_exc
()}")
return jsonify
({"error": str
(e)}
), 500
# 执行环境动作 - 添加动作白名单
ALLOWED_ACTIONS = ['create_directory', 'delete_path', 'move_
file', 'copy_
file', 'execute_command']
@app.route
('/api/execute', methods=['POST'])
def execute_action
():
"""执行环境动作"""
if not ai_agent:
return jsonify
({"error": "Agent not initialized"}
), 503
try:
data = request.get_json
()
action =
data.get
('action')
params =
data.get
('parameters', {})
if not action:
return jsonify
({"error": "Missing action parameter"}
), 400
# 检查动作是否在允许列表中
if action not in ALLOWED_ACTIONS:
logger.warning
(f"尝试执行未授权动作: {action}")
return jsonify
({"error": "未授权的动作"}
), 403
logger.info
(f"执行动作: {action}, 参数: {params}")
# 调用智能体的执行方法
result = ai_agent.execute_action
(action, params)
# 记录执行结果
if result.get
("success"):
logger.info
(f"动作执行成功: {action}")
else:
logger.warning
(f"动作执行失败: {action} - {result.get
('message')}")
return jsonify
(result)
except Exception as e:
logger.error
(f"动作执行失败: {action} - {traceback.format_exc
()}")
return jsonify
({"error": str
(e)}
), 500
# 环境状态路由 - 添加性能指标
@app.route
('/api/environment_status')
def environment_status
():
"""获取环境状态信息（包含可用动作列表）"""
if not ai_agent:
return jsonify
({"error": "Agent not initialized"}
), 503
try:
# 获取智能体状态中的环境部分
status = ai_agent.get_status
()
env_status = status.get
('environment', {})
# 获取探索历史
history = ai_agent.environment.get_exploration_history
()
# 获取可用动作列表
available_actions = ai_agent.get_available_actions
()
# 添加性能指标
import psutil
env_status['performance'] = {
"cpu_usage": psutil.cpu_percent
(),
"memory_usage": psutil.virtual_memory
().percent,
"disk_usage": psutil.disk_usage
('/').percent
}
return jsonify
({
"system": env_status,
"exploration_history": history,
"available_actions": available_actions,
"timestamp": time.time
()
})
except Exception as e:
logger.error
(f"获取环境状态失败: {traceback.format_exc
()}")
return jsonify
({"error": str
(e)}
), 500
# ========== 智能体管理路由 - 完全保留原有功能 ==========
# 获取智能体详细状态 - 添加任务统计
@app.route
('/api/agent_status')
def agent_status
():
"""获取智能体的详细状态（核心状态、环境状态、任务队列）"""
if not ai_agent:
return jsonify
({"error": "Agent not initialized"}
), 503
try:
# 获取核心状态
core_status = ai_agent.get_status
()
# 获取环境状态
env_status = ai_agent.get_environment_status
()
# 获取任务队列
task_queue = ai_agent.get_task_queue
()
# 添加任务统计
task_stats = {
"total": len
(task_queue
),
"pending": sum
(1 for t in task_queue if t['status'] == 'pending'
),
"running": sum
(1 for t in task_queue if t['status'] == 'running'
),
"completed": sum
(1 for t in task_queue if t['status'] == 'completed'
),
"failed": sum
(1 for t in task_queue if t['status'] == 'failed')
}
return jsonify
({
"core_status": core_status,
"environment_status": env_status,
"task_queue": task_queue,
"task_stats": task_stats,
"timestamp": time.time
()
})
except Exception as e:
logger.error
(f"获取智能体详细状态失败: {traceback.format_exc
()}")
return jsonify
({"error": str
(e)}
), 500
# 创建目录 - 添加路径规范化
@app.route
('/api/create_directory', methods=['POST'])
def create_directory
():
"""在指定路径创建新目录"""
if not ai_agent:
return jsonify
({"error": "Agent not initialized"}
), 503
try:
data = request.get_json
()
path =
data.get
('path')
if not path:
return jsonify
({"error": "Missing path parameter"}
), 400
# 规范化路径
path = os.path.normpath
(path)
logger.info
(f"创建目录: {path}")
# 调用智能体的目录创建方法
result = ai_agent.execute_action
("create_directory", {"path": path})
if result.get
("success"):
logger.info
(f"目录创建成功: {path}")
return jsonify
({"success": True, "message": f"目录 '{path}' 创建成功"})
else:
logger.warning
(f"目录创建失败: {path} - {result.get
('message')}")
return jsonify
({"error": result.get
("message", "目录创建失败")}
), 500
except Exception as e:
logger.error
(f"创建目录失败: {traceback.format_exc
()}")
return jsonify
({"error": str
(e)}
), 500
# 删除路径 - 添加二次确认
@app.route
('/api/delete_path', methods=['POST'])
def delete_path
():
"""删除指定路径（
文件或目录）"""
if not ai_agent:
return jsonify
({"error": "Agent not initialized"}
), 503
try:
data = request.get_json
()
path =
data.get
('path')
confirm =
data.get
('confirm', False) # 添加确认标志
if not path:
return jsonify
({"error": "Missing path parameter"}
), 400
# 重要操作需要二次确认
if not confirm:
return jsonify
({
"warning": "此操作不可逆，请确认",
"confirmation_required": True,
"path": path
}
), 202 # 202 Accepted
logger.warning
(f"删除路径: {path}")
# 调用智能体的删除方法
result = ai_agent.execute_action
("delete_path", {"path": path})
if result.get
("success"):
logger.info
(f"路径删除成功: {path}")
return jsonify
({"success": True, "message": f"路径 '{path}' 删除成功"})
else:
logger.error
(f"路径删除失败: {path} - {result.get
('message')}")
return jsonify
({"error": result.get
("message", "路径删除失败")}
), 500
except Exception as e:
logger.error
(f"删除路径失败: {traceback.format_exc
()}")
return jsonify
({"error": str
(e)}
), 500
# 获取当前任务列表 - 添加分页支持
@app.route
('/api/current_tasks')
def current_tasks
():
"""获取当前正在运行的任务列表"""
if not ai_agent:
return jsonify
({"error": "Agent not initialized"}
), 503
try:
page = int
(request.args.get
('page', 1))
per_page = int
(request.args.get
('per_page', 10))
tasks = ai_agent.get_current_tasks
()
# 分页处理
total_tasks = len
(tasks)
start_index =
(page - 1) * per_page
end_index = start_index + per_page
paginated_tasks = tasks[start_index:end_index]
return jsonify
({
"tasks": paginated_tasks,
"pagination": {
"page": page,
"per_page": per_page,
"total": total_tasks,
"total_pages":
(total_tasks + per_page - 1) // per_page
}
})
except Exception as e:
logger.error
(f"获取当前任务失败: {traceback.format_exc
()}")
return jsonify
({"error": str
(e)}
), 500
# ========== 新增辅助功能 ==========
# 健康检查端点
@app.route
('/health')
def health_check
():
"""系统健康检查"""
status = {
"status": "ok",
"agent_initialized": bool
(ai_agent
),
"timestamp": time.time
(),
"uptime": time.time
() - start_time
}
return jsonify
(status)
# 服务器信息端点
@app.route
('/server_info')
def server_info
():
"""获取服务器信息"""
import platform
return jsonify
({
"python_version": sys.version,
"platform": platform.platform
(),
"flask_version": Flask.__version__,
"system_encoding": sys.getdefaultencoding
(),
"start_time": start_time
})
if __name__ == '__main__':
start_time = time.time
()
logger.info
("=" * 50)
logger.info
("🚀 启动Web服务: http://0.0.0.0:8000")
try:
app.run
(host='0.0.0.0', port=8000, threaded=True)
except Exception as e:
logger.critical
(f"‼️ 服务器启动失败: {traceback.format_exc
()}")
在应用中使用统一配置
在 web_ui/server.py 中，改完的完整版 发我一下呗
本文介绍了解决Android DDMS连接真机后无法查看data/data文件夹的方法。主要原因是权限不足，可通过RE文件管理器调整权限为777，或者使用adb shell命令逐个设置文件夹权限。
扫一扫
1105
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
