Spaces:
Sleeping
Sleeping
Fix preprocessor when words are disabled
Browse files- src/utils.py +9 -8
src/utils.py
CHANGED
|
@@ -105,16 +105,17 @@ def __subtitle_preprocessor_iterator(transcript: Iterator[dict], maxLineWidth: i
|
|
| 105 |
words = segment.get('words', [])
|
| 106 |
|
| 107 |
if len(words) == 0:
|
| 108 |
-
# Yield the segment as-is
|
| 109 |
if maxLineWidth is None or maxLineWidth < 0:
|
| 110 |
yield segment
|
| 111 |
-
|
| 112 |
-
|
| 113 |
-
|
| 114 |
-
|
| 115 |
-
|
| 116 |
-
|
| 117 |
-
|
|
|
|
| 118 |
|
| 119 |
subtitle_start = segment['start']
|
| 120 |
subtitle_end = segment['end']
|
|
|
|
| 105 |
words = segment.get('words', [])
|
| 106 |
|
| 107 |
if len(words) == 0:
|
| 108 |
+
# Yield the segment as-is or processed
|
| 109 |
if maxLineWidth is None or maxLineWidth < 0:
|
| 110 |
yield segment
|
| 111 |
+
else:
|
| 112 |
+
yield {
|
| 113 |
+
'start': segment['start'],
|
| 114 |
+
'end': segment['end'],
|
| 115 |
+
'text': process_text(segment['text'].strip(), maxLineWidth)
|
| 116 |
+
}
|
| 117 |
+
# We are done
|
| 118 |
+
continue
|
| 119 |
|
| 120 |
subtitle_start = segment['start']
|
| 121 |
subtitle_end = segment['end']
|