Add new lines in TXT segments (#436)

chidiwilliams · May 1, 2023 · 5892e3a · 5892e3a
1 parent 0804c27
commit 5892e3a
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 5 deletions.
diff --git a/buzz/transcriber.py b/buzz/transcriber.py
@@ -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')

diff --git a/tests/transcriber_test.py b/tests/transcriber_test.py
@@ -209,7 +209,7 @@ def test_transcribe(self):
 @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'),

diff --git a/tests/widgets/transcription_viewer_test.py b/tests/widgets/transcription_viewer_test.py
@@ -71,4 +71,4 @@ def test_should_export_segments(self, tmp_path: pathlib.Path, qtbot: QtBot, task
             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()