Creating summarized chapters from podcasts

1
import requests
2
import time

1
base_url = "https://api.assemblyai.com"
2
3
headers = {
4
    "authorization": "<YOUR_API_KEY>"
5
}

1
with open("./my-audio.mp3", "rb") as f:
2
  response = requests.post(base_url + "/v2/upload",
3
                          headers=headers,
4
                          data=f)
5
6
upload_url = response.json()["upload_url"]
7
8
data = {
9
    "audio_url": upload_url,
10
    "speech_models": ["universal-3-pro", "universal-2"],
11
    "language_detection": True
12
}
13
14
response = requests.post(base_url + "/v2/transcript", json=data, headers=headers)
15
16
transcript_id = response.json()['id']
17
polling_endpoint = base_url + "/v2/transcript/" + transcript_id
18
19
while True:
20
    transcription_result = requests.get(polling_endpoint, headers=headers).json()
21
    if transcription_result['status'] == 'completed':
22
        break
23
    elif transcription_result['status'] == 'error':
24
        raise RuntimeError(f"Transcription failed: {transcription_result['error']}")
25
    else:
26
        time.sleep(3)

1
paragraphs = requests.get(polling_endpoint + '/paragraphs', headers=headers).json()['paragraphs']
2
3
# Group paragraphs into chapters
4
step = 2  # Adjust to control chapter length
5
combined_paragraphs = []
6
7
for i in range(0, len(paragraphs), step):
8
    paragraph_group = paragraphs[i : i + step]
9
    start = paragraph_group[0]['start']
10
    end = paragraph_group[-1]['end']
11
    text = " ".join(p['text'] for p in paragraph_group)
12
    combined_paragraphs.append({"text": text, "start": start, "end": end})

1
for chapter in combined_paragraphs:
2
    llm_gateway_data = {
3
        "model": "claude-sonnet-4-6",
4
        "messages": [
5
            {"role": "user", "content": f"Provide a brief one-paragraph summary, a one-line gist, and a headline for this section of a transcript.\n\nText: {chapter['text']}"}
6
        ],
7
        "max_tokens": 500
8
    }
9
10
    response = requests.post(
11
        "https://llm-gateway.assemblyai.com/v1/chat/completions",
12
        headers=headers,
13
        json=llm_gateway_data
14
    )
15
16
    result = response.json()["choices"][0]["message"]["content"]
17
    print(f"Chapter Start Time: {chapter['start']}")
18
    print(f"Chapter End Time: {chapter['end']}")
19
    print(f"Chapter Summary: {result}\n")