utils.py

import json
from file_name import *
import os


def load_groups_json():
    if not os.path.exists(sample_groups_json) or not os.path.exists(speaker_groups_json):
        print("JSON file doesn't exist")
        return [], {}

    with open(sample_groups_json, "r", encoding="utf-8") as json_file_sample:
        sample_groups_list: list = json.load(json_file_sample)
    with open(speaker_groups_json, "r", encoding="utf-8") as json_file_speaker:
        speaker_groups_dict: dict = json.load(json_file_speaker)
    return sample_groups_list, speaker_groups_dict


def write_transcribe_subtitle_file(input_file, transcribe_txt_list: list, subtitle_txt_list: list, adjustment: bool):
    transcribe = transcribe_file
    subtitle = subtitle_file
    if adjustment:
        transcribe = transcribe_adjusted_file
        subtitle = subtitle_adjusted_file

    with open(transcribe, "w", encoding="utf-8") as file:
        file.writelines(transcribe_txt_list)
    with open(subtitle, "w", encoding="utf-8") as file:
        file.writelines(subtitle_txt_list)


def read_transcribe_subtitle_file(input_file, adjustment: bool):
    transcribe = transcribe_file
    subtitle = subtitle_file
    if adjustment:
        transcribe = transcribe_adjusted_file
        subtitle = subtitle_adjusted_file

    if not os.path.exists(transcribe) or not os.path.exists(subtitle):
        print("Transcribe or subtitle file doesn't exist")
        return [], []

    with open(transcribe, "r", encoding="utf-8") as file:
        transcribe_txt_list = file.readlines()
    with open(subtitle, "r", encoding="utf-8") as file:
        subtitle_txt_list = file.readlines()
    return transcribe_txt_list, subtitle_txt_list