fix: no longer able to unpack segment

winstxnhdw · Jan 25, 2025 · 10db8c8 · 10db8c8
1 parent 8a52521
commit 10db8c8
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/capgen/transcriber/converter.py b/capgen/transcriber/converter.py
@@ -66,9 +66,9 @@ def to_srt(self) -> Iterator[str]:
         """
         return (
             f'{id}\n'
-            f'{self.convert_seconds_to_hhmmssmmm(start, ",")} --> '
-            f'{self.convert_seconds_to_hhmmssmmm(end, ",")}\n{text[1:]}'
-            for id, _, start, end, text, *_ in self.segments
+            f'{self.convert_seconds_to_hhmmssmmm(segment.start, ",")} --> '
+            f'{self.convert_seconds_to_hhmmssmmm(segment.end, ",")}\n{segment.text[1:]}'
+            for segment in self.segments
         )
 
     def to_vtt(self) -> Iterator[str]:
@@ -87,7 +87,7 @@ def to_vtt(self) -> Iterator[str]:
         """
         yield 'WEBVTT'
         yield from (
-            f'{self.convert_seconds_to_hhmmssmmm(start, ".")} --> '
-            f'{self.convert_seconds_to_hhmmssmmm(end, ".")}\n{text[1:]}'
-            for _, _, start, end, text, *_ in self.segments
+            f'{self.convert_seconds_to_hhmmssmmm(segment.start, ".")} --> '
+            f'{self.convert_seconds_to_hhmmssmmm(segment.end, ".")}\n{segment.text[1:]}'
+            for segment in self.segments
         )