FilesExpand file tree

 master

/

speech_async_grpc.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

115 lines (93 loc) · 4.13 KB

 master

/

speech_async_grpc.py

Copy path

Top

File metadata and controls

• Code
• Blame

115 lines (93 loc) · 4.13 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

83

84

85

86

87

88

89

90

91

92

93

94

95

96

97

98

99

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

#!/usr/bin/env python

# Copyright (C) 2016 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.

"""Sample that transcribes a FLAC audio file stored in Google Cloud Storage,

using async GRPC."""

import argparse

import time

from gcloud.credentials import get_credentials

from google.cloud.speech.v1beta1 import cloud_speech_pb2

from google.longrunning import operations_grpc_pb2

from grpc.beta import implementations

# Keep the request alive for this many seconds

DEADLINE_SECS = 10

SPEECH_SCOPE = 'https://www.googleapis.com/auth/cloud-platform'

def make_channel(host, port):

"""Creates an SSL channel with auth credentials from the environment."""

# In order to make an https call, use an ssl channel with defaults

ssl_channel = implementations.ssl_channel_credentials(None, None, None)

# Grab application default credentials from the environment

creds = get_credentials().create_scoped([SPEECH_SCOPE])

# Add a plugin to inject the creds into the header

auth_header = (

'Authorization',

'Bearer ' + creds.get_access_token().access_token)

auth_plugin = implementations.metadata_call_credentials(

lambda _, cb: cb([auth_header], None),

name='google_creds')

# compose the two together for both ssl and google auth

composite_channel = implementations.composite_channel_credentials(

ssl_channel, auth_plugin)

return implementations.secure_channel(host, port, composite_channel)

def main(input_uri, encoding, sample_rate):

channel = make_channel('speech.googleapis.com', 443)

service = cloud_speech_pb2.beta_create_Speech_stub(channel)

# The method and parameters can be inferred from the proto from which the

# grpc client lib was generated. See:

# https://github.com/googleapis/googleapis/blob/master/google/cloud/speech/v1beta1/cloud_speech.proto

response = service.AsyncRecognize(cloud_speech_pb2.AsyncRecognizeRequest(

config=cloud_speech_pb2.RecognitionConfig(

encoding=encoding,

sample_rate=sample_rate,

),

audio=cloud_speech_pb2.RecognitionAudio(

uri=input_uri,

)

), DEADLINE_SECS)

# Print the longrunning operation handle.

print(response)

# Construct a long running operation endpoint.

service = operations_grpc_pb2.beta_create_Operations_stub(channel)

name = response.name

while True:

# Give the server a few seconds to process.

print('Waiting for server processing...')

time.sleep(1)

# Get the long running operation with response.

response = service.GetOperation(

operations_grpc_pb2.GetOperationRequest(name=name),

DEADLINE_SECS)

if response.done:

break

# Print the recognition results.

results = cloud_speech_pb2.AsyncRecognizeResponse()

response.response.Unpack(results)

print(results)

def _gcs_uri(text):

if not text.startswith('gs://'):

raise argparse.ArgumentTypeError(

'Cloud Storage uri must be of the form gs://bucket/path/')

return text

if __name__ == '__main__':

parser = argparse.ArgumentParser()

parser.add_argument('input_uri', type=_gcs_uri)

parser.add_argument(

'--encoding', default='FLAC', choices=[

'LINEAR16', 'FLAC', 'MULAW', 'AMR', 'AMR_WB'],

help='How the audio file is encoded. See {}#L67'.format(

'https://github.com/googleapis/googleapis/blob/master/'

'google/cloud/speech/v1beta1/cloud_speech.proto'))

parser.add_argument('--sample_rate', default=16000)

args = parser.parse_args()

main(args.input_uri, args.encoding, args.sample_rate)