from PySide6.QtWidgets import (
    QMainWindow, QWidget, QVBoxLayout, QHBoxLayout, QTextEdit, QPushButton,
    QLineEdit, QLabel, QGroupBox, QScrollArea, QToolButton, QStatusBar,
    QFileDialog, QMessageBox, QListWidget, QListWidgetItem, QFrame, QSplitter,
    QDialog, QProgressBar, QGridLayout
)
from PySide6.QtCore import Qt, QTimer, QSize, QPropertyAnimation, QEasingCurve
from PySide6.QtGui import QFont, QPalette, QColor
from utils.logger import logger
import os

class SettingsDialog(QDialog):
    """设置对话框，包含高级辅助功能"""
    def __init__(self, parent=None):
        super().__init__(parent)
        self.setWindowTitle("设置 - 高级辅助")
        self.setMinimumSize(500, 600)
        self.setModal(True)
        
        # 从父窗口获取数据
        self.parent_window = parent
        self.terms = parent.terms if parent else []
        
        # 初始化UI
        self.init_ui()
        
        # 加载现有数据
        self.load_existing_data()
    
    def init_ui(self):
        """初始化设置对话框UI"""
        main_layout = QVBoxLayout(self)
        main_layout.setSpacing(20)
        main_layout.setContentsMargins(20, 20, 20, 20)
        
        # 标题
        title_label = QLabel("高级辅助设置")
        title_font = QFont()
        title_font.setPointSize(18)
        title_font.setBold(True)
        title_label.setFont(title_font)
        title_label.setStyleSheet("color: #1a365d; margin-bottom: 10px;")
        
        # 文本背景/场景介绍
        context_group = QGroupBox("文本背景 / 场景介绍")
        context_group.setStyleSheet("""
            QGroupBox {
                font-weight: bold;
                color: #4a5568;
                border: 1px solid #e2e8f0;
                border-radius: 8px;
                margin-top: 10px;
                padding-top: 10px;
            }
            QGroupBox::title {
                subcontrol-origin: margin;
                left: 10px;
                padding: 0 5px 0 5px;
            }
        """)
        
        context_layout = QVBoxLayout(context_group)
        
        self.context_edit = QTextEdit()
        self.context_edit.setPlaceholderText("例如：这是一份关于建筑工程的合同...")
        self.context_edit.setFixedHeight(100)
        self.context_edit.setStyleSheet(
            "border: 1px solid #e2e8f0; border-radius: 6px; padding: 8px;"
        )
        
        context_layout.addWidget(self.context_edit)
        
        # 术语定义
        terms_group = QGroupBox("术语簿 (定义 A=B)")
        terms_group.setStyleSheet("""
            QGroupBox {
                font-weight: bold;
                color: #4a5568;
                border: 1px solid #e2e8f0;
                border-radius: 8px;
                margin-top: 10px;
                padding-top: 10px;
            }
            QGroupBox::title {
                subcontrol-origin: margin;
                left: 10px;
                padding: 0 5px 0 5px;
            }
        """)
        
        terms_layout = QVBoxLayout(terms_group)
        
        # 术语列表
        self.terms_list = QListWidget()
        self.terms_list.setStyleSheet(
            "border: 1px solid #e2e8f0; border-radius: 6px; min-height: 150px;"
        )
        
        # 术语输入区域
        terms_input_layout = QHBoxLayout()
        self.term_input = QLineEdit()
        self.term_input.setPlaceholderText("输入术语格式：原文 = 译文")
        self.term_input.setStyleSheet(
            "border: 1px solid #e2e8f0; border-radius: 6px; padding: 8px;"
        )
        
        add_term_btn = QPushButton("添加")
        add_term_btn.setFixedSize(80, 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)
        
        # 术语操作按钮
        terms_buttons_layout = QHBoxLayout()
        
        delete_term_btn = QPushButton("删除选中")
        delete_term_btn.setFixedSize(100, 32)
        delete_term_btn.setStyleSheet(
            "background-color: #e53e3e; color: white; border: none; border-radius: 6px;"
        )
        delete_term_btn.clicked.connect(self.delete_selected_term)
        
        clear_terms_btn = QPushButton("清空所有")
        clear_terms_btn.setFixedSize(100, 32)
        clear_terms_btn.setStyleSheet(
            "background-color: #a0aec0; color: white; border: none; border-radius: 6px;"
        )
        clear_terms_btn.clicked.connect(self.clear_all_terms)
        
        terms_buttons_layout.addWidget(delete_term_btn)
        terms_buttons_layout.addWidget(clear_terms_btn)
        terms_buttons_layout.addStretch()
        
        terms_layout.addWidget(self.terms_list)
        terms_layout.addLayout(terms_input_layout)
        terms_layout.addLayout(terms_buttons_layout)
        
        # 按钮区域
        buttons_layout = QHBoxLayout()
        
        save_btn = QPushButton("保存并应用")
        save_btn.setFixedSize(120, 40)
        save_btn.setStyleSheet(
            "background-color: #38a169; color: white; border: none; border-radius: 6px; font-weight: bold;"
        )
        save_btn.clicked.connect(self.save_and_apply)
        
        cancel_btn = QPushButton("取消")
        cancel_btn.setFixedSize(120, 40)
        cancel_btn.setStyleSheet(
            "background-color: #e2e8f0; color: #2d3748; border: none; border-radius: 6px;"
        )
        cancel_btn.clicked.connect(self.reject)
        
        buttons_layout.addStretch()
        buttons_layout.addWidget(cancel_btn)
        buttons_layout.addWidget(save_btn)
        
        # 添加到主布局
        main_layout.addWidget(title_label)
        main_layout.addWidget(context_group)
        main_layout.addWidget(terms_group)
        main_layout.addStretch()
        main_layout.addLayout(buttons_layout)
    
    def load_existing_data(self):
        """加载现有数据"""
        if self.parent_window:
            # 加载上下文
            if hasattr(self.parent_window, 'context_edit'):
                self.context_edit.setText(self.parent_window.context_edit.toPlainText())
            
            # 加载术语
            for term in self.terms:
                self.terms_list.addItem(term)
    
    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, "警告", "术语格式不正确，请使用 '原文 = 译文' 格式")
    
    def delete_selected_term(self):
        """删除选中的术语"""
        selected_items = self.terms_list.selectedItems()
        if not selected_items:
            QMessageBox.warning(self, "警告", "请先选择要删除的术语")
            return
        
        for item in selected_items:
            term_text = item.text()
            if term_text in self.terms:
                self.terms.remove(term_text)
            self.terms_list.takeItem(self.terms_list.row(item))
    
    def clear_all_terms(self):
        """清空所有术语"""
        if self.terms_list.count() == 0:
            return
        
        reply = QMessageBox.question(
            self, "确认", "确定要清空所有术语吗？",
            QMessageBox.Yes | QMessageBox.No, QMessageBox.No
        )
        
        if reply == QMessageBox.Yes:
            self.terms.clear()
            self.terms_list.clear()
    
    def save_and_apply(self):
        """保存并应用设置"""
        # 保存到父窗口
        if self.parent_window:
            # 保存上下文
            if hasattr(self.parent_window, 'context_edit'):
                self.parent_window.context_edit.setText(self.context_edit.toPlainText())
            
            # 保存术语
            self.parent_window.terms = self.terms.copy()
            
            # 更新术语列表显示（如果父窗口有显示的话）
            if hasattr(self.parent_window, 'terms_list'):
                self.parent_window.terms_list.clear()
                for term in self.terms:
                    self.parent_window.terms_list.addItem(term)
        
        QMessageBox.information(self, "成功", "设置已保存并应用")
        self.accept()

