PureBasic

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. PureBasic

IncludeFile "CkHttpResponse.pb"
IncludeFile "CkHttp.pb"
IncludeFile "CkStringBuilder.pb"
IncludeFile "CkJsonObject.pb"

Procedure ChilkatExample()

    success.i = 0

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

    http.i = CkHttp::ckCreate()
    If http.i = 0
        Debug "Failed to create object."
        ProcedureReturn
    EndIf

    ; 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.i = CkJsonObject::ckCreate()
    If json.i = 0
        Debug "Failed to create object."
        ProcedureReturn
    EndIf

    CkJsonObject::ckUpdateString(json,"prompt","A cute baby sea otter")
    CkJsonObject::ckUpdateInt(json,"n",2)
    CkJsonObject::ckUpdateString(json,"size","1024x1024")

    ; Adds the "Authorization: Bearer $OPENAI_API_KEY" header.
    ; This is NOT a real key.  Change the "sk-vi...." to your own key.
    CkHttp::setCkAuthToken(http, "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")

    resp.i = CkHttpResponse::ckCreate()
    If resp.i = 0
        Debug "Failed to create object."
        ProcedureReturn
    EndIf

    success = CkHttp::ckHttpJson(http,"POST","https://api.openai.com/v1/images/generations",json,"application/json",resp)
    If success = 0
        Debug CkHttp::ckLastErrorText(http)
        CkHttp::ckDispose(http)
        CkJsonObject::ckDispose(json)
        CkHttpResponse::ckDispose(resp)
        ProcedureReturn
    EndIf

    sbResponseBody.i = CkStringBuilder::ckCreate()
    If sbResponseBody.i = 0
        Debug "Failed to create object."
        ProcedureReturn
    EndIf

    CkHttpResponse::ckGetBodySb(resp,sbResponseBody)

    jResp.i = CkJsonObject::ckCreate()
    If jResp.i = 0
        Debug "Failed to create object."
        ProcedureReturn
    EndIf

    CkJsonObject::ckLoadSb(jResp,sbResponseBody)
    CkJsonObject::setCkEmitCompact(jResp, 0)

    Debug "Response Body:"
    Debug CkJsonObject::ckEmit(jResp)

    respStatusCode.i = CkHttpResponse::ckStatusCode(resp)
    Debug "Response Status Code = " + Str(respStatusCode)
    If respStatusCode >= 400
        Debug "Response Header:"
        Debug CkHttpResponse::ckHeader(resp)
        Debug "Failed."
        CkHttp::ckDispose(http)
        CkJsonObject::ckDispose(json)
        CkHttpResponse::ckDispose(resp)
        CkStringBuilder::ckDispose(sbResponseBody)
        CkJsonObject::ckDispose(jResp)
        ProcedureReturn
    EndIf

    ; 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

    url.s
    sbLocalFilePath.i = CkStringBuilder::ckCreate()
    If sbLocalFilePath.i = 0
        Debug "Failed to create object."
        ProcedureReturn
    EndIf

    ; Download each image..
    CkHttp::ckClearHeaders(http)
    CkHttp::setCkAuthToken(http, "")

    created.i = CkJsonObject::ckIntOf(jResp,"created")
    i.i = 0
    count_i.i = CkJsonObject::ckSizeOfArray(jResp,"data")
    While i < count_i
        CkJsonObject::setCkI(jResp, i)
        url = CkJsonObject::ckStringOf(jResp,"data[i].url")

        CkStringBuilder::ckSetString(sbLocalFilePath,"c:/aaworkarea/image_")
        CkStringBuilder::ckAppendInt(sbLocalFilePath,i + 1)
        CkStringBuilder::ckAppend(sbLocalFilePath,".png")

        success = CkHttp::ckDownload(http,url,CkStringBuilder::ckGetAsString(sbLocalFilePath))
        If success = 0
            Debug CkHttp::ckLastErrorText(http)
            CkHttp::ckDispose(http)
            CkJsonObject::ckDispose(json)
            CkHttpResponse::ckDispose(resp)
            CkStringBuilder::ckDispose(sbResponseBody)
            CkJsonObject::ckDispose(jResp)
            CkStringBuilder::ckDispose(sbLocalFilePath)
            ProcedureReturn
        EndIf

        i = i + 1
    Wend


    CkHttp::ckDispose(http)
    CkJsonObject::ckDispose(json)
    CkHttpResponse::ckDispose(resp)
    CkStringBuilder::ckDispose(sbResponseBody)
    CkJsonObject::ckDispose(jResp)
    CkStringBuilder::ckDispose(sbLocalFilePath)


    ProcedureReturn
EndProcedure