Caption Engine Serverless API

 1import requests
 2import time
 3import json
 4
 5api_key = "YOUR_API_KEY"
 6url = "https://api.segmind.com/workflows/v2/6a7051e330c6484376739c5f-v1"
 7
 8data = {
 9    "input_video": "https://images.segmind.com/uploads/a6baef49-7280-4b43-99c8-36f24706ad9a/3f8a513d-a92e-4096-84c4-7b2f4e4fd994-Sequence_01_4.mp4",
10    "input_prompt": "Font Style - Roboto font\n\nFont size - medium size\n\nPosition - center-aligned in the bottom\n\nAnimation - subtle animation highlighting the words in blue color when it speaks with perfect sync to the man speaking"
11}
12
13def poll_for_result(poll_url):
14    """Poll the API until the generation is complete"""
15    while True:
16        response = requests.get(
17            poll_url,
18            headers={'Authorization': f'Bearer {api_key}'}
19        )
20        result = response.json()
21
22        if result['status'] == 'COMPLETED':
23            # Parse the output (it's a JSON string)
24            outputs = json.loads(result['output'])
25            return outputs
26        elif result['status'] == 'FAILED':
27            raise Exception(result.get('error', 'Generation failed'))
28
29        # Wait 7 seconds before polling again
30        time.sleep(7)
31
32# Make the initial request
33response = requests.post(
34    url,
35    json=data,
36    headers={
37        'Authorization': f'Bearer {api_key}',
38        'Content-Type': 'application/json'
39    }
40)
41
42if response.status_code == 200:
43    result = response.json()
44    print('Request queued:', result)
45
46    # Start polling for results
47    outputs = poll_for_result(result['poll_url'])
48    print('Generation complete:', outputs)
49else:
50    print(f"Error: {response.status_code}")
51    print(response.text)