Thumbnail Extractor from Video Serverless API

 1import requests
 2import time
 3import json
 4
 5api_key = "YOUR_API_KEY"
 6url = "https://api.segmind.com/workflows/v2/6a7051e630c6484376739c61-v2"
 7
 8data = {
 9    "input_prompt": "Find the best thumbnail from the video",
10    "input_video": "https://images.segmind.com/uploads/a6baef49-7280-4b43-99c8-36f24706ad9a/0eed3e0d-9d1e-4368-8d4b-b51ca9a96f6f-f534f-output__1_.mp4",
11    "thumbnail_style_reference": "https://images.segmind.com/uploads/a6baef49-7280-4b43-99c8-36f24706ad9a/b4dc915b-1b0e-426a-a750-8f771fc35ee3-38299ac47bbf3a8f77dd45f62e16f5a2.jpg"
12}
13
14def poll_for_result(poll_url):
15    """Poll the API until the generation is complete"""
16    while True:
17        response = requests.get(
18            poll_url,
19            headers={'Authorization': f'Bearer {api_key}'}
20        )
21        result = response.json()
22
23        if result['status'] == 'COMPLETED':
24            # Parse the output (it's a JSON string)
25            outputs = json.loads(result['output'])
26            return outputs
27        elif result['status'] == 'FAILED':
28            raise Exception(result.get('error', 'Generation failed'))
29
30        # Wait 7 seconds before polling again
31        time.sleep(7)
32
33# Make the initial request
34response = requests.post(
35    url,
36    json=data,
37    headers={
38        'Authorization': f'Bearer {api_key}',
39        'Content-Type': 'application/json'
40    }
41)
42
43if response.status_code == 200:
44    result = response.json()
45    print('Request queued:', result)
46
47    # Start polling for results
48    outputs = poll_for_result(result['poll_url'])
49    print('Generation complete:', outputs)
50else:
51    print(f"Error: {response.status_code}")
52    print(response.text)