bazarr-ai-sub-generator/auto_subtitle/utils/files.py

import os
from typing import Iterator, TextIO
from .convert import format_timestamp

def write_srt(transcript: Iterator[dict], file: TextIO):
    for i, segment in enumerate(transcript, start=1):
        print(
            f"{i}\n"
            f"{format_timestamp(segment.start, always_include_hours=True)} --> "
            f"{format_timestamp(segment.end, always_include_hours=True)}\n"
            f"{segment.text.strip().replace('-->', '->')}\n",
            file=file,
            flush=True,
        )

def filename(path: str):
    return os.path.splitext(os.path.basename(path))[0]