'
+ f'
AI助手:正在思考...'
+ )
+
+ # 自动滚动到底部
+ self.conversation_text.ensureCursorVisible()
+
+ def update_streaming_display(self):
+ """更新流式显示"""
+ if hasattr(self, 'current_streaming_content') and self.current_streaming_content:
+ # 使用更简单的方法：重新构建整个对话历史
+ self.rebuild_conversation_display()
+
+ # 自动滚动到底部
+ self.conversation_text.ensureCursorVisible()
+
+ def rebuild_conversation_display(self):
+ """重新构建对话显示"""
+ html_content = ""
+
+ for msg in self.conversation_history:
+ sender = msg["sender"]
+ message = msg["message"]
+ is_streaming = msg.get("streaming", False)
+
+ # 根据发送者设置不同的样式
+ if sender == "用户":
+ bg_color = "#e3f2fd" if self.current_theme == "light" else "#2d3e50"
+ text_color = "#000" if self.current_theme == "light" else "#fff"
+ elif sender == "AI助手":
+ bg_color = "#f5f5f5" if self.current_theme == "light" else "#3d3d3d"
+ text_color = "#000" if self.current_theme == "light" else "#fff"
+ else: # 系统消息
+ bg_color = "#fff3cd" if self.current_theme == "light" else "#5d4e00"
+ text_color = "#856404" if self.current_theme == "light" else "#ffd700"
+
+ # 格式化消息内容
+ formatted_message = message.replace('\n', '
') if message else "正在思考..."
+
+ html_content += f'''
+
+ {sender}:
+ {formatted_message}
+
+ '''
+
+ # 设置HTML内容
+ self.conversation_text.setHtml(html_content)
+
+ def call_deepseek_api_stream(self, message):
+ """调用DeepSeek API（流式版本）"""
+ url = "https://api.deepseek.com/v1/chat/completions"
+
+ headers = {
+ "Content-Type": "application/json",
+ "Authorization": f"Bearer {self.api_key}"
+ }
+
+ # 构建对话历史
+ messages = [{"role": "user", "content": message}]
+
+ data = {
+ "model": "deepseek-chat",
+ "messages": messages,
+ "stream": True, # 启用流式输出
+ "temperature": 0.7,
+ "max_tokens": 2000
+ }
+
+ try:
+ response = requests.post(url, headers=headers, json=data, stream=True, timeout=30)
+
+ if response.status_code == 200:
+ for line in response.iter_lines():
+ if line:
+ line = line.decode('utf-8')
+ if line.startswith('data: '):
+ data_str = line[6:] # 去掉 'data: ' 前缀
+ if data_str == '[DONE]':
+ break
+
+ try:
+ data_obj = json.loads(data_str)
+ if 'choices' in data_obj and len(data_obj['choices']) > 0:
+ delta = data_obj['choices'][0].get('delta', {})
+ if 'content' in delta:
+ content = delta['content']
+ self.current_streaming_content += content
+ except json.JSONDecodeError:
+ pass
+ else:
+ error_msg = f"API调用失败: {response.status_code} - {response.text}"
+ self.current_streaming_content = f"错误: {error_msg}"
+
+ except Exception as e:
+ self.current_streaming_content = f"请求失败: {str(e)}"
+
+ finally:
+ # 流式输出结束
+ self.is_streaming = False
+ if hasattr(self, 'streaming_timer'):
+ self.streaming_timer.stop()
+
+ # 更新对话历史
+ if hasattr(self, 'current_streaming_content') and self.current_streaming_content:
+ # 更新对话历史中的消息
+ for msg in self.conversation_history:
+ if msg.get('streaming') and msg.get('sender') == 'AI助手':
+ msg['message'] = self.current_streaming_content
+ msg['streaming'] = False
+ break
+
+ # 使用信号槽机制安全地更新UI
+ self.streaming_finished.emit()
+ else:
+ self.status_label.setText("接收失败")
+
+ def call_deepseek_api(self, message):
+ """调用DeepSeek API（非流式版本，保留作为备用）"""
+ url = "https://api.deepseek.com/v1/chat/completions"
+
+ headers = {
+ "Content-Type": "application/json",
+ "Authorization": f"Bearer {self.api_key}"
+ }
+
+ # 构建对话历史
+ messages = [{"role": "user", "content": message}]
+
+ data = {
+ "model": "deepseek-chat",
+ "messages": messages,
+ "stream": False,
+ "temperature": 0.7,
+ "max_tokens": 2000
+ }
+
+ response = requests.post(url, headers=headers, json=data, timeout=30)
+
+ if response.status_code == 200:
+ result = response.json()
+ return result["choices"][0]["message"]["content"]
+ else:
+ error_msg = f"API调用失败: {response.status_code} - {response.text}"
+ raise Exception(error_msg)
+
+ def add_message_to_conversation(self, sender, message):
+ """添加消息到对话显示区域"""
+ # 添加到对话历史
+ self.conversation_history.append({"sender": sender, "message": message})
+
+ # 使用新的对话显示系统
+ self.rebuild_conversation_display()
+
+ # 自动滚动到底部
+ self.conversation_text.ensureCursorVisible()
+
+ def on_streaming_finished(self):
+ """流式输出完成处理"""
+ # 安全地更新UI（在主线程中执行）
+ self.rebuild_conversation_display()
+ self.status_label.setText("消息接收完成")
+
+ def clear_conversation(self):
+ """清空对话历史"""
+ reply = QMessageBox.question(self, "确认", "确定要清空对话历史吗？",
+ QMessageBox.Yes | QMessageBox.No)
+
+ if reply == QMessageBox.Yes:
+ self.conversation_history.clear()
+ self.rebuild_conversation_display()
+ self.status_label.setText("对话已清空")
+
+ def manage_api_key(self):
+ """管理API密钥"""
+ self.show_api_key_dialog()
+
+ def closeEvent(self, event):
+ """关闭事件处理"""
+ # 发出关闭信号
+ self.closed.emit()
+
+ if self.parent:
+ # 通知父窗口对话窗口已关闭
+ if hasattr(self.parent, 'on_deepseek_dialog_closed'):
+ self.parent.on_deepseek_dialog_closed()
+ event.accept()
\ No newline at end of file
diff --git a/src/word_main_window.py b/src/word_main_window.py
index fab2880..ff65c45 100644
--- a/src/word_main_window.py
+++ b/src/word_main_window.py
@@ -93,6 +93,9 @@ class WordStyleMainWindow(QMainWindow):
self.learning_window = None # 学习模式窗口引用
self.sync_from_learning = False # 从学习模式同步内容的标记
+ # DeepSeek对话窗口引用
+ self.deepseek_dialog = None # DeepSeek对话窗口引用
+
# 统一文档内容管理
self.unified_document_content = "" # 统一文档内容
self.last_edit_mode = "typing" # 上次编辑模式
@@ -804,6 +807,12 @@ class WordStyleMainWindow(QMainWindow):
# 引用菜单
reference_menu = menubar.addMenu('引用(R)')
+ # DeepSeek AI对话功能
+ self.deepseek_dialog_action = QAction('DeepSeek AI对话', self)
+ self.deepseek_dialog_action.setShortcut('Ctrl+D')
+ self.deepseek_dialog_action.triggered.connect(self.open_deepseek_dialog)
+ reference_menu.addAction(self.deepseek_dialog_action)
+
# 邮件菜单
mail_menu = menubar.addMenu('邮件(M)')
@@ -2374,6 +2383,46 @@ class WordStyleMainWindow(QMainWindow):
self.status_bar.showMessage("学习模式窗口已关闭", 2000)
+ def open_deepseek_dialog(self):
+ """打开DeepSeek AI对话窗口"""
+ try:
+ from deepseek_dialog_window import DeepSeekDialogWindow
+
+ # 检查是否已存在对话窗口，如果存在则激活
+ if self.deepseek_dialog and self.deepseek_dialog.isVisible():
+ self.deepseek_dialog.activateWindow()
+ self.deepseek_dialog.raise_()
+ return
+
+ # 创建DeepSeek对话窗口
+ self.deepseek_dialog = DeepSeekDialogWindow(self)
+
+ # 连接对话窗口的关闭信号
+ self.deepseek_dialog.closed.connect(self.on_deepseek_dialog_closed)
+
+ # 显示对话窗口
+ self.deepseek_dialog.show()
+
+ # 更新菜单状态
+ self.deepseek_dialog_action.setChecked(True)
+
+ self.status_bar.showMessage("DeepSeek AI对话窗口已打开", 3000)
+
+ except ImportError as e:
+ QMessageBox.critical(self, "错误", f"无法加载DeepSeek对话窗口:\n{str(e)}")
+ except Exception as e:
+ QMessageBox.critical(self, "错误", f"打开DeepSeek对话窗口时出错:\n{str(e)}")
+
+ def on_deepseek_dialog_closed(self):
+ """DeepSeek对话窗口关闭时的回调"""
+ # 重置菜单状态
+ self.deepseek_dialog_action.setChecked(False)
+
+ # 清除对话窗口引用
+ self.deepseek_dialog = None
+
+ self.status_bar.showMessage("DeepSeek AI对话窗口已关闭", 2000)
+
def on_learning_content_changed(self, new_content, position):
"""学习模式内容变化时的回调 - 只在末尾追加新内容"""
# 设置同步标记，防止递归调用
--
2.34.1