Save The Date Card Creator with Nano Banana

 1import requests
 2import time
 3import json
 4
 5api_key = "YOUR_API_KEY"
 6url = "https://api.segmind.com/workflows/68ebbbead790112aa4e714ad-v3"
 7
 8data = {
 9    "Instruction": "Create a cute wedding “Save the Date” invitation. First, create a animated sticker of the couple using the image shared, with the faces exactly the same as input image. Then place the sticker in the middle of the invitation and create the invitation the style shared below.\n\nAdd stylish cursive text at the top: “Save the Date”, and below include names Sukruti & Harsha and wedding date Feb 25, 2026 in clean fonts. Soft romantic feel.\n\nStyle: Traditional south Indian/Telugu wedding style\n\n",
10    "Couple_image": "https://segmind-inference-inputs.s3.amazonaws.com/717317cd-e7ac-4375-9665-ab74beda709e-4a67a94a-d38a-4e0b-9b9b-daf7b26ca527.png"
11}
12
13def poll_for_result(poll_url):
14    """Poll the API until the generation is complete"""
15    while True:
16        response = requests.get(
17            poll_url,
18            headers={'Authorization': f'Bearer {api_key}'}
19        )
20        result = response.json()
21
22        if result['status'] == 'COMPLETED':
23            # Parse the output (it's a JSON string)
24            outputs = json.loads(result['output'])
25            return outputs
26        elif result['status'] == 'FAILED':
27            raise Exception(result.get('error', 'Generation failed'))
28
29        # Wait 7 seconds before polling again
30        time.sleep(7)
31
32# Make the initial request
33response = requests.post(
34    url,
35    json=data,
36    headers={
37        'Authorization': f'Bearer {api_key}',
38        'Content-Type': 'application/json'
39    }
40)
41
42if response.status_code == 200:
43    result = response.json()
44    print('Request queued:', result)
45
46    # Start polling for results
47    outputs = poll_for_result(result['poll_url'])
48    print('Generation complete:', outputs)
49else:
50    print(f"Error: {response.status_code}")
51    print(response.text)