AI Infographic Generator

 1import requests
 2import time
 3import json
 4
 5api_key = "YOUR_API_KEY"
 6url = "https://api.segmind.com/workflows/6839bf53659263e69c7a567a-v1"
 7
 8data = {
 9    "Text_Prompt": "[TITLE] = 'HOW TO ASK FOR A FRENCHIE PIXIE SHAG'\n[FEATURE 1] = 'Short, textured cut'\n[FEATURE 2] = 'Blending a pixie and shag '\n[FEATURE 3] = 'Soft curtain bangs'\n[INSTRUCTION 1]  ='Blend a short pixie with a soft, French-inspired shag'\n[INSTRUCTION 2]  ='Create natural lift and texture at the crown'\n[INSTRUCTION 3]  ='Cut soft curtain fringe that parts and stays out of the eyes'\n[TIP 1] = 'Razor cutting and point cutting can help achieve the airy, textured look'\n[STYLE] = 'professional'\n"
10}
11
12def poll_for_result(poll_url):
13    """Poll the API until the generation is complete"""
14    while True:
15        response = requests.get(
16            poll_url,
17            headers={'Authorization': f'Bearer {api_key}'}
18        )
19        result = response.json()
20
21        if result['status'] == 'COMPLETED':
22            # Parse the output (it's a JSON string)
23            outputs = json.loads(result['output'])
24            return outputs
25        elif result['status'] == 'FAILED':
26            raise Exception(result.get('error', 'Generation failed'))
27
28        # Wait 7 seconds before polling again
29        time.sleep(7)
30
31# Make the initial request
32response = requests.post(
33    url,
34    json=data,
35    headers={
36        'Authorization': f'Bearer {api_key}',
37        'Content-Type': 'application/json'
38    }
39)
40
41if response.status_code == 200:
42    result = response.json()
43    print('Request queued:', result)
44
45    # Start polling for results
46    outputs = poll_for_result(result['poll_url'])
47    print('Generation complete:', outputs)
48else:
49    print(f"Error: {response.status_code}")
50    print(response.text)