diff options
Diffstat (limited to 'Examples')
| -rw-r--r-- | Examples/TranscribeCS/Transcribe.cs | 37 | ||||
| -rw-r--r-- | Examples/TranscribeCS/TranscribeCS.cs | 4 |
2 files changed, 13 insertions, 28 deletions
diff --git a/Examples/TranscribeCS/Transcribe.cs b/Examples/TranscribeCS/Transcribe.cs index 6a1e500..9bbe387 100644 --- a/Examples/TranscribeCS/Transcribe.cs +++ b/Examples/TranscribeCS/Transcribe.cs @@ -69,45 +69,30 @@ namespace TranscribeCS } string speaker = ""; -#if false - if( args.diarize && pcmf32s.size() == 2 ) + if( args.diarize ) { - const size_t n_samples = pcmf32s[ 0 ].size(); - const int64_t is0 = SourceAudio::sampleFromTimestamp( seg.time.begin, n_samples ); - const int64_t is1 = SourceAudio::sampleFromTimestamp( seg.time.end, n_samples ); - - double energy0 = 0.0f; - double energy1 = 0.0f; - - for( int64_t j = is0; j < is1; j++ ) + speaker = sender.detectSpeaker( seg.time ) switch { - energy0 += fabs( pcmf32s[ 0 ][ j ] ); - energy1 += fabs( pcmf32s[ 1 ][ j ] ); - } - - if( energy0 > 1.1 * energy1 ) - speaker = "(speaker 0)"; - else if( energy1 > 1.1 * energy0 ) - speaker = "(speaker 1)"; - else - speaker = "(speaker ?)"; - - //printf("is0 = %lld, is1 = %lld, energy0 = %f, energy1 = %f, %s\n", is0, is1, energy0, energy1, speaker.c_str()); + eSpeakerChannel.Unsure => "(speaker ?)", + eSpeakerChannel.Left => "(speaker 0)", + eSpeakerChannel.Right => "(speaker 1)", + _ => "" + }; } -#endif + if( args.print_colors && AnsiCodes.enabled ) { - Console.Write( "[{0} --> {1}] ", printTime( seg.time.begin ), printTime( seg.time.end ) ); + Console.Write( "[{0} --> {1}] {2} ", printTime( seg.time.begin ), printTime( seg.time.end ), speaker ); foreach( sToken tok in res.getTokens( seg ) ) { if( !args.print_special && tok.hasFlag( eTokenFlags.Special ) ) continue; - Console.Write( "{0}{1}{2}{3}", speaker, k_colors[ colorIndex( tok ) ], tok.text, "\x1B[0m" ); + Console.Write( "{0}{1}{2}", k_colors[ colorIndex( tok ) ], tok.text, "\x1B[0m" ); } Console.WriteLine(); } else - Console.WriteLine( "[{0} --> {1}] {2}{3}", printTime( seg.time.begin ), printTime( seg.time.end ), speaker, seg.text ); + Console.WriteLine( "[{0} --> {1}] {2} {3}", printTime( seg.time.begin ), printTime( seg.time.end ), speaker, seg.text ); } } } diff --git a/Examples/TranscribeCS/TranscribeCS.cs b/Examples/TranscribeCS/TranscribeCS.cs index 9b828e3..d94ed21 100644 --- a/Examples/TranscribeCS/TranscribeCS.cs +++ b/Examples/TranscribeCS/TranscribeCS.cs @@ -32,12 +32,12 @@ namespace TranscribeCS { if( streamAudio ) { - using iAudioReader reader = mf.openAudioFile( audioFile ); + using iAudioReader reader = mf.openAudioFile( audioFile, cla.diarize ); context.runFull( reader, transcribe, null, cla.prompt ); } else { - using iAudioBuffer buffer = mf.loadAudioFile( audioFile ); + using iAudioBuffer buffer = mf.loadAudioFile( audioFile, cla.diarize ); context.runFull( buffer, transcribe, cla.prompt ); } // When asked to, produce these text files |