Spaces:
Sleeping
Sleeping
File size: 851 Bytes
09cabee |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 |
from datetime import timedelta
import os
import whisper
class SRT_segment(object):
def __init__(self, segment) -> None:
self.start_time_str = str(0)+str(timedelta(seconds=int(segment['start'])))+',000'
self.end_time_str = str(0)+str(timedelta(seconds=int(segment['end'])))+',000'
self.segment_id = segment['id']+1
self.source_text = segment['text']
self.duration = f"{self.start_time_str} --> {self.end_time_str}"
self.translation = ""
class SRT_script():
def __init__(self, segments) -> None:
self.segments = []
for seg in segments:
srt_seg = SRT_segment(seg)
self.segments.append(srt_seg)
def get_source_only():
# return a string
pass
def write_srt_file(path:str):
# write srt file to path
pass
|