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.logger import logger import os class MainWindowTest(QMainWindow): def __init__(self): super().__init__() self.setWindowTitle("PrivaTrans - UI测试") self.setMinimumSize(800, 600) # 初始化组件 self.terms = [] # 初始化UI self.init_ui() # 初始化状态栏定时器 self.init_status_bar_timer() 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 - UI测试") 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("● UI测试模式 - 跳过模型加载") self.model_status_label.setStyleSheet("color: #38a169;") self.model_info_label = QLabel("点击折叠/展开按钮测试功能") self.model_info_label.setStyleSheet("color: #2d3748; font-weight: bold;") model_layout.addWidget(self.model_status_label) model_layout.addWidget(self.model_info_label) model_layout.addStretch() # 高级辅助面板（可折叠） advanced_panel = QWidget() advanced_layout = QVBoxLayout(advanced_panel) advanced_layout.setSpacing(10) advanced_layout.setContentsMargins(0, 0, 0, 0) # 高级辅助标题和折叠按钮 advanced_header = QWidget() header_layout = QHBoxLayout(advanced_header) header_layout.setContentsMargins(0, 0, 0, 0) header_layout.setSpacing(10) advanced_title = QLabel("高级辅助 (背景与术语)") advanced_title.setStyleSheet("color: #4a5568; font-weight: bold;") self.advanced_toggle = QToolButton() self.advanced_toggle.setText("▼") self.advanced_toggle.setCheckable(True) self.advanced_toggle.setChecked(False) self.advanced_toggle.setFixedSize(20, 20) self.advanced_toggle.setStyleSheet( "border: none; background: none; color: #4a5568;" ) self.advanced_toggle.clicked.connect(self.toggle_advanced_panel) header_layout.addWidget(advanced_title) header_layout.addStretch() header_layout.addWidget(self.advanced_toggle) advanced_layout.addWidget(advanced_header) # 高级辅助内容区域 self.advanced_content = QWidget() content_layout = QVBoxLayout(self.advanced_content) content_layout.setSpacing(15) content_layout.setContentsMargins(0, 0, 0, 0) # 文本背景/场景介绍 context_label = QLabel("文本背景 / 场景介绍") context_label.setStyleSheet("color: #4a5568;") 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;") 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;" ) 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) content_layout.addWidget(context_label) content_layout.addWidget(self.context_edit) content_layout.addWidget(terms_label) content_layout.addWidget(self.terms_list) content_layout.addLayout(terms_input_layout) advanced_layout.addWidget(self.advanced_content) # 默认隐藏高级辅助内容 self.advanced_content.setVisible(False) # 将高级辅助面板添加到主布局 advanced_panel.setStyleSheet( "border: 1px solid #e2e8f0; border-radius: 8px; padding: 15px; margin-top: 10px;" ) # 原文输入区域 input_layout = QVBoxLayout() input_header_layout = QHBoxLayout() input_label = QLabel("原文内容") input_label.setStyleSheet("color: #4a5568; font-weight: bold;") input_header_layout.addWidget(input_label) input_header_layout.addStretch() self.source_text = QTextEdit() self.source_text.setPlaceholderText("在此输入要翻译的文本...") 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_test) # 译文结果区域 output_layout = QVBoxLayout() output_header_layout = QHBoxLayout() output_label = QLabel("译文结果") output_label.setStyleSheet("color: #4a5568; font-weight: bold;") output_header_layout.addWidget(output_label) output_header_layout.addStretch() self.result_text = QTextEdit() self.result_text.setReadOnly(True) self.result_text.setPlaceholderText("翻译结果将显示在这里...") 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(advanced_panel) 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("UI测试模式 - 折叠/展开按钮功能正常") 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): """更新状态栏显示""" # 在测试模式下显示固定状态 pass 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 toggle_advanced_panel(self, checked): """切换高级辅助面板的显示/隐藏状态""" self.advanced_content.setVisible(checked) self.advanced_toggle.setText("▲" if checked else "▼") # 更新状态栏显示当前状态 state = "展开" if checked else "折叠" self.status_label.setText(f"高级辅助面板已{state} - 按钮功能正常") 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 start_translation_test(self): """开始翻译测试""" source_text = self.source_text.toPlainText().strip() if not source_text: QMessageBox.warning(self, "警告", "原文内容为空") return # 在测试模式下显示模拟结果 self.result_text.setText(f"【测试模式】翻译结果预览：\n\n{source_text}\n\n（这是测试模式，实际翻译需要加载模型）") QMessageBox.information(self, "测试成功", "UI功能测试完成！\n\n✓ 折叠/展开按钮功能正常\n✓ 术语添加功能正常\n✓ 界面交互正常")