Custom Spelling

1
import requests
2
import time
3
4
base_url = "https://api.assemblyai.com"
5
6
headers = {
7
    "authorization": "<YOUR_API_KEY>"
8
}
9
10
with open("./my-audio.mp3", "rb") as f:
11
  response = requests.post(base_url + "/v2/upload",
12
                          headers=headers,
13
                          data=f)
14
15
upload_url = response.json()["upload_url"]
16
17
data = {
18
    "audio_url": upload_url, # You can also use a URL to an audio or video file on the web
19
    "speech_models": ["universal-3-pro", "universal-2"],
20
    "language_detection": True,
21
    "custom_spelling": [
22
      {
23
        "from": ["Decarlo"],
24
        "to": "DeCarlo"
25
      },
26
      {
27
        "from": ["SQL"],
28
        "to": "Sequel"
29
      }
30
    ]
31
}
32
33
url = base_url + "/v2/transcript"
34
response = requests.post(url, json=data, headers=headers)
35
36
transcript_id = response.json()['id']
37
polling_endpoint = base_url + "/v2/transcript/" + transcript_id
38
39
while True:
40
  transcription_result = requests.get(polling_endpoint, headers=headers).json()
41
42
  if transcription_result['status'] == 'completed':
43
    print(f"Transcript ID:", transcript_id)
44
    break
45
46
  elif transcription_result['status'] == 'error':
47
    raise RuntimeError(f"Transcription failed: {transcription_result['error']}")
48
49
  else:
50
    time.sleep(3)