Export SRT or VTT caption files

1
import assemblyai as aai
2
3
aai.settings.api_key = "<YOUR_API_KEY>"
4
5
# audio_file = "./local_file.mp3"
6
audio_file = "https://assembly.ai/wildfires.mp3"
7
8
config = aai.TranscriptionConfig()
9
10
transcript = aai.Transcriber(config=config).transcribe(audio_file)
11
12
if transcript.status == "error":
13
  raise RuntimeError(f"Transcription failed: {transcript.error}")
14
15
srt = transcript.export_subtitles_srt(
16
  # Optional: Customize the maximum number of characters per caption
17
  chars_per_caption=32
18
  )
19
20
with open(f"transcript_{transcript.id}.srt", "w") as srt_file:
21
  srt_file.write(srt)
22
23
# vtt = transcript.export_subtitles_vtt()
24
25
# with open(f"transcript_{transcript_id}.vtt", "w") as vtt_file:
26
#   vtt_file.write(vtt)