fix srt writer and parser for standard time format (hh:mm:ss,ms)

utils/subtitler.py

@@ -10,8 +10,8 @@ def parse_srt(srt_file):
     while i < len(lines):
         if lines[i].strip().isdigit():
             timing_str = lines[i+1].strip().split(" --> ")
-            start = timing_str[0].replace(',', '.')
-            end = timing_str[1].replace(',', '.')
+            start = timing_str[0]
+            end = timing_str[1]
             text = lines[i+2].strip()
             subtitles.append((start, end, text))
             i += 4

utils/transcriber.py

@@ -8,6 +8,19 @@ logging.basicConfig(filename='main.log',
                 datefmt='%m/%d/%Y %I:%M:%S %p')
 logging.getLogger("faster_whisper").setLevel(logging.DEBUG)
 
+def convert_seconds_to_time(seconds):
+    # Separate seconds into hours, minutes, and seconds
+    seconds = float(seconds)
+    hours, remainder = divmod(seconds, 3600)
+    minutes, remainder = divmod(remainder, 60)
+    whole_seconds = int(remainder)
+    milliseconds = int((remainder - whole_seconds) * 1000)
+
+    # Format the time string
+    time_string = f"{int(hours):02}:{int(minutes):02}:{whole_seconds:02},{milliseconds:03}"
+    
+    return time_string
+
 def write_srt(segments, srt_path, max_words_per_line):
     """Write segments to an SRT file with a maximum number of words per line."""
     with open(srt_path, "w", encoding='utf-8') as file:
@@ -19,8 +32,8 @@ def write_srt(segments, srt_path, max_words_per_line):
                 # Write the line if max words limit reached or it's the last word in the segment
                 if len(words_in_line) == max_words_per_line or w == len(segment.words) - 1:
                     if words_in_line:  # Check to avoid writing a line if there are no words
-                        start_time = words_in_line[0].start
-                        end_time = words_in_line[-1].end
+                        start_time = convert_seconds_to_time(words_in_line[0].start)
+                        end_time = convert_seconds_to_time(words_in_line[-1].end)
                         line_text = ' '.join([w.word.strip() for w in words_in_line])
                         file.write(f"{line_counter}\n{start_time} --> {end_time}\n{line_text}\n\n")
                         # Reset for the next line and increment line counter