AI-Enhanced Multi-Image Composition with Pruna P Image Edit

 1import requests
 2import time
 3import json
 4
 5api_key = "YOUR_API_KEY"
 6url = "https://api.segmind.com/workflows/692fdb953fe28df5bd277132-v1"
 7
 8data = {
 9    "Input_Imager": [
10        "https://segmind-inference-inputs.s3.amazonaws.com/9e021506-4a79-441b-bc48-86075eea4d10-d61283c9-839e-4107-9185-4ea883ecaded.webp",
11        "https://segmind-inference-inputs.s3.amazonaws.com/312ad02e-98ab-413a-8116-97c3e2cf7868-IMG_7136.webp",
12        "https://segmind-inference-inputs.s3.amazonaws.com/73a44879-1006-436d-adf3-ceebbdab5bd9-IMG_7135.webp",
13        "https://segmind-inference-inputs.s3.amazonaws.com/05d223ba-29f5-4f86-b185-5ea08e535141-IMG_7137.webp",
14        "https://segmind-inference-inputs.s3.amazonaws.com/10e3d3f4-9096-4bf9-8266-e19e4c95b3a5-IMG_7138.webp"
15    ]
16}
17
18def poll_for_result(poll_url):
19    """Poll the API until the generation is complete"""
20    while True:
21        response = requests.get(
22            poll_url,
23            headers={'Authorization': f'Bearer {api_key}'}
24        )
25        result = response.json()
26
27        if result['status'] == 'COMPLETED':
28            # Parse the output (it's a JSON string)
29            outputs = json.loads(result['output'])
30            return outputs
31        elif result['status'] == 'FAILED':
32            raise Exception(result.get('error', 'Generation failed'))
33
34        # Wait 7 seconds before polling again
35        time.sleep(7)
36
37# Make the initial request
38response = requests.post(
39    url,
40    json=data,
41    headers={
42        'Authorization': f'Bearer {api_key}',
43        'Content-Type': 'application/json'
44    }
45)
46
47if response.status_code == 200:
48    result = response.json()
49    print('Request queued:', result)
50
51    # Start polling for results
52    outputs = poll_for_result(result['poll_url'])
53    print('Generation complete:', outputs)
54else:
55    print(f"Error: {response.status_code}")
56    print(response.text)