Professional Fashion Headshots - Powered By Nano Banana

 1import requests
 2import time
 3import json
 4
 5api_key = "YOUR_API_KEY"
 6url = "https://api.segmind.com/workflows/68c6febe4af49bc4856ade70-v2"
 7
 8data = {
 9    "3Input_images__": [
10        "https://segmind-inference-inputs.s3.amazonaws.com/17ff33a8-ba90-48e3-b942-1fade0b0c42f-behrouz-sasani-FiRVZpjlX2Y-unsplash.jpg",
11        "https://segmind-inference-inputs.s3.amazonaws.com/a0dcee14-474d-4da8-a27b-791ae29b58b9-behrouz-sasani-kJKXbtnQXVY-unsplash.jpg",
12        "https://segmind-inference-inputs.s3.amazonaws.com/219f020d-0aa8-4811-9857-764f93635597-behrouz-sasani-wjIDQ71S5KU-unsplash.jpg"
13    ]
14}
15
16def poll_for_result(poll_url):
17    """Poll the API until the generation is complete"""
18    while True:
19        response = requests.get(
20            poll_url,
21            headers={'Authorization': f'Bearer {api_key}'}
22        )
23        result = response.json()
24
25        if result['status'] == 'COMPLETED':
26            # Parse the output (it's a JSON string)
27            outputs = json.loads(result['output'])
28            return outputs
29        elif result['status'] == 'FAILED':
30            raise Exception(result.get('error', 'Generation failed'))
31
32        # Wait 7 seconds before polling again
33        time.sleep(7)
34
35# Make the initial request
36response = requests.post(
37    url,
38    json=data,
39    headers={
40        'Authorization': f'Bearer {api_key}',
41        'Content-Type': 'application/json'
42    }
43)
44
45if response.status_code == 200:
46    result = response.json()
47    print('Request queued:', result)
48
49    # Start polling for results
50    outputs = poll_for_result(result['poll_url'])
51    print('Generation complete:', outputs)
52else:
53    print(f"Error: {response.status_code}")
54    print(response.text)