import os import json from PySide6.QtWidgets import (QWidget, QVBoxLayout, QHBoxLayout, QLabel, QScrollArea, QPushButton, QMessageBox, QSplitter) from PySide6.QtCore import Qt, Signal from ui_components import (CustomSlider, FloatSlider, RadioButtonGroup, AudioSelector, VideoPlayer) class SettingsTab(QWidget): """ 配置页面，允许用户设置所有处理参数并保存到config.json """ # 定义配置变更信号 config_changed = Signal(dict) def __init__(self, parent=None): super().__init__(parent) self.init_ui() self.load_config() def init_ui(self): """初始化配置页面UI""" self.layout = QVBoxLayout(self) # 创建一个滚动区域用于容纳配置项 self.scroll_area = QScrollArea() self.scroll_area.setWidgetResizable(True) self.scroll_widget = QWidget() self.scroll_layout = QVBoxLayout(self.scroll_widget) # 添加所有配置控件 self.add_config_widgets() # 创建保存配置按钮 self.button_layout = QHBoxLayout() self.save_config_button = QPushButton("保存配置") self.save_config_button.clicked.connect(self.save_config) self.save_config_button.setMinimumHeight(40) self.save_config_button.setStyleSheet("background-color: #4CAF50; color: white;") self.reset_config_button = QPushButton("重置配置") self.reset_config_button.clicked.connect(self.reset_config) self.reset_config_button.setMinimumHeight(40) self.button_layout.addWidget(self.reset_config_button) self.button_layout.addWidget(self.save_config_button) # 设置滚动区域 self.scroll_area.setWidget(self.scroll_widget) self.layout.addWidget(self.scroll_area) self.layout.addLayout(self.button_layout) self.setLayout(self.layout) def add_config_widgets(self): """添加所有配置控件""" # 视频配置 self.scroll_layout.addWidget(QLabel("=== 视频下载配置 ===")) self.scroll_layout.addWidget(QLabel("视频输出文件夹")) self.video_folder = self.add_label_value("videos", "视频输出到此文件夹") # 分辨率 self.resolution = RadioButtonGroup( ['4320p', '2160p', '1440p', '1080p', '720p', '480p', '360p', '240p', '144p'], "分辨率", '1080p' ) self.scroll_layout.addWidget(self.resolution) # 下载视频数量 self.video_count = CustomSlider(1, 100, 1, "下载视频数量", 5) self.scroll_layout.addWidget(self.video_count) # 音频处理配置 self.scroll_layout.addWidget(QLabel("=== 音频处理配置 ===")) # 模型 self.model = RadioButtonGroup( ['htdemucs', 'htdemucs_ft', 'htdemucs_6s', 'hdemucs_mmi', 'mdx', 'mdx_extra', 'mdx_q', 'mdx_extra_q', 'SIG'], "人声分离模型", 'htdemucs_ft' ) self.scroll_layout.addWidget(self.model) # 计算设备 self.device = RadioButtonGroup(['auto', 'cuda', 'cpu'], "计算设备", 'auto') self.scroll_layout.addWidget(self.device) # 移位次数 self.shifts = CustomSlider(0, 10, 1, "移位次数 Number of shifts", 5) self.scroll_layout.addWidget(self.shifts) # ASR配置 self.scroll_layout.addWidget(QLabel("=== 语音识别配置 ===")) # ASR模型选择 self.asr_model_label = QLabel("ASR模型选择") self.scroll_layout.addWidget(self.asr_model_label) self.asr_model = RadioButtonGroup(['WhisperX', 'FunASR'], "ASR模型选择", 'WhisperX') self.scroll_layout.addWidget(self.asr_model) # WhisperX模型大小 self.whisperx_size = RadioButtonGroup(['large', 'medium', 'small', 'base', 'tiny'], "WhisperX模型大小", 'large') self.scroll_layout.addWidget(self.whisperx_size) # 批处理大小 self.batch_size = CustomSlider(1, 128, 1, "批处理大小 Batch Size", 32) self.scroll_layout.addWidget(self.batch_size) # 分离多个说话人 self.separate_speakers = RadioButtonGroup([True, False], "分离多个说话人", True) self.scroll_layout.addWidget(self.separate_speakers) # 最小说话人数 self.min_speakers = RadioButtonGroup([None, 1, 2, 3, 4, 5, 6, 7, 8, 9], "最小说话人数", None) self.scroll_layout.addWidget(self.min_speakers) # 最大说话人数 self.max_speakers = RadioButtonGroup([None, 1, 2, 3, 4, 5, 6, 7, 8, 9], "最大说话人数", None) self.scroll_layout.addWidget(self.max_speakers) # 翻译配置 self.scroll_layout.addWidget(QLabel("=== 翻译配置 ===")) # 翻译方式 self.translation_method_label = QLabel("翻译方式") self.scroll_layout.addWidget(self.translation_method_label) self.translation_method = RadioButtonGroup( ['OpenAI', 'LLM', 'Google Translate', 'Bing Translate', 'Ernie', '火山引擎-deepseek', "deepseek-api", "阿里云-通义千问","Ollama"], "翻译方式", 'LLM') self.scroll_layout.addWidget(self.translation_method) # 目标语言 (翻译) self.target_language_translation_label = QLabel("目标语言 (翻译)") self.scroll_layout.addWidget(self.target_language_translation_label) self.target_language_translation = RadioButtonGroup( ['简体中文', '繁体中文', 'English', 'Cantonese', 'Japanese', 'Korean'], "目标语言 (翻译)", '简体中文') self.scroll_layout.addWidget(self.target_language_translation) # TTS配置 self.scroll_layout.addWidget(QLabel("=== 语音合成配置 ===")) # AI语音生成方法 self.tts_method_label = QLabel("AI语音生成方法") self.scroll_layout.addWidget(self.tts_method_label) self.tts_method = RadioButtonGroup(['xtts', 'cosyvoice', 'EdgeTTS'], "AI语音生成方法", 'EdgeTTS') self.scroll_layout.addWidget(self.tts_method) # 目标语言 (TTS) self.target_language_tts_label = QLabel("目标语言 (TTS)") self.scroll_layout.addWidget(self.target_language_tts_label) self.target_language_tts = RadioButtonGroup( ['中文', 'English', '粤语', 'Japanese', 'Korean', 'Spanish', 'French'], "目标语言 (TTS)", '中文') self.scroll_layout.addWidget(self.target_language_tts) # EdgeTTS声音选择 self.edge_tts_voice_label = QLabel("EdgeTTS声音选择") self.scroll_layout.addWidget(self.edge_tts_voice_label) self.edge_tts_voice = RadioButtonGroup( ['zh-CN-XiaoxiaoNeural', 'zh-CN-YunxiNeural', 'en-US-JennyNeural', 'ja-JP-NanamiNeural'], "EdgeTTS声音选择", 'zh-CN-XiaoxiaoNeural') self.scroll_layout.addWidget(self.edge_tts_voice) # 视频合成配置 self.scroll_layout.addWidget(QLabel("=== 视频合成配置 ===")) # 添加字幕 self.add_subtitles = RadioButtonGroup([True, False], "添加字幕", True) self.scroll_layout.addWidget(self.add_subtitles) # 加速倍数 self.speed_factor = FloatSlider(0.5, 2, 0.05, "加速倍数", 1.00) self.scroll_layout.addWidget(self.speed_factor) # 帧率 self.frame_rate = CustomSlider(1, 60, 1, "帧率", 30) self.scroll_layout.addWidget(self.frame_rate) # 背景音乐 self.background_music = AudioSelector("背景音乐") self.scroll_layout.addWidget(self.background_music) # 背景音乐音量 self.bg_music_volume = FloatSlider(0, 1, 0.05, "背景音乐音量", 0.5) self.scroll_layout.addWidget(self.bg_music_volume) # 视频音量 self.video_volume = FloatSlider(0, 1, 0.05, "视频音量", 1.0) self.scroll_layout.addWidget(self.video_volume) # 分辨率 (输出) self.output_resolution = RadioButtonGroup( ['4320p', '2160p', '1440p', '1080p', '720p', '480p', '360p', '240p', '144p'], "输出分辨率", '1080p' ) self.scroll_layout.addWidget(self.output_resolution) # 高级配置 self.scroll_layout.addWidget(QLabel("=== 高级配置 ===")) # Max Workers self.max_workers = CustomSlider(1, 100, 1, "Max Workers", 1) self.scroll_layout.addWidget(self.max_workers) # Max Retries self.max_retries = CustomSlider(1, 10, 1, "Max Retries", 3) self.scroll_layout.addWidget(self.max_retries) def add_label_value(self, value, tooltip=None): """为简单的文本值创建一个标签""" label = QLabel(value) if tooltip: label.setToolTip(tooltip) self.scroll_layout.addWidget(label) return label def get_config(self): """从UI控件中获取当前配置""" config = { "video_folder": self.video_folder.text(), "resolution": self.resolution.value(), "video_count": self.video_count.value(), "model": self.model.value(), "device": self.device.value(), "shifts": self.shifts.value(), "asr_model": self.asr_model.value(), "whisperx_size": self.whisperx_size.value(), "batch_size": self.batch_size.value(), "separate_speakers": self.separate_speakers.value(), "min_speakers": self.min_speakers.value(), "max_speakers": self.max_speakers.value(), "translation_method": self.translation_method.value(), "target_language_translation": self.target_language_translation.value(), "tts_method": self.tts_method.value(), "target_language_tts": self.target_language_tts.value(), "edge_tts_voice": self.edge_tts_voice.value(), "add_subtitles": self.add_subtitles.value(), "speed_factor": self.speed_factor.value(), "frame_rate": self.frame_rate.value(), "background_music": self.background_music.value(), "bg_music_volume": self.bg_music_volume.value(), "video_volume": self.video_volume.value(), "output_resolution": self.output_resolution.value(), "max_workers": self.max_workers.value(), "max_retries": self.max_retries.value() } return config def apply_config(self, config): """将配置应用到UI控件""" try: # 基本设置 self.video_folder.setText(config.get("video_folder", "videos")) # 为每个单选按钮组应用更健壮的选择逻辑 # 分辨率 resolution_value = config.get("resolution", "1080p") self._set_radio_button(self.resolution.buttons, resolution_value, "1080p") # 视频数量 self.video_count.setValue(config.get("video_count", 5)) # 模型 model_value = config.get("model", "htdemucs_ft") self._set_radio_button(self.model.buttons, model_value, "htdemucs_ft") # 设备 device_value = config.get("device", "auto") self._set_radio_button(self.device.buttons, device_value, "auto") # 移位次数 self.shifts.setValue(config.get("shifts", 5)) # ASR模型 asr_model_value = config.get("asr_model", "WhisperX") self._set_radio_button(self.asr_model.buttons, asr_model_value, "WhisperX") # WhisperX模型大小 whisperx_size_value = config.get("whisperx_size", "large") self._set_radio_button(self.whisperx_size.buttons, whisperx_size_value, "large") # 批处理大小 self.batch_size.setValue(config.get("batch_size", 32)) # 分离多个说话人 separate_speakers_value = config.get("separate_speakers", True) self._set_radio_button(self.separate_speakers.buttons, separate_speakers_value, True) # 最小说话人数 min_speakers_value = config.get("min_speakers", None) self._set_radio_button(self.min_speakers.buttons, min_speakers_value, None) # 最大说话人数 max_speakers_value = config.get("max_speakers", None) self._set_radio_button(self.max_speakers.buttons, max_speakers_value, None) # 翻译方式 translation_method_value = config.get("translation_method", "LLM") self._set_radio_button(self.translation_method.buttons, translation_method_value, "LLM") # 目标语言 (翻译) target_lang_trans_value = config.get("target_language_translation", "简体中文") self._set_radio_button(self.target_language_translation.buttons, target_lang_trans_value, "简体中文") # TTS方法 tts_method_value = config.get("tts_method", "EdgeTTS") self._set_radio_button(self.tts_method.buttons, tts_method_value, "EdgeTTS") # 目标语言 (TTS) target_lang_tts_value = config.get("target_language_tts", "中文") self._set_radio_button(self.target_language_tts.buttons, target_lang_tts_value, "中文") # EdgeTTS声音选择 edge_tts_voice_value = config.get("edge_tts_voice", "zh-CN-XiaoxiaoNeural") self._set_radio_button(self.edge_tts_voice.buttons, edge_tts_voice_value, "zh-CN-XiaoxiaoNeural") # 添加字幕 add_subtitles_value = config.get("add_subtitles", True) self._set_radio_button(self.add_subtitles.buttons, add_subtitles_value, True) # 加速倍数 self.speed_factor.setValue(config.get("speed_factor", 1.00)) # 帧率 self.frame_rate.setValue(config.get("frame_rate", 30)) # 背景音乐 if config.get("background_music"): self.background_music.file_path.setText(config.get("background_music")) # 背景音乐音量 self.bg_music_volume.setValue(config.get("bg_music_volume", 0.5)) # 视频音量 self.video_volume.setValue(config.get("video_volume", 1.0)) # 输出分辨率 output_resolution_value = config.get("output_resolution", "1080p") self._set_radio_button(self.output_resolution.buttons, output_resolution_value, "1080p") # 最大工作线程数 self.max_workers.setValue(config.get("max_workers", 1)) # 最大重试次数 self.max_retries.setValue(config.get("max_retries", 3)) except Exception as e: QMessageBox.warning(self, "配置加载错误", f"加载配置时出错: {str(e)}") def _set_radio_button(self, buttons, value, default_value): """辅助方法：安全地设置单选按钮值""" try: # 尝试找到匹配的选项并选中对应的按钮 for option, button in buttons: if option == value: button.setChecked(True) return # 如果没找到，使用默认值 for option, button in buttons: if option == default_value: button.setChecked(True) return except Exception: # 如果出现任何错误，尝试使用默认值 for option, button in buttons: if option == default_value: button.setChecked(True) return def save_config(self): """保存配置到JSON文件""" try: config = self.get_config() config_dir = os.path.dirname(os.path.abspath(__file__)) config_path = os.path.join(config_dir, "config.json") with open(config_path, 'w', encoding='utf-8') as f: json.dump(config, f, ensure_ascii=False, indent=4) QMessageBox.information(self, "保存成功", f"配置已保存到 {config_path}") # 发送配置变更信号 self.config_changed.emit(config) except Exception as e: QMessageBox.critical(self, "保存失败", f"保存配置时出错: {str(e)}") def load_config(self): """从JSON文件加载配置""" try: config_dir = os.path.dirname(os.path.abspath(__file__)) config_path = os.path.join(config_dir, "config.json") if os.path.exists(config_path): with open(config_path, 'r', encoding='utf-8') as f: config = json.load(f) self.apply_config(config) except Exception as e: QMessageBox.warning(self, "加载配置失败", f"加载配置时出错: {str(e)}") def reset_config(self): """重置配置为默认值""" if QMessageBox.question(self, "确认重置", "确定要重置所有配置到默认值吗？") == QMessageBox.Yes: # 应用默认配置 default_config = { "video_folder": "videos", "resolution": "1080p", "video_count": 5, "model": "htdemucs_ft", "device": "auto", "shifts": 5, "asr_model": "WhisperX", "whisperx_size": "large", "batch_size": 32, "separate_speakers": True, "min_speakers": None, "max_speakers": None, "translation_method": "LLM", "target_language_translation": "简体中文", "tts_method": "EdgeTTS", "target_language_tts": "中文", "edge_tts_voice": "zh-CN-XiaoxiaoNeural", "add_subtitles": True, "speed_factor": 1.00, "frame_rate": 30, "background_music": None, "bg_music_volume": 0.5, "video_volume": 1.0, "output_resolution": "1080p", "max_workers": 1, "max_retries": 3 } self.apply_config(default_config) QMessageBox.information(self, "重置成功", "所有配置已重置为默认值")