deblan-mirror/buzz
1
0
Fork 0
mirror of https://github.com/chidiwilliams/buzz.git synced 2024-06-26 11:40:09 +02:00
Code Issues Packages Projects Releases Wiki Activity

Use comma as SRT separator (#280)

Browse source
This commit is contained in:
Chidi Williams 2023-01-01 12:45:39 +00:00 committed by GitHub
parent f6ef2d5fe3
commit 5f9045c347
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 5 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
.coveragerc
View file

@ -7,4 +7,4 @@ omit =
directory = coverage/html
[report]
fail_under = 74
fail_under = 73

6
buzz/transcriber.py
View file

@ -474,7 +474,7 @@ def write_output(path: str, segments: List[Segment], should_open: bool, output_f
for (i, segment) in enumerate(segments):
file.write(f'{i + 1}\n')
file.write(
f'{to_timestamp(segment.start)} --> {to_timestamp(segment.end)}\n')
f'{to_timestamp(segment.start, ms_separator=",")} --> {to_timestamp(segment.end, ms_separator=",")}\n')
file.write(f'{segment.text}\n\n')
if should_open:
@ -495,14 +495,14 @@ def segments_to_text(segments: List[Segment]) -> str:
return result
def to_timestamp(ms: float) -> str:
def to_timestamp(ms: float, ms_separator='.') -> str:
hr = int(ms / (1000 * 60 * 60))
ms = ms - hr * (1000 * 60 * 60)
min = int(ms / (1000 * 60))
ms = ms - min * (1000 * 60)
sec = int(ms / 1000)
ms = int(ms - sec * 1000)
return f'{hr:02d}:{min:02d}:{sec:02d}.{ms:03d}'
return f'{hr:02d}:{min:02d}:{sec:02d}{ms_separator}{ms:03d}'
SUPPORTED_OUTPUT_FORMATS = 'Audio files (*.mp3 *.wav *.m4a *.ogg);;\

2
tests/transcriber_test.py
View file

@ -205,7 +205,7 @@ class TestWhisperCpp:
(OutputFormat.TXT, 'Bien venue dans\n'),
(
OutputFormat.SRT,
'1\n00:00:00.040 --> 00:00:00.299\nBien\n\n2\n00:00:00.299 --> 00:00:00.329\nvenue dans\n\n'),
'1\n00:00:00,040 --> 00:00:00,299\nBien\n\n2\n00:00:00,299 --> 00:00:00,329\nvenue dans\n\n'),
(OutputFormat.VTT,
'WEBVTT\n\n00:00:00.040 --> 00:00:00.299\nBien\n\n00:00:00.299 --> 00:00:00.329\nvenue dans\n\n'),
])