Amazing-Python-Scripts

1
def dalle(message):
2
    import requests
3
    import openai
4

5
    # Set up the OpenAI API client
6
    openai.api_key = "your-api-key"
7

8
    # Set up the model and prompt
9
    model_engine = "text-davinci-003"
10
    prompt = message
11

12
    # Generate a response
13
    response = openai.Image.create(
14
        prompt=prompt,
15
        n=1,
16
        size="1024x1024"
17
    )
18
    image_url = response['data'][0]['url']
19
    response = requests.get(image_url)
20
    with open("image.jpg", "wb") as f:
21
        f.write(response.content)
22

23
    print(image_url)
24

25

26
dalle(input("Enter something : "))
27