1import openai, requests
2from apify import Actor
3
4
5async def main():
6 async with Actor:
7 actor_input = await Actor.get_input() or {}
8
9 openai_api_key = actor_input.get('openai_api_key')
10 prompt = actor_input.get('prompt')
11 count = actor_input.get('count')
12 width = actor_input.get('width')
13 height = actor_input.get('height')
14
15 try:
16 response = openai.Image.create(
17 openai_api_key,
18 prompt = prompt,
19 n = count,
20 size = f"{width}x{height}",
21 )
22 image_url_data = response['data']
23 i = 0
24 for image_url_obj in image_url_data:
25 url = image_url_obj['url']
26 image = requests.get(url)
27 await Actor.set_value(f'image_{i}.png', image.content)
28 i += 1
29
30 except openai.error.OpenAIError as e:
31 print(e.http_status)
32 print(e.error)