Spaces:
Build error
Build error
| 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, "重置成功", "所有配置已重置为默认值") |