class MainWindowSimple(QMainWindow):
    def __init__(self):
        super().__init__()
        self.setWindowTitle("PrivaTrans - 简化测试版")
        self.setMinimumSize(800, 600)
        
        # 术语列表
        self.terms = []
        
        # 倒计时相关
        self.countdown_timer = None
        self.countdown_seconds = 60
        self.countdown_label = None
        
        # 初始化UI
        self.init_ui()
    
    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_bar = QWidget()
        title_bar_layout = QHBoxLayout(title_bar)
        title_bar_layout.setContentsMargins(0, 0, 0, 0)
        title_bar_layout.setSpacing(0)
        
        # 应用标题
        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;")
        
        # 右侧设置按钮
        self.settings_btn = QPushButton("⚙ 设置")
        self.settings_btn.setFixedSize(80, 36)
        self.settings_btn.setStyleSheet("""
            QPushButton {
                background-color: #e2e8f0;
                color: #2d3748;
                border: none;
                border-radius: 6px;
                font-weight: bold;
            }
            QPushButton:hover {
                background-color: #cbd5e0;
            }
            QPushButton:pressed {
                background-color: #a0aec0;
            }
        """)
        self.settings_btn.clicked.connect(self.open_settings)
        
        title_bar_layout.addWidget(title_label)
        title_bar_layout.addStretch()
        title_bar_layout.addWidget(self.settings_btn)
        
        # 模型信息（简化版）
        model_layout = QHBoxLayout()
        
        self.model_status_label = QLabel("● 测试模式 - 跳过模型加载")
        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()
        
        # 原文输入区域
        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; min-height: 150px;"
        )
        
        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)
        
        # 倒计时提示区域（初始隐藏）
        self.countdown_container = QWidget()
        self.countdown_container.setVisible(False)
        countdown_layout = QHBoxLayout(self.countdown_container)
        countdown_layout.setContentsMargins(0, 5, 0, 5)
        
        countdown_icon = QLabel("⏳")
        countdown_icon.setStyleSheet("font-size: 16px; color: #d69e2e;")
        
        self.countdown_label = QLabel("大模型正在翻译中，预计剩余时间：60秒")
        self.countdown_label.setStyleSheet("color: #d69e2e; font-weight: bold;")
        
        countdown_layout.addWidget(countdown_icon)
        countdown_layout.addWidget(self.countdown_label)
        countdown_layout.addStretch()
        
        # 译文结果区域
        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; min-height: 150px;"
        )
        
        output_layout.addLayout(output_header_layout)
        output_layout.addWidget(self.result_text)
        
        # 将所有组件添加到主布局
        main_layout.addWidget(title_bar)
        main_layout.addLayout(model_layout)
        main_layout.addLayout(input_layout)
        main_layout.addWidget(self.translate_btn)
        main_layout.addWidget(self.countdown_container)
        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)
        
        # 初始化上下文编辑框（用于存储设置）
        self.context_edit = QTextEdit()
        self.context_edit.setVisible(False)
    
    def open_settings(self):
        """打开设置对话框"""
        dialog = SettingsDialog(self)
        dialog.exec()
    
    def start_countdown(self):
        """开始60秒倒计时"""
        self.countdown_seconds = 60
        self.countdown_container.setVisible(True)
        self.update_countdown_display()
        
        # 创建倒计时定时器
        self.countdown_timer = QTimer()
        self.countdown_timer.setInterval(1000)  # 每秒触发一次
        self.countdown_timer.timeout.connect(self.update_countdown)
        self.countdown_timer.start()
    
    def update_countdown(self):
        """更新倒计时"""
        self.countdown_seconds -= 1
        self.update_countdown_display()
        
        if self.countdown_seconds <= 0:
            self.stop_countdown()
    
    def update_countdown_display(self):
        """更新倒计时显示"""
        if self.countdown_label:
            self.countdown_label.setText(f"大模型正在翻译中，预计剩余时间：{self.countdown_seconds}秒")
    
    def stop_countdown(self):
        """停止倒计时"""
        if self.countdown_timer:
            self.countdown_timer.stop()
            self.countdown_timer = None
        
        self.countdown_container.setVisible(False)
    
    def start_translation_test(self):
        """开始翻译测试"""
        source_text = self.source_text.toPlainText().strip()
        if not source_text:
            QMessageBox.warning(self, "警告", "原文内容为空")
            return
        
        # 禁用翻译按钮
        self.translate_btn.setEnabled(False)
        self.translate_btn.setText("翻译中...")
        
        # 开始倒计时
        self.start_countdown()
        
        # 获取上下文和术语
        context = self.context_edit.toPlainText().strip()
        terms = self.terms if self.terms else None
        
        # 模拟翻译过程（5秒后完成）
        from PySide6.QtCore import QTimer
        self.simulation_timer = QTimer()
        self.simulation_timer.setSingleShot(True)
        self.simulation_timer.setInterval(5000)  # 5秒后完成
        self.simulation_timer.timeout.connect(lambda: self.on_translation_finished(f"【测试翻译结果】\n\n原文：{source_text}\n\n译文：这是模拟的翻译结果\n\n上下文：{context if context else '无'}\n术语：{', '.join(terms) if terms else '无'}"))
        self.simulation_timer.start()
    
    def on_translation_finished(self, result):
        """翻译完成回调"""
        # 停止倒计时
        self.stop_countdown()
        
        # 启用翻译按钮
        self.translate_btn.setEnabled(True)
        self.translate_btn.setText("开始翻译 (测试倒计时)")
        
        if result:
            self.result_text.setText(result)
            QMessageBox.information(self, "测试完成", "翻译测试完成！\n\n✓ 设置按钮功能正常\n✓ 倒计时功能正常\n✓ 界面布局优化完成")

# 修复导入问题
from PySide6.QtWidgets import QApplication