From 12bea7883a984e039de2253728f7d2a49bdaa9a4 Mon Sep 17 00:00:00 2001
From: Yu-Han Liu <dizcology@hotmail.com>
Date: Fri, 22 Feb 2019 07:52:31 -0800
Subject: [PATCH] add speech api multichannel samples
[(#2003)](https://github.com/GoogleCloudPlatform/python-docs-samples/issues/2003)
* add speech api multichannel samples
* udpate copyright year
* test with multichannel audio data
* flake
* update comment
---
.../snippets/transcribe_enhanced_model.py | 7 +-
.../snippets/transcribe_multichannel.py | 95 +++++++++++++++++++
.../snippets/transcribe_multichannel_test.py | 36 +++++++
3 files changed, 134 insertions(+), 4 deletions(-)
create mode 100644 packages/google-cloud-python-speech/samples/snippets/transcribe_multichannel.py
create mode 100644 packages/google-cloud-python-speech/samples/snippets/transcribe_multichannel_test.py
diff --git a/packages/google-cloud-python-speech/samples/snippets/transcribe_enhanced_model.py b/packages/google-cloud-python-speech/samples/snippets/transcribe_enhanced_model.py
index 514d54812d03..1b233c52696c 100644
--- a/packages/google-cloud-python-speech/samples/snippets/transcribe_enhanced_model.py
+++ b/packages/google-cloud-python-speech/samples/snippets/transcribe_enhanced_model.py
@@ -22,16 +22,15 @@
"""
import argparse
-import io
def transcribe_file_with_enhanced_model(path):
"""Transcribe the given audio file using an enhanced model."""
# [START speech_transcribe_enhanced_model]
- import io
-
+ import io
+
from google.cloud import speech
-
+
client = speech.SpeechClient()
# path = 'resources/commercial_mono.wav'
diff --git a/packages/google-cloud-python-speech/samples/snippets/transcribe_multichannel.py b/packages/google-cloud-python-speech/samples/snippets/transcribe_multichannel.py
new file mode 100644
index 000000000000..e84da59ad7b3
--- /dev/null
+++ b/packages/google-cloud-python-speech/samples/snippets/transcribe_multichannel.py
@@ -0,0 +1,95 @@
+#!/usr/bin/env python
+
+# Copyright 2019 Google Inc. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""Google Cloud Speech API sample that demonstrates multichannel recognition.
+
+Example usage:
+ python transcribe_multichannel.py resources/multi.wav
+ python transcribe_multichannel.py \
+ gs://cloud-samples-tests/speech/multi.wav
+"""
+
+import argparse
+
+
+def transcribe_file_with_multichannel(speech_file):
+ """Transcribe the given audio file synchronously with
+ multi channel."""
+ # [START speech_transcribe_multichannel]
+ from google.cloud import speech
+ client = speech.SpeechClient()
+
+ with open(speech_file, 'rb') as audio_file:
+ content = audio_file.read()
+
+ audio = speech.types.RecognitionAudio(content=content)
+
+ config = speech.types.RecognitionConfig(
+ encoding=speech.enums.RecognitionConfig.AudioEncoding.LINEAR16,
+ sample_rate_hertz=44100,
+ language_code='en-US',
+ audio_channel_count=2,
+ enable_separate_recognition_per_channel=True)
+
+ response = client.recognize(config, audio)
+
+ for i, result in enumerate(response.results):
+ alternative = result.alternatives[0]
+ print('-' * 20)
+ print('First alternative of result {}'.format(i))
+ print(u'Transcript: {}'.format(alternative.transcript))
+ print(u'Channel Tag: {}'.format(result.channel_tag))
+ # [END speech_transcribe_multichannel]
+
+
+def transcribe_gcs_with_multichannel(gcs_uri):
+ """Transcribe the given audio file on GCS with
+ multi channel."""
+ # [START speech_transcribe_multichannel_gcs]
+ from google.cloud import speech
+ client = speech.SpeechClient()
+
+ audio = speech.types.RecognitionAudio(uri=gcs_uri)
+
+ config = speech.types.RecognitionConfig(
+ encoding=speech.enums.RecognitionConfig.AudioEncoding.LINEAR16,
+ sample_rate_hertz=44100,
+ language_code='en-US',
+ audio_channel_count=2,
+ enable_separate_recognition_per_channel=True)
+
+ response = client.recognize(config, audio)
+
+ for i, result in enumerate(response.results):
+ alternative = result.alternatives[0]
+ print('-' * 20)
+ print('First alternative of result {}'.format(i))
+ print(u'Transcript: {}'.format(alternative.transcript))
+ print(u'Channel Tag: {}'.format(result.channel_tag))
+ # [END speech_transcribe_multichannel_gcs]
+
+
+if __name__ == '__main__':
+ parser = argparse.ArgumentParser(
+ description=__doc__,
+ formatter_class=argparse.RawDescriptionHelpFormatter)
+ parser.add_argument(
+ 'path', help='File or GCS path for audio file to be recognized')
+ args = parser.parse_args()
+ if args.path.startswith('gs://'):
+ transcribe_gcs_with_multichannel(args.path)
+ else:
+ transcribe_file_with_multichannel(args.path)
diff --git a/packages/google-cloud-python-speech/samples/snippets/transcribe_multichannel_test.py b/packages/google-cloud-python-speech/samples/snippets/transcribe_multichannel_test.py
new file mode 100644
index 000000000000..de9558629994
--- /dev/null
+++ b/packages/google-cloud-python-speech/samples/snippets/transcribe_multichannel_test.py
@@ -0,0 +1,36 @@
+# Copyright 2019, Google, Inc.
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import os
+
+from transcribe_multichannel import (
+ transcribe_file_with_multichannel,
+ transcribe_gcs_with_multichannel)
+
+RESOURCES = os.path.join(os.path.dirname(__file__), 'resources')
+
+
+def test_transcribe_multichannel_file(capsys):
+ transcribe_file_with_multichannel(
+ os.path.join(RESOURCES, 'multi.wav'))
+ out, err = capsys.readouterr()
+
+ assert 'how are you doing' in out
+
+
+def test_transcribe_multichannel_gcs(capsys):
+ transcribe_gcs_with_multichannel(
+ 'gs://cloud-samples-data/speech/multi.wav')
+ out, err = capsys.readouterr()
+
+ assert 'how are you doing' in out