Shopify Product Showcase Maker

 1import requests
 2import time
 3import json
 4
 5api_key = "YOUR_API_KEY"
 6url = "https://api.segmind.com/workflows/6787c8aa6426baad7e5391b4-v3"
 7
 8data = {
 9    "product_image": "https://segmind-inference-inputs.s3.amazonaws.com/36b96fc5-ddb3-45d3-95ea-19f00cea637a-319561 (1).jpeg",
10    "text_prompt": "running shoes, studio photo shots",
11    "width": "1024",
12    "height": "1024"
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)