From 509c0cab65b2fbfea9285cb9c67bb07c0199caf5 Mon Sep 17 00:00:00 2001
From: Konstantin <const@const.me>
Date: Mon, 30 Jan 2023 23:49:04 +0100
Subject: “Text with timestamps” output format option
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 Examples/WhisperDesktop/TranscribeDlg.cpp |  35 ++++++++++++++++++++++--------
 Examples/WhisperDesktop/TranscribeDlg.h   |   2 +-
 Examples/WhisperDesktop/WhisperDesktop.rc | Bin 20384 -> 20384 bytes
 3 files changed, 27 insertions(+), 10 deletions(-)

diff --git a/Examples/WhisperDesktop/TranscribeDlg.cpp b/Examples/WhisperDesktop/TranscribeDlg.cpp
index 4f4944d..742c1d4 100644
--- a/Examples/WhisperDesktop/TranscribeDlg.cpp
+++ b/Examples/WhisperDesktop/TranscribeDlg.cpp
@@ -108,7 +108,8 @@ void TranscribeDlg::printModelDescription()
 void TranscribeDlg::populateOutputFormats()
 {
 	transcribeOutFormat.AddString( L"None" );
-	transcribeOutFormat.AddString( L"Text File" );
+	transcribeOutFormat.AddString( L"Text file" );
+	transcribeOutFormat.AddString( L"Text with timestamps" );
 	transcribeOutFormat.AddString( L"SubRip subtitles" );
 	transcribeOutFormat.AddString( L"WebVTT subtitles" );
 }
@@ -117,8 +118,9 @@ enum struct TranscribeDlg::eOutputFormat : uint8_t
 {
 	None = 0,
 	Text = 1,
-	SubRip = 2,
-	WebVTT = 3
+	TextTimestamps = 2,
+	SubRip = 3,
+	WebVTT = 4,
 };
 
 // CBN_SELCHANGE notification for IDC_OUTPUT_FORMAT combobox
@@ -163,10 +165,10 @@ void TranscribeDlg::onBrowseMedia()
 		setOutputPath( path );
 }
 
-static const LPCTSTR outputFilters = L"Text files (*.txt)\0*.txt\0SubRip subtitles (*.srt)\0*.srt\0WebVTT subtitles (*.vtt)\0*.vtt\0\0";
-static const std::array<LPCTSTR, 3> outputExtensions =
+static const LPCTSTR outputFilters = L"Text files (*.txt)\0*.txt\0Text with timestamps (*.txt)\0*.txt\0SubRip subtitles (*.srt)\0*.srt\0WebVTT subtitles (*.vtt)\0*.vtt\0\0";
+static const std::array<LPCTSTR, 4> outputExtensions =
 {
-	L".txt", L".srt", L".vtt"
+	L".txt", L".txt", L".srt", L".vtt"
 };
 
 void TranscribeDlg::setOutputPath( const CString& input )
@@ -414,7 +416,9 @@ HRESULT TranscribeDlg::transcribe()
 	switch( format )
 	{
 	case eOutputFormat::Text:
-		return writeTextFile( segments, len.countSegments, outputFile );
+		return writeTextFile( segments, len.countSegments, outputFile, false );
+	case eOutputFormat::TextTimestamps:
+		return writeTextFile( segments, len.countSegments, outputFile, true );
 	case eOutputFormat::SubRip:
 		return writeSubRip( segments, len.countSegments, outputFile );
 	case eOutputFormat::WebVTT:
@@ -479,14 +483,27 @@ namespace
 using Whisper::sSegment;
 
 
-HRESULT TranscribeDlg::writeTextFile( const sSegment* const segments, const size_t length, CAtlFile& file )
+HRESULT TranscribeDlg::writeTextFile( const sSegment* const segments, const size_t length, CAtlFile& file, bool timestamps )
 {
 	using namespace Whisper;
 	CHECK( writeUtf8Bom( file ) );
 	CStringA line;
 	for( size_t i = 0; i < length; i++ )
 	{
-		line = skipBlank( segments[ i ].text );
+		const sSegment& seg = segments[ i ];
+
+		if( timestamps )
+		{
+			line = "[";
+			printTimeStamp( line, seg.time.begin );
+			line += " --> ";
+			printTimeStamp( line, seg.time.end );
+			line += "]  ";
+		}
+		else
+			line = "";
+
+		line += skipBlank( seg.text );
 		line += "\r\n";
 		CHECK( write( file, line ) );
 	}
diff --git a/Examples/WhisperDesktop/TranscribeDlg.h b/Examples/WhisperDesktop/TranscribeDlg.h
index 800d8eb..4ea2916 100644
--- a/Examples/WhisperDesktop/TranscribeDlg.h
+++ b/Examples/WhisperDesktop/TranscribeDlg.h
@@ -116,7 +116,7 @@ private:
 	HRESULT transcribe();
 	void getThreadError();
 	
-	static HRESULT writeTextFile( const Whisper::sSegment* const segments, const size_t length, CAtlFile& file );
+	static HRESULT writeTextFile( const Whisper::sSegment* const segments, const size_t length, CAtlFile& file, bool timestamps );
 	static HRESULT writeSubRip( const Whisper::sSegment* const segments, const size_t length, CAtlFile& file );
 	static HRESULT writeWebVTT( const Whisper::sSegment* const segments, const size_t length, CAtlFile& file );
 
diff --git a/Examples/WhisperDesktop/WhisperDesktop.rc b/Examples/WhisperDesktop/WhisperDesktop.rc
index 3e9aeb4..5de9967 100644
Binary files a/Examples/WhisperDesktop/WhisperDesktop.rc and b/Examples/WhisperDesktop/WhisperDesktop.rc differ
-- 
cgit v1.2.3