157 lines
4.3 KiB
Python
157 lines
4.3 KiB
Python
"""
|
|
AI助手Web客户端 - Flask应用
|
|
100%还原docs/ai-assistant.html原型设计
|
|
"""
|
|
|
|
import os
|
|
import sys
|
|
import json
|
|
import logging
|
|
from flask import Flask, render_template, request, jsonify, Response
|
|
from flask_cors import CORS
|
|
|
|
# 添加父目录到路径，以便导入api_client
|
|
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
|
|
|
try:
|
|
from api_client import AIAssistantClient
|
|
from config import APPLICATIONS, AI_ASSISTANT_CONFIG
|
|
except ImportError:
|
|
# 如果导入失败，使用模拟数据
|
|
AIAssistantClient = None
|
|
APPLICATIONS = {}
|
|
AI_ASSISTANT_CONFIG = {}
|
|
|
|
# 配置日志
|
|
logging.basicConfig(
|
|
level=logging.INFO,
|
|
format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
|
|
)
|
|
logger = logging.getLogger(__name__)
|
|
|
|
# 创建Flask应用
|
|
app = Flask(__name__)
|
|
app.config['SECRET_KEY'] = 'ai-assistant-web-client-secret-key'
|
|
CORS(app) # 启用CORS
|
|
|
|
# 初始化API客户端
|
|
api_client = None
|
|
if AIAssistantClient:
|
|
try:
|
|
api_client = AIAssistantClient()
|
|
logger.info("API客户端初始化成功")
|
|
except Exception as e:
|
|
logger.error(f"API客户端初始化失败: {e}")
|
|
|
|
|
|
@app.route('/')
|
|
def index():
|
|
"""主页面"""
|
|
return render_template('index.html')
|
|
|
|
|
|
@app.route('/api/applications', methods=['GET'])
|
|
def get_applications():
|
|
"""获取应用列表"""
|
|
try:
|
|
if api_client:
|
|
apps = api_client.get_app_list()
|
|
return jsonify({
|
|
'code': 200,
|
|
'message': 'success',
|
|
'data': apps
|
|
})
|
|
else:
|
|
# 返回模拟数据
|
|
return jsonify({
|
|
'code': 200,
|
|
'message': 'success',
|
|
'data': list(APPLICATIONS.values()) if APPLICATIONS else []
|
|
})
|
|
except Exception as e:
|
|
logger.error(f"获取应用列表失败: {e}")
|
|
return jsonify({
|
|
'code': 500,
|
|
'message': str(e),
|
|
'data': []
|
|
}), 500
|
|
|
|
|
|
@app.route('/api/chat/send', methods=['POST'])
|
|
def send_message():
|
|
"""发送聊天消息"""
|
|
try:
|
|
data = request.get_json()
|
|
app_id = data.get('appId')
|
|
message = data.get('message')
|
|
chat_id = data.get('chatId')
|
|
stream = data.get('stream', False)
|
|
|
|
if not message:
|
|
return jsonify({
|
|
'code': 400,
|
|
'message': '消息内容不能为空',
|
|
'data': None
|
|
}), 400
|
|
|
|
if api_client and app_id:
|
|
# 调用真实API
|
|
response = api_client.send_message(
|
|
app_id=app_id,
|
|
message=message,
|
|
chat_id=chat_id,
|
|
stream=stream
|
|
)
|
|
|
|
if stream:
|
|
# 流式响应
|
|
def generate():
|
|
for chunk in response:
|
|
yield f"data: {json.dumps(chunk, ensure_ascii=False)}\n\n"
|
|
yield "data: [DONE]\n\n"
|
|
|
|
return Response(generate(), mimetype='text/event-stream')
|
|
else:
|
|
# 普通响应
|
|
return jsonify(response)
|
|
else:
|
|
# 返回模拟响应
|
|
return jsonify({
|
|
'code': 200,
|
|
'message': 'success',
|
|
'data': {
|
|
'answer': f'这是对"{message}"的模拟回复。实际使用时会调用真实的AI助手API。',
|
|
'chatId': chat_id or 'mock_chat_id_001',
|
|
'messageId': 'mock_message_id_001'
|
|
}
|
|
})
|
|
except Exception as e:
|
|
logger.error(f"发送消息失败: {e}")
|
|
return jsonify({
|
|
'code': 500,
|
|
'message': str(e),
|
|
'data': None
|
|
}), 500
|
|
|
|
|
|
@app.route('/api/health', methods=['GET'])
|
|
def health_check():
|
|
"""健康检查"""
|
|
return jsonify({
|
|
'code': 200,
|
|
'message': 'OK',
|
|
'data': {
|
|
'status': 'healthy',
|
|
'api_client_available': api_client is not None
|
|
}
|
|
})
|
|
|
|
|
|
if __name__ == '__main__':
|
|
port = int(os.environ.get('PORT', 5000))
|
|
debug = os.environ.get('DEBUG', 'True').lower() == 'true'
|
|
|
|
logger.info(f"启动AI助手Web客户端，端口: {port}, 调试模式: {debug}")
|
|
app.run(host='0.0.0.0', port=port, debug=debug)
|
|
|