Chilkat2-Python

OpenAI (ChatGPT) Create Image

See more OpenAI ChatGPT Examples

Show how to send a POST to create an image. Gets the JSON response with URLs and then downloads images to files. Chilkat2-Python

import sys
import chilkat2

success = False

# This example assumes the Chilkat API to have been previously unlocked.
# See Global Unlock Sample for sample code.

http = chilkat2.Http()

# Implements the following CURL command:

# curl https://api.openai.com/v1/images/generations \
#   -H "Content-Type: application/json" \
#   -H "Authorization: Bearer $OPENAI_API_KEY" \
#   -d '{
#     "prompt": "A cute baby sea otter",
#     "n": 2,
#     "size": "1024x1024"
#   }'

# Use the following online tool to generate HTTP code from a CURL command
# Convert a cURL Command to HTTP Source Code

# Use this online tool to generate code from sample JSON:
# Generate Code to Create JSON

# The following JSON is sent in the request body.

# {
#   "prompt": "A cute baby sea otter",
#   "n": 2,
#   "size": "1024x1024"
# }

json = chilkat2.JsonObject()
json.UpdateString("prompt","A cute baby sea otter")
json.UpdateInt("n",2)
json.UpdateString("size","1024x1024")

# Adds the "Authorization: Bearer $OPENAI_API_KEY" header.
# This is NOT a real key.  Change the "sk-vi...." to your own key.
http.AuthToken = "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"

resp = chilkat2.HttpResponse()
success = http.HttpJson("POST","https://api.openai.com/v1/images/generations",json,"application/json",resp)
if (success == False):
    print(http.LastErrorText)
    sys.exit()

sbResponseBody = chilkat2.StringBuilder()
resp.GetBodySb(sbResponseBody)

jResp = chilkat2.JsonObject()
jResp.LoadSb(sbResponseBody)
jResp.EmitCompact = False

print("Response Body:")
print(jResp.Emit())

respStatusCode = resp.StatusCode
print("Response Status Code = " + str(respStatusCode))
if (respStatusCode >= 400):
    print("Response Header:")
    print(resp.Header)
    print("Failed.")
    sys.exit()

# Sample JSON response:
# (Sample code for parsing the JSON response is shown below)

# {
#   "created": 1589478378,
#   "data": [
#     {
#       "url": "https://..."
#     },
#     {
#       "url": "https://..."
#     }
#   ]
# }

# Sample code for parsing the JSON response...
# Use the following online tool to generate parsing code from sample JSON:
# Generate Parsing Code from JSON

sbLocalFilePath = chilkat2.StringBuilder()

# Download each image..
http.ClearHeaders()
http.AuthToken = ""

created = jResp.IntOf("created")
i = 0
count_i = jResp.SizeOfArray("data")
while i < count_i :
    jResp.I = i
    url = jResp.StringOf("data[i].url")

    sbLocalFilePath.SetString("c:/aaworkarea/image_")
    sbLocalFilePath.AppendInt(i + 1)
    sbLocalFilePath.Append(".png")

    success = http.Download(url,sbLocalFilePath.GetAsString())
    if (success == False):
        print(http.LastErrorText)
        sys.exit()

    i = i + 1