본문 바로가기
카테고리 없음

OpenAI Dall-e 이미지 API 연동 (파이썬) 방법

by 스마트파베르 2024. 3. 9.
반응형

ㅇ OpenAI 가입, API key 발급, 모델 선택 등에 대한 설명은 이전에 만든 티스토리 글을 참조해 주세요. 

  https://ai-democracy.tistory.com/18

 

ChatGPT API 연동 (파이썬) 방법 - 텍스트 응답 처리 방식 2가지

ㅇ OpenAI 가입하고, API key를 발급받는다. - 아래 사이트에서 +Create new secret key를 눌러서 키 발급 받음 : https://platform.openai.com/api-keys ㅇ openAI 에서 API 사용 레퍼런스를 제공한다. 기본적인 text를 만

ai-democracy.tistory.com

 

ㅇ Dall-e 이미지 연동 API 코딩

from openai import OpenAI

client = OpenAI(api_key = "발급 받은 Key")

completion = client.images.generate(
  model="dall-e-3",
  #promp#t="A cute baby sea otter",
  prompt = "나는 노후에 바닷가에 살고 싶다. 바닷가에 지을 작은 주택을 설계하여 주라, 집은 2층이며, 마당이 있고, 뒷 뜰에는 벗나무가 있고. 마당 앞에는 해변이 있다",
  n = 1,
  size="1024x1024"
)

""" response 형식
{
  "created": 1589478378,
  "data": [
    {"url": "https://..."},
    {"url": "https://..."}
  ]
}

"""

response_img = completion.data[0].url
print(response_img)   #  URL로 접속하면 이미지를 브라우저로 볼수 있다

# 이미지를 화면에 직접 표시 방법
res = requests.get(response_img)
img = Image.open(BytesIO(res.content))
img

 

ㅇ 응답

   - URL 응답  :  https://oaidalleapiprodscus.blob.core.windows.net/private/org-8tTpIMD2w6g.....

  - 이미지 표시 :

OpenAI 이미지 API의 Dall-e model로 그린 그림

 

반응형