Combine Keyterms Prompting with Prompts

1
import requests
2
import time
3
4
base_url = "https://api.assemblyai.com"
5
headers = {"authorization": "<YOUR_API_KEY>"}
6
7
# Define your base prompt and keyterms
8
base_prompt = "This is a YouTube video describing common sports injuries."
9
keyterms = ["Sprained ankle", "ACL tear", "Hamstring strain", "Rotator cuff injury", "Tennis elbow", "Shin splints", "Concussion", "Groin pull", "Achilles tendonitis", "Meniscus tear"]
10
11
# Append context with keyterms to the prompt
12
prompt_with_context = f"{base_prompt}\n\nContext: {', '.join(keyterms)}"
13
14
data = {
15
    "audio_url": "https://assembly.ai/sports_injuries.mp3",
16
    "speech_models": ["universal-3-pro"],
17
    "language_detection": True,
18
    "prompt": prompt_with_context
19
}
20
21
response = requests.post(base_url + "/v2/transcript", headers=headers, json=data)
22
23
if response.status_code != 200:
24
    print(f"Error: {response.status_code}, Response: {response.text}")
25
    response.raise_for_status()
26
27
transcript_response = response.json()
28
transcript_id = transcript_response["id"]
29
polling_endpoint = f"{base_url}/v2/transcript/{transcript_id}"
30
31
while True:
32
    transcript = requests.get(polling_endpoint, headers=headers).json()
33
    if transcript["status"] == "completed":
34
        print(transcript["text"])
35
        break
36
    elif transcript["status"] == "error":
37
        raise RuntimeError(f"Transcription failed: {transcript['error']}")
38
    else:
39
        time.sleep(3)