# E:\AI_System\web_ui\server.py
import sys
import os
import time
import logging
import json
import traceback
import threading
import platform
import psutil
import datetime
from pathlib import Path
from flask import Flask, jsonify, request, render_template
from logging.handlers import TimedRotatingFileHandler
from flask_socketio import SocketIO, emit
# 在server.py中添加健康检查路由
from flask import jsonify
@app.route('/system/health')
def health_check():
"""系统健康检查接口"""
components = {
'ai_core': True, # 实际应替换为状态检测函数
'hardware_manager': True,
'scheduler': True,
'environment': True,
'evolution_monitor': True
}
status = all(components.values())
return jsonify(
status="running" if status else "degraded",
components=components,
timestamp=datetime.now().isoformat()
)
# ========== 配置系统 ==========
class SystemConfig:
def __init__(self):
self.BASE_DIR = Path(__file__).resolve().parent.parent
self.HOST = '127.0.0.1'
self.PORT = 5000
self.LOG_LEVEL = 'INFO'
self.SECRET_KEY = 'your_secret_key_here'
self.DEBUG = True
self.USE_GPU = False
self.DEFAULT_MODEL = 'gpt-3.5-turbo'
# 目录配置
self.LOG_DIR = self.BASE_DIR / 'logs'
self.LOG_DIR.mkdir(parents=True, exist_ok=True)
self.CONFIG_DIR = self.BASE_DIR / 'config'
self.CONFIG_DIR.mkdir(parents=True, exist_ok=True)
self.AGENT_PATH = self.BASE_DIR / 'agent'
self.MODEL_CACHE_DIR = self.BASE_DIR / 'model_cache'
self.MODEL_CACHE_DIR.mkdir(parents=True, exist_ok=True)
def __str__(self):
return f"SystemConfig(HOST={self.HOST}, PORT={self.PORT})"
config = SystemConfig()
# ========== 全局协调器 ==========
coordinator = None
def register_coordinator(coord):
"""注册意识系统协调器"""
global coordinator
coordinator = coord
if coordinator and hasattr(coordinator, 'connect_to_ui'):
coordinator.connect_to_ui(update_ui)
def update_ui(event):
"""更新UI事件处理"""
if 'socketio' in globals():
socketio.emit('system_event', event)
# ========== 初始化日志系统 ==========
def setup_logger():
"""配置全局日志系统"""
logger = logging.getLogger('WebServer')
logger.setLevel(getattr(logging, config.LOG_LEVEL.upper(), logging.INFO))
# 日志格式
log_formatter = logging.Formatter(
'%(asctime)s - %(name)s - %(levelname)s - %(message)s',
datefmt='%Y-%m-%d %H:%M:%S'
)
# 文件日志处理器 (每天轮换，保留30天)
file_handler = TimedRotatingFileHandler(
config.LOG_DIR / 'web_server.log',
when='midnight',
backupCount=30,
encoding='utf-8'
)
file_handler.setFormatter(log_formatter)
logger.addHandler(file_handler)
# 控制台日志处理器
console_handler = logging.StreamHandler()
console_handler.setFormatter(log_formatter)
logger.addHandler(console_handler)
# 安全日志处理装饰器
def safe_logger(func):
def wrapper(*args, **kwargs):
try:
return func(*args, **kwargs)
except UnicodeEncodeError:
new_args = []
for arg in args:
if isinstance(arg, str):
new_args.append(arg.encode('ascii', 'ignore').decode('ascii'))
else:
new_args.append(arg)
return func(*new_args, **kwargs)
return wrapper
# 应用安全日志处理
for level in ['debug', 'info', 'warning', 'error', 'critical']:
setattr(logger, level, safe_logger(getattr(logger, level)))
return logger
# 初始化日志
logger = setup_logger()
# ========== 系统初始化 ==========
class SystemInitializer:
"""负责初始化系统核心组件"""
def __init__(self):
self.base_dir = Path(__file__).resolve().parent.parent
self.ai_core = None
self.hardware_manager = None
self.life_scheduler = None
self.ai_agent = None
self.start_time = time.time()
self.environment_manager = None
def initialize_system_paths(self):
"""初始化系统路径"""
sys.path.insert(0, str(self.base_dir))
logger.info(f"项目根目录: {self.base_dir}")
sub_dirs = ['agent', 'core', 'utils', 'config', 'cognitive_arch', 'environment']
for sub_dir in sub_dirs:
full_path = self.base_dir / sub_dir
if full_path.exists():
sys.path.insert(0, str(full_path))
logger.info(f"添加路径: {full_path}")
else:
logger.warning(f"目录不存在: {full_path}")
def initialize_environment_manager(self):
"""初始化环境管理器"""
try:
# 环境管理器模拟实现
class EnvironmentManager:
def __init__(self, config):
self.config = config
self.state = {
'temperature': 22.5,
'humidity': 45.0,
'light_level': 75,
'objects': [],
'last_updated': datetime.datetime.now().isoformat()
}
def start(self):
logger.info("环境管理器已启动")
def get_state(self):
# 更新模拟数据
self.state['temperature'] = 20 + 5 * (time.time() % 10) / 10
self.state['humidity'] = 40 + 10 * (time.time() % 10) / 10
self.state['light_level'] = 70 + 10 * (time.time() % 10) / 10
self.state['last_updated'] = datetime.datetime.now().isoformat()
return self.state
def execute_action(self, action, params):
logger.info(f"执行环境动作: {action} 参数: {params}")
if action == "adjust_temperature":
self.state['temperature'] = params.get('value', 22.0)
return True
elif action == "adjust_light":
self.state['light_level'] = params.get('level', 70)
return True
return False
env_config = {'update_interval': 1.0, 'spatial': {'grid_size': 1.0}}
self.environment_manager = EnvironmentManager(env_config)
self.environment_manager.start()
logger.info("✅ 环境管理器初始化成功")
return self.environment_manager
except Exception as e:
logger.error(f"❌ 环境管理器初始化失败: {str(e)}")
logger.warning("⚠️ 环境交互功能将不可用")
return None
def initialize_ai_core(self):
"""初始化AI核心系统"""
class AICore:
def __init__(self, base_dir):
self.base_dir = Path(base_dir)
self.genetic_code = self.load_genetic_code()
self.physical_state = {
"health": 100,
"energy": 100,
"mood": "calm"
}
self.dependencies = self.scan_dependencies()
def load_genetic_code(self):
code_path = self.base_dir / "core" / "genetic_code.py"
try:
if not code_path.exists():
# 创建默认遗传代码文件
default_code = """# 默认遗传代码
class AICore:
def __init__(self):
self.version = "1.0.0"
self.capabilities = ["learning", "reasoning", "problem_solving"]
def evolve(self):
print("系统正在进化...")
"""
with open(code_path, "w", encoding="utf-8") as f:
f.write(default_code)
with open(code_path, "r", encoding="utf-8") as f:
return f.read()
except Exception as e:
logger.error(f"加载遗传代码失败: {str(e)}")
return "# 默认遗传代码\nclass AICore:\n pass"
def scan_dependencies(self):
return {
"nervous_system": "flask",
"memory": "sqlite",
"perception": "opencv",
"reasoning": "transformers"
}
def mutate(self, new_code):
try:
code_path = self.base_dir / "core" / "genetic_code.py"
with open(code_path, "w", encoding="utf-8") as f:
f.write(new_code)
return True, "核心代码更新成功，系统已进化"
except Exception as e:
return False, f"进化失败: {str(e)}"
def wear_dependency(self, dependency_name, version):
self.dependencies[dependency_name] = version
return f"已穿戴 {dependency_name}@{version}"
def get_state(self):
return {
"genetic_code_hash": hash(self.genetic_code),
"dependencies": self.dependencies,
"physical_state": self.physical_state,
"hardware_environment": self.get_hardware_info()
}
def get_hardware_info(self):
return {
"cpu": platform.processor(),
"cpu_cores": psutil.cpu_count(logical=False),
"cpu_threads": psutil.cpu_count(logical=True),
"memory_gb": round(psutil.virtual_memory().total / (1024 ** 3), 1),
"storage_gb": round(psutil.disk_usage('/').total / (1024 ** 3), 1),
"os": f"{platform.system()} {platform.release()}"
}
self.ai_core = AICore(self.base_dir)
logger.info("✅ AI核心系统初始化完成")
return self.ai_core
def initialize_hardware_manager(self):
"""初始化硬件管理器"""
try:
# 硬件管理器模拟实现
class HardwareManager:
def __init__(self):
self.available_hardware = {
"cpu": ["Intel i9-13900K", "AMD Ryzen 9 7950X", "Apple M2 Max"],
"gpu": ["NVIDIA RTX 4090", "AMD Radeon RX 7900 XTX", "Apple M2 GPU"],
"memory": [16, 32, 64, 128],
"storage": ["1TB SSD", "2TB SSD", "4TB SSD", "8TB HDD"],
"peripherals": ["4K Camera", "3D Scanner", "High-Fidelity Microphone"]
}
self.current_setup = {
"cpu": platform.processor(),
"gpu": "Integrated Graphics",
"memory": round(psutil.virtual_memory().total / (1024 ** 3), 1),
"storage": round(psutil.disk_usage('/').total / (1024 ** 3), 1)
}
def request_hardware(self, hardware_type, specification):
if hardware_type not in self.available_hardware:
return False, f"不支持硬件类型: {hardware_type}"
if specification not in self.available_hardware[hardware_type]:
return False, f"不支持的规格: {specification}"
self.current_setup[hardware_type] = specification
return True, f"已请求 {hardware_type}: {specification}。请管理员完成安装。"
def get_current_setup(self):
return self.current_setup
def get_performance_metrics(self):
return {
"cpu_usage": psutil.cpu_percent(),
"memory_usage": psutil.virtual_memory().percent,
"disk_usage": psutil.disk_usage('/').percent,
"cpu_temp": 45.0,
"gpu_temp": 55.0,
"network_io": {
"sent": psutil.net_io_counters().bytes_sent,
"received": psutil.net_io_counters().bytes_recv
},
"last_updated": datetime.datetime.now().isoformat()
}
self.hardware_manager = HardwareManager()
logger.info("✅ 硬件管理器初始化成功")
return self.hardware_manager
except Exception as e:
logger.error(f"❌ 硬件管理器初始化失败: {str(e)}")
logger.warning("⚠️ 使用内置简单硬件管理器")
return None
def initialize_life_scheduler(self):
"""初始化生活调度器"""
try:
# 生活调度器模拟实现
class LifeScheduler:
def __init__(self):
self.daily_schedule = {
"wake_up": "07:00",
"breakfast": "08:00",
"lunch": "12:30",
"dinner": "19:00",
"sleep": "23:00"
}
self.current_activity = "awake"
self.energy_level = 100
self.mood = "calm"
self.activity_log = []
def wake_up(self):
self.current_activity = "awake"
self.log_activity("醒来")
def have_meal(self, meal_type):
self.current_activity = f"eating_{meal_type}"
self.energy_level = min(100, self.energy_level + 20)
self.log_activity(f"用餐: {meal_type}")
def go_to_sleep(self):
self.current_activity = "sleeping"
self.log_activity("睡觉")
def log_activity(self, activity):
timestamp = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
self.activity_log.append(f"{timestamp} - {activity}")
# 保留最近的100条记录
if len(self.activity_log) > 100:
self.activity_log.pop(0)
def adjust_schedule(self, adjustments):
for activity, new_time in adjustments.items():
if activity in self.daily_schedule:
self.daily_schedule[activity] = new_time
def get_current_state(self):
return {
"current_activity": self.current_activity,
"next_scheduled": self._get_next_scheduled(),
"energy_level": self.energy_level,
"mood": self.mood
}
def get_recent_activities(self, count=10):
return self.activity_log[-count:]
def _get_next_scheduled(self):
now = datetime.datetime.now()
current_time = now.strftime("%H:%M")
# 找到下一个计划活动
schedule_times = sorted(
[(k, v) for k, v in self.daily_schedule.items()],
key=lambda x: x[1]
)
for activity, time_str in schedule_times:
if time_str > current_time:
return f"{activity} at {time_str}"
# 如果没有找到，返回第二天的第一个活动
return f"{schedule_times[0][0]} at {schedule_times[0][1]} tomorrow"
self.life_scheduler = LifeScheduler()
logger.info("✅ 生活调度器初始化成功")
# 启动生活系统后台线程
life_thread = threading.Thread(
target=self._update_life_status,
daemon=True,
name="LifeSystemThread"
)
life_thread.start()
logger.info("✅ 生活系统后台线程已启动")
return self.life_scheduler
except Exception as e:
logger.error(f"❌ 生活调度器初始化失败: {str(e)}")
return None
def _update_life_status(self):
logger.info("🚦 生活系统后台线程启动")
while True:
try:
now = datetime.datetime.now()
current_hour = now.hour
current_minute = now.minute
current_time = f"{current_hour:02d}:{current_minute:02d}"
# 根据时间更新活动状态
if 7 <= current_hour < 8 and self.life_scheduler.current_activity != "awake":
self.life_scheduler.wake_up()
elif 12 <= current_hour < 13 and self.life_scheduler.current_activity != "eating_lunch":
self.life_scheduler.have_meal("lunch")
elif 19 <= current_hour < 20 and self.life_scheduler.current_activity != "eating_dinner":
self.life_scheduler.have_meal("dinner")
elif (23 <= current_hour or current_hour < 6) and self.life_scheduler.current_activity != "sleeping":
self.life_scheduler.go_to_sleep()
# 自然能量消耗
if self.life_scheduler.current_activity != "sleeping":
self.life_scheduler.energy_level = max(0, self.life_scheduler.energy_level - 0.1)
time.sleep(60)
except Exception as e:
logger.error(f"生活系统更新失败: {str(e)}", exc_info=True)
time.sleep(300)
def initialize_ai_agent(self):
"""初始化AI智能体"""
try:
# AI智能体模拟实现
class AutonomousAgent:
def __init__(self, model_path, cache_dir, use_gpu, default_model):
self.model_name = default_model
self.cache_dir = cache_dir
self.use_gpu = use_gpu
self.conversation_history = {}
def process_input(self, user_input, user_id):
# 初始化用户对话历史
if user_id not in self.conversation_history:
self.conversation_history[user_id] = []
# 添加用户输入到历史
self.conversation_history[user_id].append({"role": "user", "content": user_input})
# 生成AI响应（模拟）
if "你好" in user_input or "hello" in user_input:
response = "你好！我是AI助手，有什么可以帮您的吗？"
elif "时间" in user_input:
response = f"现在是 {datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S')}"
elif "状态" in user_input:
response = "系统运行正常，所有组件都在线。"
else:
response = f"我已收到您的消息: '{user_input}'。这是一个模拟响应，实际系统中我会分析您的问题并提供专业解答。"
# 添加AI响应到历史
self.conversation_history[user_id].append({"role": "assistant", "content": response})
return response
def get_status(self):
return {
"model": self.model_name,
"cache_dir": str(self.cache_dir),
"use_gpu": self.use_gpu,
"active_conversations": len(self.conversation_history)
}
self.ai_agent = AutonomousAgent(
model_path=config.AGENT_PATH,
cache_dir=config.MODEL_CACHE_DIR,
use_gpu=config.USE_GPU,
default_model=config.DEFAULT_MODEL
)
logger.info("✅ AI智能体初始化成功")
return self.ai_agent
except Exception as e:
logger.error(f"❌ AI智能体初始化失败: {str(e)}")
return None
def start_evolution_monitor(self):
def monitor():
while True:
try:
cpu_usage = psutil.cpu_percent()
mem_usage = psutil.virtual_memory().percent
# 根据系统负载调整AI状态
if cpu_usage > 80:
self.ai_core.physical_state["energy"] = max(20, self.ai_core.physical_state["energy"] - 5)
self.ai_core.physical_state["mood"] = "strained"
elif cpu_usage < 30:
self.ai_core.physical_state["energy"] = min(100, self.ai_core.physical_state["energy"] + 2)
time.sleep(60)
except Exception as e:
logging.error(f"进化监控错误: {str(e)}")
time.sleep(300)
monitor_thread = threading.Thread(target=monitor, daemon=True)
monitor_thread.start()
logger.info("✅ 进化监控线程已启动")
def initialize_all(self):
logger.info("=" * 50)
logger.info("🚀 开始初始化AI系统")
logger.info("=" * 50)
self.initialize_system_paths()
self.initialize_ai_core()
self.initialize_hardware_manager()
self.initialize_life_scheduler()
self.initialize_ai_agent()
self.initialize_environment_manager()
self.start_evolution_monitor()
logger.info("✅ 所有系统组件初始化完成")
return {
"ai_core": self.ai_core,
"hardware_manager": self.hardware_manager,
"life_scheduler": self.life_scheduler,
"ai_agent": self.ai_agent,
"environment_manager": self.environment_manager
}
# ========== Flask应用工厂 ==========
def create_app():
app = Flask(
__name__,
template_folder='templates',
static_folder='static',
static_url_path='/static'
)
app.secret_key = config.SECRET_KEY
system_initializer = SystemInitializer()
components = system_initializer.initialize_all()
app.config['SYSTEM_COMPONENTS'] = components
app.config['START_TIME'] = system_initializer.start_time
app.config['BASE_DIR'] = system_initializer.base_dir
# 初始化SocketIO
socketio = SocketIO(app, cors_allowed_origins="*", async_mode='threading')
app.config['SOCKETIO'] = socketio
# 注册路由
register_routes(app)
register_error_handlers(app)
return app, socketio
# ========== 环境交互路由 ==========
def register_environment_routes(app):
@app.route('/environment')
def environment_view():
return render_template('environment_view.html')
@app.route('/api/environment/state', methods=['GET'])
def get_environment_state():
env_manager = app.config['SYSTEM_COMPONENTS'].get('environment_manager')
if not env_manager:
return jsonify({"success": False, "error": "环境管理器未初始化"}), 503
try:
state = env_manager.get_state()
return jsonify(state)
except Exception as e:
app.logger.error(f"获取环境状态失败: {traceback.format_exc()}")
return jsonify({"success": False, "error": str(e)}), 500
@app.route('/api/environment/action', methods=['POST'])
def execute_environment_action():
env_manager = app.config['SYSTEM_COMPONENTS'].get('environment_manager')
if not env_manager:
return jsonify({"success": False, "error": "环境管理器未初始化"}), 503
try:
data = request.json
action = data.get('action')
params = data.get('params', {})
if not action:
return jsonify({"success": False, "error": "缺少动作参数"}), 400
success = env_manager.execute_action(action, params)
return jsonify({"success": success, "action": action})
except Exception as e:
app.logger.error(f"执行环境动作失败: {traceback.format_exc()}")
return jsonify({"success": False, "error": str(e)}), 500
# ========== 环境状态广播 ==========
def setup_environment_broadcast(app):
socketio = app.config['SOCKETIO']
@socketio.on('connect', namespace='/environment')
def handle_environment_connect():
app.logger.info('客户端已连接环境WebSocket')
@socketio.on('disconnect', namespace='/environment')
def handle_environment_disconnect():
app.logger.info('客户端已断开环境WebSocket')
def broadcast_environment_state():
env_manager = app.config['SYSTEM_COMPONENTS'].get('environment_manager')
if not env_manager:
return
while True:
try:
state = env_manager.get_state()
socketio.emit('environment_update', state, namespace='/environment')
time.sleep(1)
except Exception as e:
app.logger.error(f"环境状态广播失败: {str(e)}")
time.sleep(5)
broadcast_thread = threading.Thread(
target=broadcast_environment_state,
daemon=True,
name="EnvironmentBroadcastThread"
)
broadcast_thread.start()
app.logger.info("✅ 环境状态广播线程已启动")
# ========== 路由注册 ==========
def register_routes(app):
register_environment_routes(app)
setup_environment_broadcast(app)
@app.route('/')
def index():
return render_template('agent_interface.html')
@app.route('/status')
def status():
try:
components = app.config['SYSTEM_COMPONENTS']
status_data = {
"server": {
"status": "running",
"uptime": time.time() - app.config['START_TIME'],
"version": "1.0.0",
"config": {
"host": config.HOST,
"port": config.PORT,
"log_level": config.LOG_LEVEL,
"default_model": config.DEFAULT_MODEL
}
},
"core": components['ai_core'].get_state(),
"hardware": components['hardware_manager'].get_current_setup()
}
if components['environment_manager']:
try:
status_data["environment"] = components['environment_manager'].get_state()
except Exception as e:
status_data["environment"] = {"error": str(e)}
if components['life_scheduler']:
try:
status_data["life_system"] = components['life_scheduler'].get_current_state()
except Exception as e:
status_data["life_system"] = {"error": str(e)}
if components['ai_agent']:
try:
status_data["agent"] = components['ai_agent'].get_status()
except Exception as e:
status_data["agent"] = {"error": str(e)}
return jsonify(status_data)
except Exception as e:
app.logger.error(f"获取状态失败: {traceback.format_exc()}")
return jsonify({"error": "内部错误", "details": str(e)}), 500
# 核心系统路由
@app.route('/api/core/state')
def get_core_state():
return jsonify(app.config['SYSTEM_COMPONENTS']['ai_core'].get_state())
@app.route('/api/core/mutate', methods=['POST'])
def mutate_core():
data = request.get_json()
new_code = data.get('genetic_code')
if not new_code:
return jsonify({"success": False, "error": "缺少遗传代码"}), 400
success, message = app.config['SYSTEM_COMPONENTS']['ai_core'].mutate(new_code)
return jsonify({"success": success, "message": message})
@app.route('/api/core/wear', methods=['POST'])
def wear_dependency():
data = request.get_json()
dep_name = data.get('dependency')
version = data.get('version', 'latest')
if not dep_name:
return jsonify({"success": False, "error": "缺少依赖名称"}), 400
result = app.config['SYSTEM_COMPONENTS']['ai_core'].wear_dependency(dep_name, version)
return jsonify({"success": True, "message": result})
# 硬件管理路由
@app.route('/api/hardware/catalog')
def get_hardware_catalog():
return jsonify(app.config['SYSTEM_COMPONENTS']['hardware_manager'].available_hardware)
@app.route('/api/hardware/request', methods=['POST'])
def request_hardware():
data = request.get_json()
hw_type = data.get('type')
spec = data.get('specification')
if not hw_type or not spec:
return jsonify({"success": False, "error": "缺少硬件类型或规格"}), 400
success, message = app.config['SYSTEM_COMPONENTS']['hardware_manager'].request_hardware(hw_type, spec)
return jsonify({"success": success, "message": message})
@app.route('/api/hardware/current')
def get_current_hardware():
return jsonify(app.config['SYSTEM_COMPONENTS']['hardware_manager'].get_current_setup())
# 生活系统路由
@app.route('/life')
def life_dashboard():
return render_template('life_dashboard.html')
@app.route('/api/life/status')
def get_life_status():
components = app.config['SYSTEM_COMPONENTS']
if not components['life_scheduler']:
return jsonify({"success": False, "error": "生活系统未初始化"}), 503
try:
current_state = components['life_scheduler'].get_current_state()
recent_activities = components['life_scheduler'].get_recent_activities(10)
return jsonify({
"success": True,
"current_activity": current_state.get("current_activity", "未知"),
"next_scheduled": current_state.get("next_scheduled", "未知"),
"recent_activities": recent_activities,
"energy_level": current_state.get("energy_level", 100),
"mood": current_state.get("mood", "平静")
})
except Exception as e:
app.logger.error(f"获取生活状态失败: {traceback.format_exc()}")
return jsonify({"success": False, "error": str(e)}), 500
@app.route('/adjust_schedule', methods=['POST'])
def adjust_schedule():
components = app.config['SYSTEM_COMPONENTS']
if not components['life_scheduler']:
return jsonify({"success": False, "error": "生活系统未初始化"}), 503
try:
data = request.json
adjustments = data.get("adjustments", {})
valid_activities = ["wake_up", "breakfast", "lunch", "dinner", "sleep"]
for activity, new_time in adjustments.items():
if activity not in valid_activities:
return jsonify({"success": False, "error": f"无效的活动类型: {activity}"}), 400
if not isinstance(new_time, str) or len(new_time) != 5 or new_time[2] != ':':
return jsonify({"success": False, "error": f"无效的时间格式: {new_time}"}), 400
components['life_scheduler'].adjust_schedule(adjustments)
return jsonify({
"success": True,
"message": "计划表已更新",
"new_schedule": components['life_scheduler'].daily_schedule
})
except Exception as e:
app.logger.error(f"调整作息时间失败: {traceback.format_exc()}")
return jsonify({"success": False, "error": str(e)}), 400
# 聊天路由
@app.route('/chat', methods=['POST'])
def chat():
components = app.config['SYSTEM_COMPONENTS']
if not components['ai_agent']:
return jsonify({"error": "Agent未初始化"}), 503
try:
data = request.get_json()
user_input = data.get('message', '')
user_id = data.get('user_id', 'default')
if not user_input:
return jsonify({"error": "消息内容不能为空"}), 400
app.logger.info(f"聊天请求: 用户={user_id}, 内容长度={len(user_input)}")
response = components['ai_agent'].process_input(user_input, user_id)
return jsonify({"response": response})
except Exception as e:
app.logger.error(f"聊天处理失败: {traceback.format_exc()}")
return jsonify({"error": "聊天处理失败", "details": str(e)}), 500
# 家具管理路由
furniture_cache = {}
CACHE_DURATION = 3600 # 1小时
@app.route('/api/furniture')
def get_furniture():
try:
room = request.args.get('room', 'workshop')
app.logger.info(f"获取家具数据: 房间={room}")
current_time = time.time()
if room in furniture_cache and current_time - furniture_cache[room]['timestamp'] < CACHE_DURATION:
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}}
],
"living_room": [
{"type": "sofa", "position": {"x": 0, "y": 0, "z": -2}, "rotation": {"x": 0, "y": 0, "z": 0}},
{"type": "tv", "position": {"x": 0, "y": 1.5, "z": -3}, "rotation": {"x": 0, "y": 0, "z": 0}}
],
"bedroom": [
{"type": "bed", "position": {"x": 0, "y": 0, "z": -3}, "rotation": {"x": 0, "y": 0, "z": 0}},
{"type": "nightstand", "position": {"x": 1.5, "y": 0, "z": -2.5},
"rotation": {"x": 0, "y": 0, "z": 0}}
]
}
furniture_cache[room] = {
'timestamp': current_time,
'data': furniture_data.get(room, [])
}
return jsonify(furniture_cache[room]['data'])
except Exception as e:
app.logger.error(f"获取家具数据失败: {traceback.format_exc()}")
return jsonify({"error": "内部错误", "details": str(e)}), 500
# ========== 错误处理器 ==========
def register_error_handlers(app):
@app.errorhandler(404)
def page_not_found(error):
app.logger.warning(f"404错误: {request.path}")
return jsonify({
"error": "资源未找到",
"path": request.path,
"method": request.method
}), 404
@app.errorhandler(500)
def internal_server_error(error):
app.logger.error(f"500错误: {str(error)}")
return jsonify({
"error": "服务器内部错误",
"message": "系统遇到意外错误，请稍后重试"
}), 500
@app.errorhandler(Exception)
def handle_general_exception(error):
app.logger.error(f"未处理异常: {traceback.format_exc()}")
return jsonify({
"error": "未处理的异常",
"type": type(error).__name__,
"message": str(error)
}), 500
# ========== WebSocket处理 ==========
def setup_websocket_handlers(socketio):
@socketio.on('connect')
def handle_connect():
logger.info('客户端已连接')
socketio.emit('system_status', {'status': 'ready'})
@socketio.on('disconnect')
def handle_disconnect():
logger.info('客户端已断开连接')
@socketio.on('user_message')
def handle_user_message(data):
user_id = data.get('user_id', 'guest')
message = data.get('message', '')
logger.info(f"收到来自 {user_id} 的消息: {message}")
# 处理消息逻辑
response = f"已收到您的消息: {message}"
# 如果有协调器，使用协调器处理
global coordinator
if coordinator:
try:
response = coordinator.process_message(message)
except Exception as e:
logger.error(f"协调器处理消息失败: {str(e)}")
socketio.emit('agent_response', {
'user_id': user_id,
'response': response
})
# ========== 主程序入口 ==========
if __name__ == '__main__':
try:
app, socketio = create_app()
# 设置WebSocket处理器
setup_websocket_handlers(socketio)
# 启动服务器
socketio.run(
app,
host=config.HOST,
port=config.PORT,
debug=config.DEBUG,
use_reloader=False
)
logger.info(f"服务器运行在 http://{config.HOST}:{config.PORT}")
except KeyboardInterrupt:
logger.info("服务器关闭")
except Exception as e:
logger.critical(f"服务器启动失败: {str(e)}")
logger.error(traceback.format_exc()) 帮我改好 发我完整版哦
扫一扫
8021
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
