import json
import re


def parse_vtt(vtt_filename):
    with open(vtt_filename, "r", encoding="utf-8") as file:
        lines = file.readlines()

    time_pattern = re.compile(r"(\d+\.\d{3}) --> (\d+\.\d{3})")

    subtitles = []
    current_subtitle = {}

    for line in lines[1:]:
        match = time_pattern.match(line)
        if match:
            current_subtitle["start"] = float(match.group(1))
            current_subtitle["end"] = float(match.group(2))
            current_subtitle["content"] = ""
        elif line.strip() == "":
            if current_subtitle:
                if current_subtitle["content"][-1] == "\n":
                    current_subtitle["content"] = current_subtitle["content"][:-1]
                subtitles.append(current_subtitle)
                current_subtitle = {}
        else:
            current_subtitle["content"] += (
                line.strip() + "\n"
            )  # Space to separate lines

    if current_subtitle:
        if current_subtitle["content"][-1] == "\n":
            current_subtitle["content"] = current_subtitle["content"][:-1]
        subtitles.append(current_subtitle)

    return subtitles


def subtitles_to_backup(subtitles):
    backup_data = {
        "subtitles": subtitles,
        "script_lines": [],
        "line_index": len(subtitles),
        "current_subtitle": {},
        "play": 0,
    }
    return backup_data


def main(vtt_filename, output_filename):
    subtitles = parse_vtt(vtt_filename)
    backup_data = subtitles_to_backup(subtitles)

    with open(output_filename, "w", encoding="utf-8") as json_file:
        json.dump(backup_data, json_file, indent=2)


vtt_filename = "audio.vtt"
output_filename = "backup2.json"
main(vtt_filename, output_filename)