戒酒的李白

Integrating the OpenAI API for in-depth comment analysis, with usability to be debugged.

from sqlalchemy import Column, Integer, String, Float, DateTime, Text, JSON
from sqlalchemy.ext.declarative import declarative_base
from datetime import datetime
Base = declarative_base()
class AIAnalysis(Base):
__tablename__ = 'ai_analysis'
id = Column(Integer, primary_key=True)
message_id = Column(Integer, nullable=False)
sentiment = Column(String(10), nullable=False)
sentiment_score = Column(Float, nullable=False)
keywords = Column(JSON, nullable=False)
key_points = Column(Text, nullable=False)
influence_analysis = Column(Text, nullable=False)
risk_level = Column(String(10), nullable=False)
created_at = Column(DateTime, default=datetime.now)
def to_dict(self):
return {
'id': self.id,
'message_id': self.message_id,
'sentiment': self.sentiment,
'sentiment_score': f"{self.sentiment_score:.2%}",
'keywords': self.keywords,
'key_points': self.key_points,
'influence': self.influence_analysis,
'risk_level': self.risk_level,
'analysis_time': self.created_at.strftime('%Y-%m-%d %H:%M:%S')
}
\ No newline at end of file
... ...
import openai
import json
from typing import List, Dict
import os
from datetime import datetime
from utils.logger import app_logger as logging
class AIAnalyzer:
def __init__(self):
# 从环境变量获取API密钥
self.api_key = os.getenv('OPENAI_API_KEY')
if not self.api_key:
raise ValueError("请设置OPENAI_API_KEY环境变量")
openai.api_key = self.api_key
# 系统提示词,限制AI的输出格式
self.system_prompt = """你是一个专业的舆情分析助手。你的任务是分析每条消息的情感倾向、关键词和潜在影响。
请严格按照以下JSON格式返回分析结果:
{
"analysis_results": [
{
"message_id": "消息ID",
"sentiment": "情感倾向 (积极/消极/中性)",
"sentiment_score": "情感分数 (0-1)",
"keywords": ["关键词1", "关键词2", "关键词3"],
"key_points": "核心观点概述",
"influence_analysis": "潜在影响分析",
"risk_level": "风险等级 (低/中/高)",
"timestamp": "分析时间戳"
}
]
}
请确保每个字段都有值,并保持JSON格式的一致性。"""
async def analyze_messages(self, messages: List[Dict]) -> List[Dict]:
"""分析一批消息并返回分析结果"""
try:
# 构建输入消息
formatted_messages = []
for msg in messages:
formatted_messages.append(f"消息ID: {msg['id']}\n内容: {msg['content']}")
messages_text = "\n---\n".join(formatted_messages)
# 调用OpenAI API
response = await openai.ChatCompletion.acreate(
model="gpt-3.5-turbo",
messages=[
{"role": "system", "content": self.system_prompt},
{"role": "user", "content": f"请分析以下消息:\n{messages_text}"}
],
temperature=0.3, # 降低随机性
max_tokens=2000,
n=1
)
# 解析返回结果
try:
result = json.loads(response.choices[0].message.content)
# 验证结果格式
if not isinstance(result, dict) or 'analysis_results' not in result:
raise ValueError("AI返回格式不正确")
return result['analysis_results']
except json.JSONDecodeError:
logging.error("AI返回结果解析失败")
return []
except Exception as e:
logging.error(f"AI分析过程出错: {e}")
return []
def format_analysis_for_display(self, analysis: Dict) -> Dict:
"""将分析结果格式化为前端显示格式"""
return {
'id': analysis['message_id'],
'sentiment': analysis['sentiment'],
'sentiment_score': f"{float(analysis['sentiment_score']):.2%}",
'keywords': ', '.join(analysis['keywords']),
'key_points': analysis['key_points'],
'influence': analysis['influence_analysis'],
'risk_level': analysis['risk_level'],
'analysis_time': datetime.fromtimestamp(
float(analysis['timestamp'])
).strftime('%Y-%m-%d %H:%M:%S')
}
# 创建全局AI分析器实例
ai_analyzer = AIAnalyzer()
\ No newline at end of file
... ...
... ... @@ -9,6 +9,11 @@ from utils.getTopicPageData import *
from utils.yuqingpredict import *
from utils.logger import app_logger as logging
from utils.cache_manager import prediction_cache
from utils.ai_analyzer import ai_analyzer
from models.ai_analysis import AIAnalysis
from sqlalchemy.orm import Session
from sqlalchemy import create_engine
import asyncio
import torch
from BCAT_front.predict import model_manager
... ... @@ -31,6 +36,11 @@ try:
except Exception as e:
logging.error(f"模型加载失败: {e}")
# 数据库配置
DATABASE_URL = "sqlite:///ai_analysis.db"
engine = create_engine(DATABASE_URL)
AIAnalysis.metadata.create_all(engine)
def predict_sentiment(text):
"""使用改进版模型预测单个文本的情感"""
try:
... ... @@ -294,3 +304,99 @@ def articleChar(id):
except Exception as e:
logging.error(f"获取文章详情时发生错误: {e}")
return render_template('error.html', error_message="加载文章详情失败")
@pb.route('/api/analyze_messages', methods=['POST'])
async def analyze_messages():
try:
# 获取最近50条消息
messages = getRecentMessages(50) # 需要实现这个函数
# 调用AI进行分析
analysis_results = await ai_analyzer.analyze_messages(messages)
# 保存到数据库
with Session(engine) as session:
for result in analysis_results:
analysis = AIAnalysis(
message_id=result['message_id'],
sentiment=result['sentiment'],
sentiment_score=float(result['sentiment_score']),
keywords=result['keywords'],
key_points=result['key_points'],
influence_analysis=result['influence_analysis'],
risk_level=result['risk_level']
)
session.add(analysis)
session.commit()
# 格式化结果用于显示
display_results = [
ai_analyzer.format_analysis_for_display(result)
for result in analysis_results
]
return jsonify({
'success': True,
'data': display_results
})
except Exception as e:
logging.error(f"AI分析过程出错: {e}")
return jsonify({
'success': False,
'error': str(e)
}), 500
@pb.route('/api/get_analysis/<int:message_id>')
def get_message_analysis(message_id):
"""获取特定消息的分析结果"""
try:
with Session(engine) as session:
analysis = session.query(AIAnalysis)\
.filter(AIAnalysis.message_id == message_id)\
.order_by(AIAnalysis.created_at.desc())\
.first()
if analysis:
return jsonify({
'success': True,
'data': analysis.to_dict()
})
else:
return jsonify({
'success': False,
'error': '未找到分析结果'
}), 404
except Exception as e:
logging.error(f"获取分析结果时出错: {e}")
return jsonify({
'success': False,
'error': str(e)
}), 500
def getRecentMessages(limit=50):
"""获取最近的消息"""
# 这里需要根据你的数据库结构实现具体的查询逻辑
messages = []
try:
# 示例查询逻辑
with Session(engine) as session:
results = session.execute(
"""
SELECT id, content
FROM comments
ORDER BY created_at DESC
LIMIT :limit
""",
{'limit': limit}
).fetchall()
messages = [
{'id': row[0], 'content': row[1]}
for row in results
]
except Exception as e:
logging.error(f"获取最近消息时出错: {e}")
return messages
... ...
... ... @@ -445,8 +445,157 @@
</div>
</div>
</div>
<!-- AI分析结果展示区域 -->
<div class="row">
<div class="col-lg-12">
<div class="card">
<div class="card-header d-flex justify-content-between">
<div class="header-title">
<h4 class="card-title">AI深度分析</h4>
</div>
<button class="btn btn-primary" onclick="requestAIAnalysis()">
开始AI分析
</button>
</div>
<div class="card-body">
<div id="ai-analysis-results" class="analysis-container">
<!-- 分析结果将在这里动态显示 -->
</div>
</div>
</div>
</div>
</div>
<!-- 添加必要的CSS样式 -->
<style>
.analysis-container {
max-height: 600px;
overflow-y: auto;
}
.analysis-card {
border: 1px solid #eee;
border-radius: 8px;
padding: 15px;
margin-bottom: 15px;
background-color: #fff;
box-shadow: 0 2px 4px rgba(0,0,0,0.1);
}
.analysis-card:hover {
box-shadow: 0 4px 8px rgba(0,0,0,0.15);
}
.risk-level {
padding: 4px 8px;
border-radius: 4px;
font-weight: bold;
}
.risk-low {
background-color: #e8f5e9;
color: #2e7d32;
}
.risk-medium {
background-color: #fff3e0;
color: #f57c00;
}
.risk-high {
background-color: #ffebee;
color: #c62828;
}
.keywords-container {
display: flex;
flex-wrap: wrap;
gap: 8px;
margin: 10px 0;
}
.keyword-tag {
background-color: #e3f2fd;
color: #1976d2;
padding: 4px 8px;
border-radius: 16px;
font-size: 0.9em;
}
</style>
<!-- 添加必要的JavaScript代码 -->
<script>
async function requestAIAnalysis() {
try {
const response = await fetch('/page/api/analyze_messages', {
method: 'POST',
headers: {
'Content-Type': 'application/json'
}
});
const result = await response.json();
if (result.success) {
displayAnalysisResults(result.data);
} else {
alert('分析失败: ' + result.error);
}
} catch (error) {
console.error('AI分析请求失败:', error);
alert('请求失败,请稍后重试');
}
}
function displayAnalysisResults(results) {
const container = document.getElementById('ai-analysis-results');
container.innerHTML = ''; // 清空现有结果
results.forEach(analysis => {
const card = document.createElement('div');
card.className = 'analysis-card';
const riskLevelClass =
analysis.risk_level === '高' ? 'risk-high' :
analysis.risk_level === '中' ? 'risk-medium' : 'risk-low';
card.innerHTML = `
<div class="d-flex justify-content-between align-items-center">
<h5 class="mb-2">消息ID: ${analysis.id}</h5>
<span class="risk-level ${riskLevelClass}">
风险等级: ${analysis.risk_level}
</span>
</div>
<div class="mb-2">
<strong>情感倾向:</strong> ${analysis.sentiment}
<span class="ml-2">(${analysis.sentiment_score})</span>
</div>
<div class="keywords-container">
${analysis.keywords.split(',').map(keyword =>
`<span class="keyword-tag">${keyword.trim()}</span>`
).join('')}
</div>
<div class="mb-2">
<strong>核心观点:</strong>
<p class="mb-1">${analysis.key_points}</p>
</div>
<div class="mb-2">
<strong>影响分析:</strong>
<p class="mb-1">${analysis.influence}</p>
</div>
<div class="text-muted">
分析时间: ${analysis.analysis_time}
</div>
`;
container.appendChild(card);
});
}
// 页面加载完成后自动请求一次AI分析
document.addEventListener('DOMContentLoaded', requestAIAnalysis);
</script>
{% endblock %}
{% block echarts %}
... ...