Fix webvtt subtitle writing

jhj0517 · Dec 19, 2024 · 4fef683 · 4fef683
1 parent 46990fb
commit 4fef683
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/modules/utils/subtitle_manager.py b/modules/utils/subtitle_manager.py
@@ -253,9 +253,11 @@ class WriteVTT(SubtitlesWriter):
     def write_result(
         self, result: dict, file: TextIO, options: Optional[dict] = None, **kwargs
     ):
-        print("WEBVTT\n", file=file)
-        for start, end, text in self.iterate_result(result, options, **kwargs):
-            print(f"{start} --> {end}\n{text}\n", file=file, flush=True)
+        print("WEBVTT\n", file=file, flush=True)
+        for i, (start, end, text) in enumerate(
+            self.iterate_result(result, options, **kwargs), start=1
+        ):
+            print(f"{i}\n{start} --> {end}\n{text}\n", file=file, flush=True)
 
     def to_segments(self, file_path: str) -> List[Segment]:
         segments = []
@@ -268,7 +270,7 @@ def to_segments(self, file_path: str) -> List[Segment]:
                 index = lines[0]
                 time_line = lines[1].split(" --> ")
                 start, end = time_str_to_seconds(time_line[0], self.decimal_marker), time_str_to_seconds(time_line[1], self.decimal_marker)
-                sentence = ' '.join(lines[1:])
+                sentence = ' '.join(lines[2:])
 
                 segments.append(Segment(
                     start=start,