video-to-subs / write_srt.py
abnerh's picture
Add necessary file
262d511
import os
import datetime
def write_to_file(file_handle, inferred_text, line_count, limits):
"""Write the inferred text to SRT file
Follows a specific format for SRT files
Args:
file_handle : SRT file handle
inferred_text : text to be written
line_count : subtitle line count
limits : starting and ending times for text
"""
sep = ','
d = str(datetime.timedelta(seconds=float(limits[0])))
try:
from_dur = "0" + str(d.split(".")[0]) + sep + str(d.split(".")[-1][:2])
except:
from_dur = "0" + str(d) + sep + "00"
d = str(datetime.timedelta(seconds=float(limits[1])))
try:
to_dur = "0" + str(d.split(".")[0]) + sep + str(d.split(".")[-1][:2])
except:
to_dur = "0" + str(d) + sep + "00"
file_handle.write(str(line_count) + "\n")
file_handle.write(from_dur + " --> " + to_dur + "\n")
file_handle.write(inferred_text + "\n\n")