deblan-mirror/buzz
1
0
Fork 0
mirror of https://github.com/chidiwilliams/buzz.git synced 2026-03-18 08:29:53 +01:00
Code Issues Projects Releases Packages Wiki Activity

Add new lines in TXT segments (#436)

Browse source
This commit is contained in:
Chidi Williams 2023-05-01 10:15:41 +00:00 • committed by GitHub
commit 5892e3a2a1
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 3 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

4
buzz/transcriber.py
View file

@ -596,9 +596,7 @@ def write_output(path: str, segments: List[Segment], output_format: OutputFormat
if output_format == OutputFormat.TXT:
for (i, segment) in enumerate(segments):
file.write(segment.text)
if i < len(segments) - 1:
file.write(' ')
file.write('\n')
file.write('\n')
elif output_format == OutputFormat.VTT:
file.write('WEBVTT\n\n')

2
tests/transcriber_test.py
View file

@ -209,7 +209,7 @@ class TestWhisperCpp:
@pytest.mark.parametrize(
'output_format,output_text',
[
(OutputFormat.TXT, 'Bien venue dans\n'),
(OutputFormat.TXT, 'Bien\nvenue 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'),

2
tests/widgets/transcription_viewer_test.py
View file

@ -71,4 +71,4 @@ class TestTranscriptionViewerWidget:
export_button.menu().actions()[0].trigger()
output_file = open(output_file_path, 'r', encoding='utf-8')
assert 'Bien venue dans' in output_file.read()
assert 'Bien\nvenue dans' in output_file.read()