Life Style Product Photo Generator

 1import requests
 2import time
 3import json
 4
 5api_key = "YOUR_API_KEY"
 6url = "https://api.segmind.com/workflows/6780bee21cd362b65eb04173-v7"
 7
 8data = {
 9    "product_image": "https://segmind-inference-inputs.s3.amazonaws.com/21c47010-78b4-490c-88c2-cb6fbced65a7-GOMBEWboQGqMWhKHovI4Sw.jpg",
10    "prompt": "woman holding a detergent bottle",
11    "product_type": "bottle",
12    "seed": "4715087"
13}
14
15def poll_for_result(poll_url):
16    """Poll the API until the generation is complete"""
17    while True:
18        response = requests.get(
19            poll_url,
20            headers={'Authorization': f'Bearer {api_key}'}
21        )
22        result = response.json()
23
24        if result['status'] == 'COMPLETED':
25            # Parse the output (it's a JSON string)
26            outputs = json.loads(result['output'])
27            return outputs
28        elif result['status'] == 'FAILED':
29            raise Exception(result.get('error', 'Generation failed'))
30
31        # Wait 7 seconds before polling again
32        time.sleep(7)
33
34# Make the initial request
35response = requests.post(
36    url,
37    json=data,
38    headers={
39        'Authorization': f'Bearer {api_key}',
40        'Content-Type': 'application/json'
41    }
42)
43
44if response.status_code == 200:
45    result = response.json()
46    print('Request queued:', result)
47
48    # Start polling for results
49    outputs = poll_for_result(result['poll_url'])
50    print('Generation complete:', outputs)
51else:
52    print(f"Error: {response.status_code}")
53    print(response.text)