models/subtitles.py

def save_subtitles(prediction, output_file):
    ## Save prediction to subtitles file format
    with open(output_file, "w") as f:
        for i, chunk in enumerate(prediction):
            start_time = format_time(chunk['timestamp'][0])
            end_time = format_time(chunk['timestamp'][1])
            f.write(f"{i+1}\n")
            f.write(f"{start_time} --> {end_time}\n")
            f.write(f"{chunk['text']}\n")
            f.write("\n")

def format_time(seconds):
    hours = int(seconds // 3600)
    minutes = int((seconds % 3600) // 60)
    seconds = seconds % 60
    return f"{hours:02d}:{minutes:02d}:{seconds:04.1f}"