diff --git a/Ref_Audio_Selector/common/common.py b/Ref_Audio_Selector/common/common.py index 998ae6a..3d74674 100644 --- a/Ref_Audio_Selector/common/common.py +++ b/Ref_Audio_Selector/common/common.py @@ -2,6 +2,7 @@ from tools import my_utils import glob import os + class RefAudioListManager: def __init__(self, root_dir): self.audio_dict = {'default': []} @@ -45,6 +46,7 @@ class RefAudioListManager: audio_info_list.append(audio_info) return audio_info_list + def batch_clean_paths(paths): """ 批量处理路径列表，对每个路径调用 clean_path() 函数。 @@ -102,4 +104,4 @@ def write_text_to_file(text, output_file_path): except IOError as e: print(f"Error occurred while writing to the file: {e}") else: - print(f"Text successfully written to file: {output_file_path}") \ No newline at end of file + print(f"Text successfully written to file: {output_file_path}") diff --git a/Ref_Audio_Selector/ref_audio_selector_webui.py b/Ref_Audio_Selector/ref_audio_selector_webui.py index ef0ac0a..c96523a 100644 --- a/Ref_Audio_Selector/ref_audio_selector_webui.py +++ b/Ref_Audio_Selector/ref_audio_selector_webui.py @@ -6,6 +6,7 @@ import gradio as gr import Ref_Audio_Selector.tool.audio_similarity as audio_similarity import Ref_Audio_Selector.tool.audio_inference as audio_inference import Ref_Audio_Selector.tool.audio_config as audio_config +import Ref_Audio_Selector.tool.delete_inference_with_ref as delete_inference_with_ref import Ref_Audio_Selector.common.common as common from tools.i18n.i18n import I18nAuto from config import python_exec, is_half @@ -23,7 +24,7 @@ p_text_similarity = None # 校验基础信息 def check_base_info(text_work_space_dir): if text_work_space_dir is None or text_work_space_dir == '': - raise Exception(i18n("工作目录不能为空")) + raise Exception("工作目录不能为空") # 从list文件，提取参考音频 @@ -34,13 +35,13 @@ def convert_from_list(text_work_space_dir, text_list_input): try: check_base_info(text_work_space_dir) if text_list_input is None or text_list_input == '': - raise Exception(i18n("list文件路径不能为空")) + raise Exception("list文件路径不能为空") audio_similarity.convert_from_list(text_list_input, ref_audio_all) except Exception as e: traceback.print_exc() text_convert_from_list_info = f"发生异常：{e}" text_sample_dir = '' - return [text_convert_from_list_info, text_sample_dir] + return i18n(text_convert_from_list_info), text_sample_dir def start_similarity_analysis(work_space_dir, sample_dir, base_voice_path, need_similarity_output): @@ -83,19 +84,19 @@ def sample(text_work_space_dir, text_sample_dir, text_base_voice_path, try: check_base_info(text_work_space_dir) if text_sample_dir is None or text_sample_dir == '': - raise Exception(i18n("参考音频抽样目录不能为空，请先完成上一步操作")) + raise Exception("参考音频抽样目录不能为空，请先完成上一步操作") if text_base_voice_path is None or text_base_voice_path == '': - raise Exception(i18n("基准音频路径不能为空")) + raise Exception("基准音频路径不能为空") if text_subsection_num is None or text_subsection_num == '': - raise Exception(i18n("分段数不能为空")) + raise Exception("分段数不能为空") if text_sample_num is None or text_sample_num == '': - raise Exception(i18n("每段随机抽样个数不能为空")) + raise Exception("每段随机抽样个数不能为空") similarity_list, _, _ = start_similarity_analysis(text_work_space_dir, text_sample_dir, text_base_voice_path, checkbox_similarity_output) if similarity_list is None: - raise Exception(i18n("相似度分析失败")) + raise Exception("相似度分析失败") audio_similarity.sample(ref_audio_dir, similarity_list, int(text_subsection_num), int(text_sample_num)) @@ -106,8 +107,7 @@ def sample(text_work_space_dir, text_sample_dir, text_base_voice_path, text_model_inference_voice_dir = ref_audio_dir text_sync_ref_audio_dir = ref_audio_dir text_sync_ref_audio_dir2 = ref_audio_dir - return [text_sample_info, text_model_inference_voice_dir, text_sync_ref_audio_dir, - text_sync_ref_audio_dir2] + return i18n(text_sample_info), text_model_inference_voice_dir, text_sync_ref_audio_dir, text_sync_ref_audio_dir2 # 根据参考音频和测试文本，执行批量推理 @@ -120,31 +120,31 @@ def model_inference(text_work_space_dir, text_model_inference_voice_dir, text_ur try: check_base_info(text_work_space_dir) if text_model_inference_voice_dir is None or text_model_inference_voice_dir == '': - raise Exception(i18n("待推理的参考音频所在目录不能为空，请先完成上一步操作")) + raise Exception("待推理的参考音频所在目录不能为空，请先完成上一步操作") if text_url is None or text_url == '': - raise Exception(i18n("推理服务请求地址不能为空")) + raise Exception("推理服务请求地址不能为空") if text_text is None or text_text == '': - raise Exception(i18n("文本参数名不能为空")) + raise Exception("文本参数名不能为空") if text_test_content is None or text_test_content == '': - raise Exception(i18n("待推理文本路径不能为空")) + raise Exception("待推理文本路径不能为空") if (text_ref_path is None or text_ref_path == '') and (text_ref_text is None or text_ref_text == '') and ( text_emotion is None or text_emotion == ''): - raise Exception(i18n("参考音频路径/文本和角色情绪二选一填写，不能全部为空")) + raise Exception("参考音频路径/文本和角色情绪二选一填写，不能全部为空") url_composer = audio_inference.URLComposer(text_url, text_emotion, text_text, text_ref_path, text_ref_text) url_composer.is_valid() text_list = common.read_text_file_to_list(text_test_content) if text_list is None or len(text_list) == 0: - raise Exception(i18n("待推理文本内容不能为空")) + raise Exception("待推理文本内容不能为空") ref_audio_manager = common.RefAudioListManager(text_model_inference_voice_dir) if len(ref_audio_manager.get_audio_list()) == 0: - raise Exception(i18n("待推理的参考音频不能为空")) + raise Exception("待推理的参考音频不能为空") audio_inference.generate_audio_files(url_composer, text_list, ref_audio_manager.get_ref_audio_list(), inference_dir) except Exception as e: traceback.print_exc() text_model_inference_info = f"发生异常：{e}" text_asr_audio_dir = '' - return [text_model_inference_info, text_asr_audio_dir] + return i18n(text_model_inference_info), text_asr_audio_dir # 对推理生成音频执行asr @@ -156,13 +156,13 @@ def asr(text_work_space_dir, text_asr_audio_dir, dropdown_asr_model, try: check_base_info(text_work_space_dir) if text_asr_audio_dir is None or text_asr_audio_dir == '': - raise Exception(i18n("待asr的音频所在目录不能为空，请先完成上一步操作")) + raise Exception("待asr的音频所在目录不能为空，请先完成上一步操作") if dropdown_asr_model is None or dropdown_asr_model == '': - raise Exception(i18n("asr模型不能为空")) + raise Exception("asr模型不能为空") if dropdown_asr_size is None or dropdown_asr_size == '': - raise Exception(i18n("asr模型大小不能为空")) + raise Exception("asr模型大小不能为空") if dropdown_asr_lang is None or dropdown_asr_lang == '': - raise Exception(i18n("asr语言不能为空")) + raise Exception("asr语言不能为空") asr_file = open_asr(text_asr_audio_dir, text_work_space_dir, dropdown_asr_model, dropdown_asr_size, dropdown_asr_lang) text_text_similarity_analysis_path = asr_file @@ -171,7 +171,7 @@ def asr(text_work_space_dir, text_asr_audio_dir, dropdown_asr_model, traceback.print_exc() text_asr_info = f"发生异常：{e}" text_text_similarity_analysis_path = '' - return [text_asr_info, text_text_similarity_analysis_path] + return i18n(text_asr_info), text_text_similarity_analysis_path def open_asr(asr_inp_dir, asr_opt_dir, asr_model, asr_model_size, asr_lang): @@ -213,12 +213,12 @@ def text_similarity_analysis(text_work_space_dir, try: check_base_info(text_work_space_dir) if text_text_similarity_analysis_path is None or text_text_similarity_analysis_path == '': - raise Exception(i18n("asr生成的文件路径不能为空，请先完成上一步操作")) + raise Exception("asr生成的文件路径不能为空，请先完成上一步操作") open_text_similarity_analysis(text_text_similarity_analysis_path, similarity_dir) except Exception as e: traceback.print_exc() text_text_similarity_analysis_info = f"发生异常：{e}" - return text_text_similarity_analysis_info + return i18n(text_text_similarity_analysis_info) def open_text_similarity_analysis(asr_file_path, output_dir, similarity_enlarge_boundary=0.8): @@ -247,38 +247,39 @@ def similarity_audio_output(text_work_space_dir, text_base_audio_path, try: check_base_info(text_work_space_dir) if text_base_audio_path is None or text_base_audio_path == '': - raise Exception(i18n("基准音频路径不能为空")) + raise Exception("基准音频路径不能为空") if text_compare_audio_dir is None or text_compare_audio_dir == '': - raise Exception(i18n("待分析的音频所在目录不能为空")) + raise Exception("待分析的音频所在目录不能为空") similarity_list, similarity_file, similarity_file_dir = start_similarity_analysis( text_work_space_dir, text_compare_audio_dir, text_base_audio_path, True) if similarity_list is None: - raise Exception(i18n("相似度分析失败")) + raise Exception("相似度分析失败") text_similarity_audio_output_info = f'相似度分析成功：生成目录{similarity_file_dir}，文件{similarity_file}' except Exception as e: traceback.print_exc() text_similarity_audio_output_info = f"发生异常：{e}" - return text_similarity_audio_output_info + return i18n(text_similarity_audio_output_info) # 根据参考音频目录的删除情况，将其同步到推理生成的音频目录中，即参考音频目录下，删除了几个参考音频，就在推理目录下，将这些参考音频生成的音频文件移除 def sync_ref_audio(text_work_space_dir, text_sync_ref_audio_dir, text_sync_inference_audio_dir): - text_sync_ref_audio_info = "同步参考音频成功：生成目录XXX" + text_sync_ref_audio_info = None try: check_base_info(text_work_space_dir) if text_sync_ref_audio_dir is None or text_sync_ref_audio_dir == '': - raise Exception(i18n("参考音频目录不能为空")) + raise Exception("参考音频目录不能为空") if text_sync_inference_audio_dir is None or text_sync_inference_audio_dir == '': - raise Exception(i18n("推理生成的音频目录不能为空")) - pass + raise Exception("推理生成的音频目录不能为空") + delete_text_wav_num, delete_emotion_dir_num = delete_inference_with_ref.sync_ref_audio(text_sync_ref_audio_dir, text_sync_inference_audio_dir) + text_sync_ref_audio_info = f"推理音频目录{text_sync_inference_audio_dir}下，text目录删除了{delete_text_wav_num}个参考音频，emotion目录下，删除了{delete_emotion_dir_num}个目录" except Exception as e: traceback.print_exc() text_sync_ref_audio_info = f"发生异常：{e}" - return text_sync_ref_audio_info + return i18n(text_sync_ref_audio_info) # 根据模板和参考音频目录，生成参考音频配置内容 @@ -288,15 +289,15 @@ def create_config(text_work_space_dir, text_template, text_sync_ref_audio_dir2): try: check_base_info(text_work_space_dir) if text_template is None or text_template == '': - raise Exception(i18n("参考音频抽样目录不能为空")) + raise Exception("参考音频抽样目录不能为空") if text_sync_ref_audio_dir2 is None or text_sync_ref_audio_dir2 == '': - raise Exception(i18n("参考音频目录不能为空")) + raise Exception("参考音频目录不能为空") ref_audio_manager = common.RefAudioListManager(text_sync_ref_audio_dir2) audio_config.generate_audio_config(text_work_space_dir, text_template, ref_audio_manager.get_ref_audio_list(), config_file) except Exception as e: traceback.print_exc() text_create_config_info = f"发生异常：{e}" - return text_create_config_info + return i18n(text_create_config_info) # 基于请求路径和参数，合成完整的请求路径 diff --git a/Ref_Audio_Selector/tool/audio_asr.py b/Ref_Audio_Selector/tool/audio_asr.py deleted file mode 100644 index fd40910..0000000 --- a/Ref_Audio_Selector/tool/audio_asr.py +++ /dev/null @@ -1,4 +0,0 @@ - - - - diff --git a/Ref_Audio_Selector/tool/audio_inference.py b/Ref_Audio_Selector/tool/audio_inference.py index c2a4634..01adee7 100644 --- a/Ref_Audio_Selector/tool/audio_inference.py +++ b/Ref_Audio_Selector/tool/audio_inference.py @@ -2,8 +2,6 @@ import os import requests import itertools from urllib.parse import urlparse, parse_qs, urlencode, urlunparse, quote -from tools.i18n.i18n import I18nAuto -i18n = I18nAuto() class URLComposer: @@ -16,13 +14,13 @@ class URLComposer: def is_valid(self): if self.base_url is None or self.base_url == '': - raise ValueError(i18n("请输入url")) + raise ValueError("请输入url") if self.text_param_name is None or self.text_param_name == '': - raise ValueError(i18n("请输入text参数名")) + raise ValueError("请输入text参数名") if self.emotion_param_name is None and self.ref_path_param_name is None and self.ref_text_param_name is None: - raise ValueError(i18n("请输入至少一个参考or情绪的参数")) + raise ValueError("请输入至少一个参考or情绪的参数") def is_emotion(self): return self.emotion_param_name is not None and self.emotion_param_name != '' diff --git a/Ref_Audio_Selector/tool/delete_inference_with_ref.py b/Ref_Audio_Selector/tool/delete_inference_with_ref.py new file mode 100644 index 0000000..d653499 --- /dev/null +++ b/Ref_Audio_Selector/tool/delete_inference_with_ref.py @@ -0,0 +1,58 @@ +import os +import shutil +import Ref_Audio_Selector.common.common as common + + +def remove_matching_audio_files_in_text_dir(text_dir, emotions_list): + count = 0 + for root, dirs, files in os.walk(text_dir): + for emotion_dict in emotions_list: + emotion_tag = emotion_dict['emotion'] + wav_file_name = f"{emotion_tag}.wav" + file_path = os.path.join(root, wav_file_name) + if os.path.exists(file_path): + print(f"Deleting file: {file_path}") + try: + os.remove(file_path) + count += 1 + except Exception as e: + print(f"Error deleting file {file_path}: {e}") + return count + + +def delete_emotion_subdirectories(emotion_dir, emotions_list): + """ + 根据给定的情绪数组，删除emotion目录下对应情绪标签的子目录。 + + 参数: + emotions_list (List[Dict]): 每个字典包含'emotion'字段。 + base_dir (str): 子目录所在的基础目录，默认为'emotion')。 + + 返回: + None + """ + count = 0 + for emotion_dict in emotions_list: + emotion_folder = emotion_dict['emotion'] + folder_path = os.path.join(emotion_dir, emotion_folder) + + # 检查emotion子目录是否存在 + if os.path.isdir(folder_path): + print(f"Deleting directory: {folder_path}") + try: + # 使用shutil.rmtree删除整个子目录及其内容 + shutil.rmtree(folder_path) + count += 1 + except Exception as e: + print(f"Error deleting directory {folder_path}: {e}") + return count + + +def sync_ref_audio(ref_audio_dir, inference_audio_dir): + ref_audio_manager = common.RefAudioListManager(ref_audio_dir) + ref_list = ref_audio_manager.get_ref_audio_list() + text_dir = os.path.join(inference_audio_dir, 'text') + emotion_dir = os.path.join(inference_audio_dir, 'emotion') + delete_text_wav_num = remove_matching_audio_files_in_text_dir(text_dir, ref_list) + delete_emotion_dir_num = delete_emotion_subdirectories(emotion_dir, ref_list) + return delete_text_wav_num, delete_emotion_dir_num diff --git a/Ref_Audio_Selector/tool/text_similarity.py b/Ref_Audio_Selector/tool/text_similarity.py deleted file mode 100644 index e69de29..0000000