Merge pull request #41 from AssemblyAI/niels/update-sample

Update sample, increase HTTP request timeout

Author: Swimburger

sample-app/src/main/java/sample/App.java

@@ -3,68 +3,83 @@
 import com.assemblyai.api.AssemblyAI;
 import com.assemblyai.api.RealtimeTranscriber;
 import com.assemblyai.api.resources.files.types.UploadedFile;
-import com.assemblyai.api.resources.transcripts.requests.TranscriptParams;
-import com.assemblyai.api.resources.transcripts.requests.WordSearchParams;
-import com.assemblyai.api.resources.transcripts.types.ParagraphsResponse;
-import com.assemblyai.api.resources.transcripts.types.SentencesResponse;
-import com.assemblyai.api.resources.transcripts.types.SubtitleFormat;
-import com.assemblyai.api.resources.transcripts.types.Transcript;
-import com.assemblyai.api.resources.transcripts.types.TranscriptList;
-import com.assemblyai.api.resources.transcripts.types.WordSearchResponse;
+import com.assemblyai.api.resources.lemur.requests.LemurTaskParams;
+import com.assemblyai.api.resources.lemur.types.LemurTaskResponse;
+import com.assemblyai.api.resources.transcripts.requests.*;
+import com.assemblyai.api.resources.transcripts.types.*;
 import java.io.File;
 import java.io.FileInputStream;
 import java.io.IOException;
 import java.nio.file.Files;
+import java.util.List;
 
 public final class App {
 
     public static void main(String... args) throws IOException {
-        AssemblyAI aai = AssemblyAI.builder()
+        AssemblyAI client = AssemblyAI.builder()
                 .apiKey(System.getenv("ASSEMBLYAI_API_KEY"))
                 .build();
 
-        Transcript transcript =
-                aai.transcripts().transcribe("https://storage.googleapis.com/aai-docs-samples/nbc.mp3");
-        System.out.println(transcript);
+        Transcript transcript = client.transcripts().transcribe(
+                "https://storage.googleapis.com/client-docs-samples/nbc.mp3",
+                TranscriptOptionalParams.builder()
+                        .sentimentAnalysis(true)
+                        .build()
+        );
+
+        for(SentimentAnalysisResult result: transcript.getSentimentAnalysisResults().get())
+        {
+            System.out.println("Text: " + result.getText());
+            System.out.println("Sentiment: " + result.getSentiment());
+            System.out.println("Confidence: " + result.getConfidence());
+            System.out.printf("Timestamp: %s - %s", result.getStart(), result.getEnd());
+        }
 
-        SentencesResponse sentences = aai.transcripts().getSentences(transcript.getId());
+        SentencesResponse sentences = client.transcripts().getSentences(transcript.getId());
         System.out.println("Get transcript sentences. " + sentences);
 
-        ParagraphsResponse paragraphs = aai.transcripts().getParagraphs(transcript.getId());
+        ParagraphsResponse paragraphs = client.transcripts().getParagraphs(transcript.getId());
         System.out.println("Get transcript paragraphs. " + paragraphs);
 
-        String srt = aai.transcripts().getSubtitles(transcript.getId(), SubtitleFormat.SRT);
+        String srt = client.transcripts().getSubtitles(transcript.getId(), SubtitleFormat.SRT);
         System.out.println("Get transcript srt. " + srt);
 
-        String vtt = aai.transcripts().getSubtitles(transcript.getId(), SubtitleFormat.VTT);
+        String vtt = client.transcripts().getSubtitles(transcript.getId(), SubtitleFormat.VTT);
         System.out.println("Get transcript vtt. " + vtt);
 
-        WordSearchResponse search = aai.transcripts().wordSearch(transcript.getId(), WordSearchParams.builder()
+        WordSearchResponse search = client.transcripts().wordSearch(transcript.getId(), WordSearchParams.builder()
                 .words("NBC")
                 .build());
         System.out.println("Search transcript. " + search);
 
-        transcript = aai.transcripts().delete(transcript.getId());
+        transcript = client.transcripts().delete(transcript.getId());
         System.out.println("Delete transcript. " + transcript);
 
         File file = new File("sample-app/src/main/resources/nZP7pb_t4oA.mp3");
-        UploadedFile uploadedFile = aai.files().upload(Files.readAllBytes(file.toPath()));
+        UploadedFile uploadedFile = client.files().upload(Files.readAllBytes(file.toPath()));
         System.out.println("Uploaded file" + uploadedFile);
 
-        transcript = aai.transcripts().submit(TranscriptParams.builder()
-                .audioUrl("https://storage.googleapis.com/aai-docs-samples/nbc.mp3")
+        transcript = client.transcripts().submit(TranscriptParams.builder()
+                .audioUrl("https://storage.googleapis.com/client-docs-samples/nbc.mp3")
                 .build());
         System.out.println("Created transcript " + transcript);
 
-        transcript = aai.transcripts().get(transcript.getId());
+        transcript = client.transcripts().get(transcript.getId());
         System.out.println("Got transcript. " + transcript);
 
-        TranscriptList transcripts = aai.transcripts().list();
+        TranscriptList transcripts = client.transcripts().list();
         System.out.println("List transcript. " + transcripts);
 
+        LemurTaskResponse response = client.lemur().task(LemurTaskParams.builder()
+                .prompt("Summarize this transcript.")
+                .transcriptIds(List.of(transcript.getId()))
+                .build());
+
+        System.out.println("Summary: " + response.getResponse());
+
         RealtimeTranscriber realtimeTranscriber = RealtimeTranscriber.builder()
+                .apiKey(System.getenv("ASSEMBLYAI_API_KEY"))
                 .onSessionStart(System.out::println)
-                .apiKey(System.getenv("ASSEMBLY_AI_API_KEY"))
                 .onPartialTranscript(System.out::println)
                 .onFinalTranscript(System.out::println)
                 .onError((err) -> System.out.println(err.getMessage()))
@@ -81,8 +96,9 @@ public static void streamFile(String filePath, RealtimeTranscriber realtimeTrans
 
             while (fileInputStream.read(buffer) != -1) {
                 realtimeTranscriber.sendAudio(buffer);
+                Thread.sleep(300);
             }
-        } catch (IOException e) {
+        } catch (IOException | InterruptedException e) {
             e.printStackTrace();
         }
     }

src/main/java/com/assemblyai/api/core/ClientOptions.java

@@ -5,6 +5,7 @@
 
 import java.util.HashMap;
 import java.util.Map;
+import java.util.concurrent.TimeUnit;
 import java.util.function.Supplier;
 import okhttp3.OkHttpClient;
 
@@ -34,7 +35,6 @@ private ClientOptions(
                 "JAVA"));
         this.headerSuppliers = headerSuppliers;
         this.httpClient = httpClient;
-        ;
     }
 
     public Environment environment() {
@@ -85,6 +85,8 @@ public Builder addHeader(String key, Supplier<String> value) {
         public ClientOptions build() {
             OkHttpClient okhttpClient = new OkHttpClient.Builder()
                     .addInterceptor(new RetryInterceptor(3))
+                    // set longer timeout for LeMUR
+                    .connectTimeout(30, TimeUnit.SECONDS)
                     .build();
             return new ClientOptions(environment, headers, headerSuppliers, okhttpClient);
         }