Adding support for modifying maximum sentence length when outputting to .srt via CLI argument (#14)

Author: jwcounts

Sources/yap/OutputFormat.swift

@@ -15,7 +15,7 @@ enum OutputFormat: String, EnumerableFlag {
         }
     }
 
-    func text(for transcript: AttributedString) -> String {
+    func text(for transcript: AttributedString, maxLength: Int) -> String {
         switch self {
         case .txt:
             return String(transcript.characters)
@@ -28,7 +28,7 @@ enum OutputFormat: String, EnumerableFlag {
                 return String(format: "%0.2d:%0.2d:%0.2d,%0.3d", h, m, s, ms)
             }
 
-            return transcript.sentences(maxLength: 40).compactMap { (sentence: AttributedString) -> (CMTimeRange, String)? in
+            return transcript.sentences(maxLength: maxLength).compactMap { (sentence: AttributedString) -> (CMTimeRange, String)? in
                 guard let timeRange = sentence.audioTimeRange else { return nil }
                 return (timeRange, String(sentence.characters))
             }.enumerated().map { index, run in

Sources/yap/Transcribe.swift

@@ -31,6 +31,11 @@ import Speech
         transform: URL.init(fileURLWithPath:)
     ) var outputFile: URL?
 
+    @Option(
+        name: .shortAndLong,
+        help: "Maximum sentence length in characters. If not provided, it will be set to 40.",
+    ) var maxLength: Int = 40
+
     mutating func run() async throws {
         let piped = isatty(STDOUT_FILENO) == 0
         struct DevNull: StandardPipelining { func write(content _: String) {} }
@@ -112,7 +117,7 @@ import Speech
         }
 
         if let outputFile {
-            try outputFormat.text(for: transcript).write(
+            try outputFormat.text(for: transcript, maxLength: maxLength).write(
                 to: outputFile,
                 atomically: false,
                 encoding: .utf8
@@ -121,7 +126,7 @@ import Speech
         }
 
         if piped || outputFile == nil {
-            print(outputFormat.text(for: transcript))
+            print(outputFormat.text(for: transcript, maxLength: maxLength))
         }
     }
 }