-
Notifications
You must be signed in to change notification settings - Fork 26
/
main.go
84 lines (69 loc) · 1.73 KB
/
main.go
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
package main
import (
"bufio"
"fmt"
"log"
"net/http"
"reflect"
"time"
"github.com/Jeffail/gabs/v2"
"github.com/deepgram-devs/deepgram-go-sdk/deepgram"
"github.com/gorilla/websocket"
)
const (
DEEPGRAM_API_KEY = "DEEPGRAM_API_KEY"
STREAM_URL = "http://stream.live.vc.bbcmedia.co.uk/bbc_world_service"
CHUNK_SIZE = 1024 * 2
TEN_MILLISECONDS_SLEEP = 10 * time.Millisecond
)
func main() {
client := new(http.Client)
dg := *deepgram.NewClient(DEEPGRAM_API_KEY)
res, err := client.Get(STREAM_URL)
if err != nil {
log.Println("ERROR getting stream", err)
return
}
defer res.Body.Close()
fmt.Println("Stream is up and running ", reflect.TypeOf(res))
reader := bufio.NewReader(res.Body)
liveTranscriptionOptions := deepgram.LiveTranscriptionOptions{
Language: "en-US",
Punctuate: true,
}
dgConn, _, err := dg.LiveTranscription(liveTranscriptionOptions)
if err != nil {
log.Println("ERROR creating LiveTranscription connection:", err)
return
}
defer dgConn.Close()
chunk := make([]byte, CHUNK_SIZE)
go func() {
for {
_, message, err := dgConn.ReadMessage()
if err != nil {
log.Println("ERROR reading message:", err)
return
}
jsonParsed, jsonErr := gabs.ParseJSON(message)
if jsonErr != nil {
log.Println("ERROR parsing JSON message:", err)
return
}
log.Printf("recv: %s", jsonParsed.Path("channel.alternatives.0.transcript").String())
}
}()
for {
bytesRead, err := reader.Read(chunk)
if err != nil {
log.Println("ERROR reading chunk:", err)
return
}
err = dgConn.WriteMessage(websocket.BinaryMessage, chunk[:bytesRead])
if err != nil {
log.Println("ERROR writing message:", err)
return
}
time.Sleep(TEN_MILLISECONDS_SLEEP)
}
}