GPT-SoVITS/Ref_Audio_Selector/common/common.py

from tools import my_utils
from config import python_exec, is_half
import subprocess
import sys
import os


class RefAudioListManager:
    def __init__(self, root_dir):
        self.audio_dict = {'default': []}
        absolute_root = os.path.abspath(root_dir)

        for subdir, dirs, files in os.walk(absolute_root):
            relative_path = os.path.relpath(subdir, absolute_root)

            if relative_path == '.':
                category = 'default'
            else:
                category = relative_path.replace(os.sep, '')

            for file in files:
                if file.endswith('.wav'):
                    # 将相对路径转换为绝对路径
                    audio_abs_path = os.path.join(subdir, file)
                    if category not in self.audio_dict:
                        self.audio_dict[category] = []
                    self.audio_dict[category].append(audio_abs_path)

    def get_audio_list(self):
        return self.audio_dict

    def get_flattened_audio_list(self):
        all_audio_files = []
        for category_audios in self.audio_dict.values():
            all_audio_files.extend(category_audios)
        return all_audio_files

    def get_ref_audio_list(self):
        audio_info_list = []
        for category, audio_paths in self.audio_dict.items():
            for audio_path in audio_paths:
                filename_without_extension = os.path.splitext(os.path.basename(audio_path))[0]
                audio_info = {
                    'emotion': f"{category}-{filename_without_extension}",
                    'ref_path': audio_path,
                    'ref_text': filename_without_extension,
                }
                audio_info_list.append(audio_info)
        return audio_info_list


def batch_clean_paths(paths):
    """
    批量处理路径列表，对每个路径调用 clean_path() 函数。

    参数:
        paths (list[str]): 包含待处理路径的列表。

    返回:
        list[str]: 经过 clean_path() 处理后的路径列表。
    """
    cleaned_paths = []
    for path in paths:
        cleaned_paths.append(my_utils.clean_path(path))
    return cleaned_paths


def read_text_file_to_list(file_path):
    # 按照UTF-8编码打开文件（确保能够正确读取中文）
    with open(file_path, mode='r', encoding='utf-8') as file:
        # 读取所有行并存储到一个列表中
        lines = file.read().splitlines()
    return lines


def get_filename_without_extension(file_path):
    """
    Given a file path string, returns the file name without its extension.

    Parameters:
    file_path (str): The full path to the file.

    Returns:
    str: The file name without its extension.
    """
    base_name = os.path.basename(file_path)  # Get the base name (file name with extension)
    file_name, file_extension = os.path.splitext(base_name)  # Split the base name into file name and extension
    return file_name  # Return the file name without extension


def read_file(file_path):
    # 使用with语句打开并读取文件
    with open(file_path, 'r', encoding='utf-8') as file:  # 'r' 表示以读取模式打开文件
        # 一次性读取文件所有内容
        file_content = file.read()

    # 文件在with语句结束时会自动关闭
    # 现在file_content变量中存储了文件的所有文本内容
    return file_content


def write_text_to_file(text, output_file_path):
    try:
        with open(output_file_path, 'w', encoding='utf-8') as file:
            file.write(text)
    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}")


def check_path_existence_and_return(path):
    """
    检查给定路径（文件或目录）是否存在。如果存在，返回该路径；否则，返回空字符串。
    :param path: 待检查的文件或目录路径（字符串）
    :return: 如果路径存在，返回原路径；否则，返回空字符串
    """
    if os.path.exists(path):
        return path
    else:
        return ""


def open_file(filepath):
    if sys.platform.startswith('darwin'):
        subprocess.run(['open', filepath])  # macOS
    elif os.name == 'nt':  # For Windows
        os.startfile(filepath)
    elif os.name == 'posix':  # For Linux, Unix, etc.
        subprocess.run(['xdg-open', filepath])


def start_new_service(script_path):
    # 对于Windows系统
    if sys.platform.startswith('win'):
        cmd = f'start cmd /k {python_exec} {script_path}'
    # 对于Mac或者Linux系统
    else:
        cmd = f'xterm -e {python_exec} {script_path}'

    proc = subprocess.Popen(cmd, shell=True)

    # 关闭之前启动的子进程
    # proc.terminate()

    # 或者如果需要强制关闭可以使用
    # proc.kill()

    return proc


if __name__ == '__main__':
    dir = r'C:\Users\Administrator\Desktop/test'
    dir2 = r'"C:\Users\Administrator\Desktop\test2"'
    dir, dir2 = batch_clean_paths([dir, dir2])
    print(dir, dir2)