FazBrowse GitHub Viewer | Trending |
URL:
| Home
Tools: [Download Repo ZIP]   [Original HTTPS Page]

Adds streaming example and tests. · lkhore/java-docs-samples@61e91da · GitHub

/ java-docs-samples Public
forked from GoogleCloudPlatform/java-docs-samples

Commit 61e91da

Browse files
gguuss
committed
Adds streaming example and tests.
1 parent 6fe4238 commit 61e91da

2 files changed

Lines changed: 101 additions & 6 deletions

File tree

‎speech/cloud-client/src/main/java/com/example/speech/Recognize.java‎

Lines changed: 91 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,9 @@
1616

1717
package com.example.speech;
1818

19+
import com.google.api.gax.core.ApiStreamObserver;
1920
import com.google.api.gax.grpc.OperationFuture;
21+
import com.google.api.gax.grpc.StreamingCallable;
2022
import com.google.cloud.speech.spi.v1.SpeechClient;
2123
import com.google.cloud.speech.v1.LongRunningRecognizeResponse;
2224
import com.google.cloud.speech.v1.RecognitionAudio;
@@ -25,6 +27,11 @@
2527
import com.google.cloud.speech.v1.RecognizeResponse;
2628
import com.google.cloud.speech.v1.SpeechRecognitionAlternative;
2729
import com.google.cloud.speech.v1.SpeechRecognitionResult;
30+
import com.google.cloud.speech.v1.StreamingRecognitionConfig;
31+
import com.google.cloud.speech.v1.StreamingRecognitionResult;
32+
import com.google.cloud.speech.v1.StreamingRecognizeRequest;
33+
import com.google.cloud.speech.v1.StreamingRecognizeResponse;
34+
import com.google.common.util.concurrent.SettableFuture;
2835
import com.google.protobuf.ByteString;
2936

3037
import java.io.IOException;
@@ -40,7 +47,7 @@ public static void main(String... args) throws Exception {
4047
System.out.printf(
4148
"\tjava %s \"<command>\" \"<path-to-image>\"\n"
4249
+ "Commands:\n"
43-
+ "\tsyncrecognize | asyncrecognize\n"
50+
+ "\tsyncrecognize | asyncrecognize | streamrecognize\n"
4451
+ "Path:\n\tA file path (ex: ./resources/audio.raw) or a URI "
4552
+ "for a Cloud Storage resource (gs://...)\n",
4653
Recognize.class.getCanonicalName());
@@ -62,7 +69,11 @@ public static void main(String... args) throws Exception {
6269
} else {
6370
asyncRecognizeFile(path);
6471
}
72+
} else if (command.equals("streamrecognize")) {
73+
streamingRecognizeFile(path);
74+
//streamingRecognizeEasy(path);
6575
}
76+
6677
}
6778

