AI Youtube Thumbnail Generator

 1import requests
 2import time
 3import json
 4
 5api_key = "YOUR_API_KEY"
 6url = "https://api.segmind.com/workflows/682a1dbc00eb0e1658c0b687-v1"
 7
 8data = {
 9    "ref_image": "https://segmind-inference-inputs.s3.amazonaws.com/3c962cf5-de7b-4714-a5b8-177011158e94-istockphoto-1688741946-612x612.jpg",
10    "prompt": "Create a vibrant and eye-catching realistic YouTube thumbnail titled ‘Who Benches More?’ Feature two people on opposite sides of a gym bench: one wearing (a man ) white glasses with a sad expression (struggling to lift a small weight), and the other wearing (a woman ) black glasses with a confident, happy smile (lifting a massive weight). \nAdd bold, playful text like ‘Gym Showdown!!!’ or  ‘White Glasses vs Black Glasses. Use bright colors, dynamic poses, and include gym equipment in the background for context. \nEnsure the design is bold and contrasts well to grab attention."
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)