utils.py

import json
from file_name import *
import os


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

    with open(get_sample_groups_json(), "r", encoding="utf-8") as json_file_sample:
        sample_groups_list: list = json.load(json_file_sample)
    with open(get_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(transcribe_txt_list: list, subtitle_txt_list: list, adjustment: bool):
    transcribe = get_transcribe_file()
    subtitle = get_subtitle_file()
    if adjustment:
        transcribe = get_transcribe_adjusted_file()
        subtitle = get_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(adjustment: bool):
    transcribe = get_transcribe_file()
    subtitle = get_subtitle_file()
    if adjustment:
        transcribe = get_transcribe_adjusted_file()
        subtitle = get_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