6879
/**
@@ -180,13 +191,13 @@ public static void asyncRecognizeFile(String fileName) throws Exception, IOExcep
180191
* Performs non-blocking speech recognition on remote FLAC file and prints
181192
* the transcription.
182193
*
183-
* @param gcsUri the path to the remote FLAC audio file to transcribe.
194+
* @param gcsUri the path to the remote LINEAR16 audio file to transcribe.
184195
*/
185196
public static void asyncRecognizeGcs(String gcsUri) throws Exception, IOException {
186197
// Instantiates a client with GOOGLE_APPLICATION_CREDENTIALS
187198
SpeechClient speech = SpeechClient.create();
188199

189-
// Configure remote file request for FLAC file
200+
// Configure remote file request for Linear16
190201
RecognitionConfig config = RecognitionConfig.newBuilder()
191202
.setEncoding(AudioEncoding.FLAC)
192203
.setLanguageCode("en-US")
@@ -214,4 +225,81 @@ public static void asyncRecognizeGcs(String gcsUri) throws Exception, IOExceptio
214225
}
215226
speech.close();
216227
}
228+
229+
public static void streamingRecognizeFile(String fileName) throws Exception, IOException {
230+
Path path = Paths.get(fileName);
231+
byte[] data = Files.readAllBytes(path);
232+
233+
// Instantiates a client with GOOGLE_APPLICATION_CREDENTIALS
234+
SpeechClient speech = SpeechClient.create();
235+
236+
// Configure request with local raw PCM audio
237+
RecognitionConfig recConfig = RecognitionConfig.newBuilder()
238+
.setEncoding(AudioEncoding.LINEAR16)
239+
.setLanguageCode("en-US")
240+
.setSampleRateHertz(16000)
241+
.build();
242+
StreamingRecognitionConfig config = StreamingRecognitionConfig.newBuilder()
243+
.setConfig(recConfig)
244+
.build();
245+
246+
class ResponseApiStreamingObserver<T> implements ApiStreamObserver<T> {
247+
private final SettableFuture<List<T>> future = SettableFuture.create();
248+
private final List<T> messages = new java.util.ArrayList<T>();
249+
250+
@Override
251+
public void onNext(T message) {
252+
messages.add(message);
253+
}
254+
255+
@Override
256+
public void onError(Throwable t) {
257+
future.setException(t);
258+
}
259+
260+
@Override
261+
public void onCompleted() {
262+
future.set(messages);
263+
}
264+
265+
// Returns the SettableFuture object to get received messages / exceptions.
266+
public SettableFuture<List<T>> future() {
267+
return future;
268+
}
269+
}
270+
271+
ResponseApiStreamingObserver<StreamingRecognizeResponse> responseObserver =
272+
new ResponseApiStreamingObserver<StreamingRecognizeResponse>();
273+
274+
StreamingCallable<StreamingRecognizeRequest,StreamingRecognizeResponse> callable =
275+
speech.streamingRecognizeCallable();
276+
277+
ApiStreamObserver<StreamingRecognizeRequest> requestObserver =
278+
callable.bidiStreamingCall(responseObserver);
279+
280+
// The first request must **only** contain the audio configuration:
281+
requestObserver.onNext(StreamingRecognizeRequest.newBuilder()
282+
.setStreamingConfig(config)
283+
.build());
284+
285+
// Subsequent requests must **only** contain the audio data.
286+
requestObserver.onNext(StreamingRecognizeRequest.newBuilder()
287+
.setAudioContent(ByteString.copyFrom(data))
288+
.build());
289+
290+
// Mark transmission as completed after sending the data.
291+
requestObserver.onCompleted();
292+
293+
List<StreamingRecognizeResponse> responses = responseObserver.future().get();
294+
295+
for (StreamingRecognizeResponse response: responses) {
296+
for (StreamingRecognitionResult result: response.getResultsList()) {
297+
for (SpeechRecognitionAlternative alternative : result.getAlternativesList()) {
298+
System.out.println(alternative.getTranscript());
299+
}
300+
}
301+
}
302+
speech.close();
303+
}
304+
217305
}

‎speech/cloud-client/src/test/java/com/example/speech/RecognizeIT.java‎

Lines changed: 10 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -61,25 +61,32 @@ public void testRecognizeFile() throws Exception {
6161
String got = bout.toString();
6262
assertThat(got).contains("how old is the Brooklyn Bridge");
6363
}
64-
64+
6565
@Test
6666
public void testRecognizeGcs() throws Exception {
6767
Recognize.syncRecognizeGcs(gcsPath);
6868
String got = bout.toString();
6969
assertThat(got).contains("how old is the Brooklyn Bridge");
7070
}
71-
71+
7272
@Test
7373
public void testAsyncRecognizeFile() throws Exception {
7474
Recognize.asyncRecognizeFile(fileName);
7575
String got = bout.toString();
7676
assertThat(got).contains("how old is the Brooklyn Bridge");
7777
}
78-
78+
7979
@Test
8080
public void testAsyncRecognizeGcs() throws Exception {
8181
Recognize.asyncRecognizeGcs(gcsPath);
8282
String got = bout.toString();
8383
assertThat(got).contains("how old is the Brooklyn Bridge");
8484
}
85+
86+
@Test
87+
public void testStreamRecognize() throws Exception {
88+
Recognize.streamingRecognizeFile(fileName);
89+
String got = bout.toString();
90+
assertThat(got).contains("how old is the Brooklyn Bridge");
91+
}
8592
}

0 commit comments

Comments
 (0)
Back | FazBrowse Home | New Git URL