feat: 实现基于PySide6的翻译GUI工具初始版本

添加主程序入口、GUI界面、翻译核心逻辑、Word文件处理、系统监控和日志模块
This commit is contained in:
2026-01-14 15:10:19 +08:00
commit 136994db90
9 changed files with 762 additions and 0 deletions

27
.gitignore vendored Normal file
View File

@@ -0,0 +1,27 @@
# 虚拟环境
venv/
env/
*.venv/
*.env/
# Python编译文件
*.pyc
__pycache__/
# 日志文件
logs/
*.log
# IDE配置
.idea/
.vscode/
*.swp
*.swo
*~
# 操作系统文件
.DS_Store
Thumbs.db
# 模型文件
models/*.gguf

View File

@@ -0,0 +1,88 @@
# 实现基于PySide6的翻译GUI工具
## 1. 项目结构设计
```
translate/
├── main.py # 主程序入口
├── translator.py # 翻译核心逻辑
├── models/ # 模型文件目录
│ └── HY-MT1.5-1.8B_bf16_Q4_K_M-GGUF.gguf # 模型文件
├── ui/
│ └── main_window.py # GUI界面设计
├── utils/
│ ├── word_handler.py # Word文件处理
│ ├── system_monitor.py # 系统资源监控
│ └── logger.py # 日志配置
└── requirements.txt # 依赖库列表
```
## 2. 核心功能实现
### 2.1 依赖库安装
- PySide6GUI框架
- llama-cpp-python运行GGUF模型
- python-docxWord文件处理
- loguru日志管理
- psutil系统资源监控
- GPUtilGPU资源监控如果需要
### 2.2 系统资源监控
- 实现CPU占用率监控
- 实现内存占用率监控
- 实现显卡内存占用率监控
- 定期更新状态栏显示
### 2.3 模型管理
- 实现模型加载和初始化
- 支持模型文件选择和切换
- 显示模型状态(就绪/加载中)
### 2.4 翻译核心
- 使用llama-cpp-python调用模型进行翻译
- 支持添加文本背景和场景介绍
- 支持术语定义功能
### 2.5 GUI界面设计
- 实现与参考图一致的界面布局
- 顶部模型信息和更换按钮
- 可折叠的高级辅助面板
- 文本背景/场景介绍输入框
- 术语定义列表和管理
- 原文输入区域支持Word导入
- 翻译按钮
- 译文结果区域支持Word导出和复制
- 底部状态栏,显示系统资源占用情况
### 2.6 Word文件处理
- 实现docx文件的导入提取文本内容
- 实现译文的docx文件导出
- 保持文档格式(尽可能)
## 3. 代码实现步骤
1. 创建项目目录结构
2. 编写requirements.txt文件
3. 实现日志配置模块
4. 实现系统资源监控模块
5. 实现Word文件处理模块
6. 实现翻译核心逻辑
7. 实现GUI界面设计包括状态栏
8. 实现主程序入口,整合所有功能
9. 测试各功能模块
## 4. 注意事项
- 确保模型文件路径正确配置
- 处理模型加载和翻译过程中的异常
- 优化GUI响应速度避免翻译过程中界面卡顿
- 实现良好的错误提示和日志记录
- 支持Windows系统的文件路径格式
- 系统监控模块要低开销运行,避免影响翻译性能
## 5. 未来扩展考虑
- 支持更多模型格式
- 实现批量翻译功能
- 支持更多文档格式
- 添加翻译历史记录
- 实现翻译质量评估

27
main.py Normal file
View File

@@ -0,0 +1,27 @@
import sys
from PySide6.QtWidgets import QApplication
from ui.main_window import MainWindow
from utils.logger import logger
def main():
"""主程序入口"""
logger.info("启动PrivaTrans翻译工具")
# 创建应用程序
app = QApplication(sys.argv)
# 设置应用程序样式
app.setStyle("Fusion")
# 创建主窗口
window = MainWindow()
window.show()
# 运行应用程序
exit_code = app.exec()
logger.info(f"PrivaTrans翻译工具退出退出码: {exit_code}")
sys.exit(exit_code)
if __name__ == "__main__":
main()

6
requirements.txt Normal file
View File

@@ -0,0 +1,6 @@
PySide6
llama-cpp-python
python-docx
loguru
psutil
GPUtil

105
translator.py Normal file
View File

@@ -0,0 +1,105 @@
from llama_cpp import Llama
import os
from utils.logger import logger
class Translator:
def __init__(self, model_path=None):
self.model = None
self.model_path = model_path
self.is_ready = False
self.model_name = ""
def load_model(self, model_path=None):
"""加载模型"""
if model_path:
self.model_path = model_path
if not self.model_path:
logger.error("未提供模型路径")
return False
if not os.path.exists(self.model_path):
logger.error(f"模型文件不存在: {self.model_path}")
return False
try:
logger.info(f"开始加载模型: {self.model_path}")
self.model = Llama(
model_path=self.model_path,
n_ctx=2048,
n_threads=4,
n_gpu_layers=100 # 尽可能使用GPU加速
)
self.is_ready = True
self.model_name = os.path.basename(self.model_path)
logger.info(f"模型加载成功: {self.model_name}")
return True
except Exception as e:
logger.error(f"模型加载失败: {e}")
self.is_ready = False
return False
def translate(self, text, context="", terms=None):
"""执行翻译"""
if not self.is_ready or not self.model:
logger.error("模型未就绪,无法执行翻译")
return ""
try:
# 构建翻译提示词
prompt = self._build_prompt(text, context, terms)
logger.info(f"开始翻译,输入长度: {len(text)} 字符")
# 调用模型进行翻译
output = self.model(
prompt,
max_tokens=2048,
temperature=0.7,
top_p=0.95,
stop=["\n原文:", "\n译文:", "\n###"]
)
translated_text = output["choices"][0]["text"].strip()
logger.info(f"翻译完成,输出长度: {len(translated_text)} 字符")
return translated_text
except Exception as e:
logger.error(f"翻译失败: {e}")
return ""
def _build_prompt(self, text, context="", terms=None):
"""构建翻译提示词"""
prompt = "你是一个专业的翻译助手,根据以下要求将中文翻译成英文:\n"
if context:
prompt += f"\n文本背景/场景介绍:{context}\n"
if terms:
prompt += "\n术语定义:\n"
for term in terms:
prompt += f"{term}\n"
prompt += f"\n原文:{text}\n译文:"
return prompt
def unload_model(self):
"""卸载模型"""
if self.model:
try:
del self.model
self.model = None
self.is_ready = False
logger.info("模型已卸载")
return True
except Exception as e:
logger.error(f"模型卸载失败: {e}")
return False
return True
def get_model_info(self):
"""获取模型信息"""
if self.is_ready:
return f"{self.model_name.split('.')[0]}"
return "未加载模型"

386
ui/main_window.py Normal file
View File

@@ -0,0 +1,386 @@
from PySide6.QtWidgets import (
QMainWindow, QWidget, QVBoxLayout, QHBoxLayout, QTextEdit, QPushButton,
QLineEdit, QLabel, QGroupBox, QScrollArea, QToolButton, QStatusBar,
QFileDialog, QMessageBox, QListWidget, QListWidgetItem, QFrame, QSplitter
)
from PySide6.QtCore import Qt, QTimer, QSize
from PySide6.QtGui import QFont, QPalette, QColor
from utils.word_handler import WordHandler
from utils.system_monitor import SystemMonitor
from translator import Translator
from utils.logger import logger
import os
class MainWindow(QMainWindow):
def __init__(self):
super().__init__()
self.setWindowTitle("PrivaTrans")
self.setMinimumSize(800, 600)
# 初始化组件
self.translator = Translator()
self.system_monitor = SystemMonitor()
self.word_handler = WordHandler()
# 默认模型路径
self.default_model_path = os.path.join(os.path.dirname(os.path.dirname(__file__)), "models", "HY-MT1.5-1.8B_bf16_Q4_K_M.gguf")
# 术语列表
self.terms = []
# 初始化UI
self.init_ui()
# 初始化状态栏定时器
self.init_status_bar_timer()
# 加载默认模型
self.load_default_model()
def init_ui(self):
"""初始化UI界面"""
# 主布局
central_widget = QWidget()
self.setCentralWidget(central_widget)
main_layout = QVBoxLayout(central_widget)
main_layout.setSpacing(20)
main_layout.setContentsMargins(20, 20, 20, 10)
# 顶部标题
title_label = QLabel("PrivaTrans")
title_font = QFont()
title_font.setPointSize(24)
title_font.setBold(True)
title_label.setFont(title_font)
title_label.setStyleSheet("color: #1a365d;")
# 模型信息和更换按钮
model_layout = QHBoxLayout()
self.model_status_label = QLabel()
self.model_status_label.setStyleSheet("color: #2d3748;")
self.model_info_label = QLabel()
self.model_info_label.setStyleSheet("color: #2d3748; font-weight: bold;")
self.change_model_btn = QPushButton("更换")
self.change_model_btn.setFixedSize(80, 32)
self.change_model_btn.setStyleSheet(
"background-color: #e2e8f0; color: #2d3748; border: none; border-radius: 6px;"
)
self.change_model_btn.clicked.connect(self.change_model)
model_layout.addWidget(self.model_status_label)
model_layout.addWidget(self.model_info_label)
model_layout.addStretch()
model_layout.addWidget(self.change_model_btn)
# 高级辅助面板(可折叠)
self.advanced_group = QGroupBox("高级辅助 (背景与术语)")
self.advanced_group.setCheckable(True)
self.advanced_group.setChecked(False)
self.advanced_group.setStyleSheet(
"QGroupBox { border: 1px solid #e2e8f0; border-radius: 8px; margin-top: 10px; }"
"QGroupBox::title { subcontrol-origin: margin; left: 10px; padding: 0 5px 0 5px; }"
)
advanced_layout = QVBoxLayout(self.advanced_group)
advanced_layout.setSpacing(15)
advanced_layout.setContentsMargins(15, 15, 15, 15)
# 文本背景/场景介绍
context_label = QLabel("文本背景 / 场景介绍")
context_label.setStyleSheet("color: #4a5568; font-weight: bold;")
self.context_edit = QTextEdit()
self.context_edit.setPlaceholderText("例如:这是一份关于建筑工程的合同...")
self.context_edit.setFixedHeight(80)
self.context_edit.setStyleSheet(
"border: 1px solid #e2e8f0; border-radius: 6px; padding: 8px;"
)
# 术语定义
terms_label = QLabel("术语簿 (定义 A=B)")
terms_label.setStyleSheet("color: #4a5568; font-weight: bold;")
self.terms_list = QListWidget()
self.terms_list.setStyleSheet(
"border: 1px solid #e2e8f0; border-radius: 6px;"
)
terms_input_layout = QHBoxLayout()
self.term_input = QLineEdit()
self.term_input.setPlaceholderText("你好 = what's up")
self.term_input.setStyleSheet(
"border: 1px solid #e2e8f0; border-radius: 6px; padding: 8px; flex: 1;"
)
add_term_btn = QPushButton("+")
add_term_btn.setFixedSize(32, 32)
add_term_btn.setStyleSheet(
"background-color: #3182ce; color: white; border: none; border-radius: 6px;"
)
add_term_btn.clicked.connect(self.add_term)
terms_input_layout.addWidget(self.term_input)
terms_input_layout.addWidget(add_term_btn)
# 术语列表项右键菜单
self.terms_list.setContextMenuPolicy(Qt.CustomContextMenu)
self.terms_list.customContextMenuRequested.connect(self.show_term_context_menu)
advanced_layout.addWidget(context_label)
advanced_layout.addWidget(self.context_edit)
advanced_layout.addWidget(terms_label)
advanced_layout.addWidget(self.terms_list)
advanced_layout.addLayout(terms_input_layout)
# 原文输入区域
input_layout = QVBoxLayout()
input_header_layout = QHBoxLayout()
input_label = QLabel("原文内容")
input_label.setStyleSheet("color: #4a5568; font-weight: bold;")
import_word_btn = QPushButton("导入 Word")
import_word_btn.setFixedSize(120, 32)
import_word_btn.setStyleSheet(
"background-color: #e2e8f0; color: #2d3748; border: none; border-radius: 6px;"
)
import_word_btn.clicked.connect(self.import_word)
input_header_layout.addWidget(input_label)
input_header_layout.addStretch()
input_header_layout.addWidget(import_word_btn)
self.source_text = QTextEdit()
self.source_text.setStyleSheet(
"border: 1px solid #e2e8f0; border-radius: 6px; padding: 10px;"
)
input_layout.addLayout(input_header_layout)
input_layout.addWidget(self.source_text)
# 翻译按钮
self.translate_btn = QPushButton("开始翻译")
self.translate_btn.setFixedHeight(50)
self.translate_btn.setStyleSheet(
"background-color: #3182ce; color: white; border: none; border-radius: 8px; font-size: 16px; font-weight: bold;"
)
self.translate_btn.clicked.connect(self.start_translation)
# 译文结果区域
output_layout = QVBoxLayout()
output_header_layout = QHBoxLayout()
output_label = QLabel("译文结果")
output_label.setStyleSheet("color: #4a5568; font-weight: bold;")
output_buttons_layout = QHBoxLayout()
export_word_btn = QPushButton("导出 Word")
export_word_btn.setFixedSize(120, 32)
export_word_btn.setStyleSheet(
"background-color: #e2e8f0; color: #2d3748; border: none; border-radius: 6px; margin-right: 10px;"
)
export_word_btn.clicked.connect(self.export_word)
copy_btn = QPushButton("复制内容")
copy_btn.setFixedSize(120, 32)
copy_btn.setStyleSheet(
"background-color: #e2e8f0; color: #2d3748; border: none; border-radius: 6px;"
)
copy_btn.clicked.connect(self.copy_result)
output_buttons_layout.addWidget(export_word_btn)
output_buttons_layout.addWidget(copy_btn)
output_header_layout.addWidget(output_label)
output_header_layout.addStretch()
output_header_layout.addLayout(output_buttons_layout)
self.result_text = QTextEdit()
self.result_text.setReadOnly(True)
self.result_text.setStyleSheet(
"border: 1px solid #e2e8f0; border-radius: 6px; padding: 10px;"
)
output_layout.addLayout(output_header_layout)
output_layout.addWidget(self.result_text)
# 将所有组件添加到主布局
main_layout.addWidget(title_label)
main_layout.addLayout(model_layout)
main_layout.addWidget(self.advanced_group)
main_layout.addLayout(input_layout)
main_layout.addWidget(self.translate_btn)
main_layout.addLayout(output_layout)
# 状态栏
self.status_bar = QStatusBar()
self.setStatusBar(self.status_bar)
self.status_label = QLabel()
self.status_bar.addWidget(self.status_label)
def init_status_bar_timer(self):
"""初始化状态栏定时器"""
self.status_timer = QTimer()
self.status_timer.setInterval(1000) # 每秒更新一次
self.status_timer.timeout.connect(self.update_status_bar)
self.status_timer.start()
def update_status_bar(self):
"""更新状态栏显示"""
status_text = self.system_monitor.get_status_text()
self.status_label.setText(status_text)
def load_default_model(self):
"""加载默认模型"""
if os.path.exists(self.default_model_path):
self.model_status_label.setText("模型加载中...")
if self.translator.load_model(self.default_model_path):
self.model_status_label.setText("● 模型就绪")
self.model_status_label.setStyleSheet("color: #38a169;")
self.model_info_label.setText(f"{self.translator.get_model_info()}")
else:
self.model_status_label.setText("● 模型加载失败")
self.model_status_label.setStyleSheet("color: #e53e3e;")
else:
self.model_status_label.setText("● 模型文件不存在")
self.model_status_label.setStyleSheet("color: #e53e3e;")
def change_model(self):
"""更换模型"""
file_path, _ = QFileDialog.getOpenFileName(
self, "选择模型文件", "", "GGUF模型文件 (*.gguf)"
)
if file_path:
self.model_status_label.setText("模型加载中...")
if self.translator.load_model(file_path):
self.model_status_label.setText("● 模型就绪")
self.model_status_label.setStyleSheet("color: #38a169;")
self.model_info_label.setText(f"{self.translator.get_model_info()}")
QMessageBox.information(self, "成功", "模型加载成功")
else:
self.model_status_label.setText("● 模型加载失败")
self.model_status_label.setStyleSheet("color: #e53e3e;")
QMessageBox.critical(self, "错误", "模型加载失败")
def add_term(self):
"""添加术语定义"""
term_text = self.term_input.text().strip()
if term_text:
if "=" in term_text:
self.terms.append(term_text)
self.terms_list.addItem(term_text)
self.term_input.clear()
else:
QMessageBox.warning(self, "警告", "术语格式不正确,请使用 'A = B' 格式")
def show_term_context_menu(self, pos):
"""显示术语项右键菜单"""
from PySide6.QtWidgets import QMenu
menu = QMenu()
delete_action = menu.addAction("删除")
action = menu.exec_(self.terms_list.mapToGlobal(pos))
if action == delete_action:
selected_items = self.terms_list.selectedItems()
for item in selected_items:
self.terms.remove(item.text())
self.terms_list.takeItem(self.terms_list.row(item))
def import_word(self):
"""导入Word文件"""
file_path, _ = QFileDialog.getOpenFileName(
self, "导入Word文件", "", "Word文件 (*.docx)"
)
if file_path:
try:
text = self.word_handler.import_docx(file_path)
self.source_text.setText(text)
QMessageBox.information(self, "成功", "Word文件导入成功")
except Exception as e:
logger.error(f"导入Word文件失败: {e}")
QMessageBox.critical(self, "错误", f"Word文件导入失败: {str(e)}")
def export_word(self):
"""导出Word文件"""
result_text = self.result_text.toPlainText()
if not result_text.strip():
QMessageBox.warning(self, "警告", "译文结果为空,无法导出")
return
file_path, _ = QFileDialog.getSaveFileName(
self, "导出Word文件", "", "Word文件 (*.docx)"
)
if file_path:
try:
self.word_handler.export_docx(file_path, result_text)
QMessageBox.information(self, "成功", "Word文件导出成功")
except Exception as e:
logger.error(f"导出Word文件失败: {e}")
QMessageBox.critical(self, "错误", f"Word文件导出失败: {str(e)}")
def copy_result(self):
"""复制译文结果"""
result_text = self.result_text.toPlainText()
if result_text.strip():
from PySide6.QtGui import QClipboard
clipboard = QApplication.clipboard()
clipboard.setText(result_text)
QMessageBox.information(self, "成功", "译文已复制到剪贴板")
else:
QMessageBox.warning(self, "警告", "译文结果为空,无法复制")
def start_translation(self):
"""开始翻译"""
source_text = self.source_text.toPlainText().strip()
if not source_text:
QMessageBox.warning(self, "警告", "原文内容为空,无法翻译")
return
if not self.translator.is_ready:
QMessageBox.warning(self, "警告", "模型未就绪,无法翻译")
return
# 禁用翻译按钮
self.translate_btn.setEnabled(False)
self.translate_btn.setText("翻译中...")
# 获取上下文和术语
context = self.context_edit.toPlainText().strip()
terms = self.terms if self.terms else None
# 执行翻译在后台线程中执行避免UI卡顿
from PySide6.QtCore import QThread, Signal
class TranslationThread(QThread):
finished = Signal(str)
def __init__(self, translator, text, context, terms):
super().__init__()
self.translator = translator
self.text = text
self.context = context
self.terms = terms
def run(self):
result = self.translator.translate(self.text, self.context, self.terms)
self.finished.emit(result)
self.translation_thread = TranslationThread(self.translator, source_text, context, terms)
self.translation_thread.finished.connect(self.on_translation_finished)
self.translation_thread.start()
def on_translation_finished(self, result):
"""翻译完成回调"""
# 启用翻译按钮
self.translate_btn.setEnabled(True)
self.translate_btn.setText("开始翻译")
if result:
self.result_text.setText(result)
else:
QMessageBox.critical(self, "错误", "翻译失败,请检查日志")
# 修复导入问题
from PySide6.QtWidgets import QApplication

27
utils/logger.py Normal file
View File

@@ -0,0 +1,27 @@
from loguru import logger
import os
import sys
# 创建logs目录
logs_dir = os.path.join(os.path.dirname(os.path.dirname(__file__)), 'logs')
os.makedirs(logs_dir, exist_ok=True)
# 配置日志
logger.remove()
logger.add(
sys.stdout,
level="INFO",
format="{time:YYYY-MM-DD HH:mm:ss} | {level} | {message}",
colorize=True
)
logger.add(
os.path.join(logs_dir, "translate.log"),
level="DEBUG",
format="{time:YYYY-MM-DD HH:mm:ss} | {level} | {module}.{function}:{line} | {message}",
rotation="10 MB",
retention="7 days",
compression="zip"
)
# 导出logger实例
__all__ = ["logger"]

57
utils/system_monitor.py Normal file
View File

@@ -0,0 +1,57 @@
import psutil
from .logger import logger
# 尝试导入GPUtil如果失败则设置has_gpu为False
try:
import GPUtil
GPUtil_available = True
except ImportError:
logger.warning("GPUtil模块未找到将禁用GPU监控")
GPUtil_available = False
class SystemMonitor:
def __init__(self):
self.cpu_usage = 0.0
self.memory_usage = 0.0
self.gpu_memory_usage = 0.0
self.has_gpu = self._check_gpu()
def _check_gpu(self):
"""检查是否有可用的GPU"""
if not GPUtil_available:
return False
try:
gpus = GPUtil.getGPUs()
return len(gpus) > 0
except Exception as e:
logger.warning(f"GPU检查失败: {e}")
return False
def update_metrics(self):
"""更新系统资源使用情况"""
try:
# 更新CPU使用率
self.cpu_usage = psutil.cpu_percent(interval=0.1)
# 更新内存使用率
memory = psutil.virtual_memory()
self.memory_usage = memory.percent
# 更新GPU内存使用率如果有GPU
if self.has_gpu:
try:
gpus = GPUtil.getGPUs()
if gpus:
self.gpu_memory_usage = gpus[0].memoryUtil * 100
except Exception as e:
logger.warning(f"GPU内存使用率获取失败: {e}")
except Exception as e:
logger.error(f"系统资源监控更新失败: {e}")
def get_status_text(self):
"""获取状态栏显示文本"""
self.update_metrics()
status = f"CPU: {self.cpu_usage:.1f}% | 内存: {self.memory_usage:.1f}%"
if self.has_gpu:
status += f" | GPU内存: {self.gpu_memory_usage:.1f}%"
return status

39
utils/word_handler.py Normal file
View File

@@ -0,0 +1,39 @@
from docx import Document
from .logger import logger
class WordHandler:
@staticmethod
def import_docx(file_path):
"""从docx文件中提取文本内容"""
try:
doc = Document(file_path)
text = []
for paragraph in doc.paragraphs:
if paragraph.text.strip():
text.append(paragraph.text)
logger.info(f"成功从 {file_path} 导入文本,共 {len(text)}")
return "\n\n".join(text)
except Exception as e:
logger.error(f"导入Word文件失败: {e}")
raise
@staticmethod
def export_docx(file_path, content):
"""将文本内容导出为docx文件"""
try:
doc = Document()
# 将内容按段落分割并添加到文档
paragraphs = content.split("\n\n")
for para_text in paragraphs:
if para_text.strip():
doc.add_paragraph(para_text.strip())
doc.save(file_path)
logger.info(f"成功将文本导出到 {file_path},共 {len(paragraphs)}")
return True
except Exception as e:
logger.error(f"导出Word文件失败: {e}")
raise