AI Style Makeover with Personality Fit

 1import requests
 2import time
 3import json
 4
 5api_key = "YOUR_API_KEY"
 6url = "https://api.segmind.com/workflows/682a05de00eb0e1658c0b672-v1"
 7
 8data = {
 9    "input_image": "https://segmind-inference-inputs.s3.amazonaws.com/d3276015-9872-4508-ab01-2b87f038a2a9-varun.png",
10    "optional_promot": "Based on this photo, give me a style makeover that fits my [personality], [ambition], and [lifestyle]based on my uploaded image. Make it look like me. full body image with proper head to body proportion (don't generate big head with small body), good background.\n\n\nmake sure only my face is generated in the final image. no other person should be there only one person"
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)