[AI] Open-AI를 통한 이미지 만들기

파이썬으로 openai 모듈을 이용해 Chat-gpt와 DALL-E 를 이용하여 이미지 만들기

---

Open AI 설치

pip install openai

가상환경에 openai를 설치한다.

코드작성

openai 이용방법

import openai

openai 를 improt 하여 사용할 준비를 만들어 준다.

openai.api_key = "API_KEY"

openai를 이용하기 위해서는 api key가 필수 !!
5 달러까진 무료이며, 이후에는 과금이 되니 남용하지 않게 주의 필요하다.
api key 생성은 openai 공식홈페이지에서 가능하다!

Chat_GPT 이용방법

gpt_prompt = []

gpt_prompt.append({
    "role":"system",
    "content":"Imagine the detail appearance of the input. Response shortly. Translating Korean into English in detail."
})

• gpt에 보낼 내용을 리스트 형태로 만든다.
• role에 system을 입력하여 시스템이 해야할 일(content)을 전달

gpt_prompt.append({
    "role":"user",
    "content":"피아노 치는 고양이"
})

• role에 user을 입력하여 유저가 입력한 정보(content)를 전달

prompt = openai.ChatCompletion.create(model="gpt-3.5-turbo", messages=gpt_prompt)
prompt = prompt["choices"][0]["message"]["content"]

• prompt에 gpt-3.5-turbo를 사용한 답변을 저장
• json 형태로 오기 때문에 필요한 답변만 다시 저장

DALL-E 전달

result = openai.Image.create(prompt=prompt, size="512x512")
print(result['data'][0]['url'])

• openai.Image.create를 사용하여 이미지를 원하는 크기(size="512x512")에 맞춰 생성
• 또한, json 형태로 오기에 원하는 이미지(url)를 찾아서 출력해준다.
• 출력된 url 클릭하면 사진이 나온다.

---

공부중.. 사진이 1시간만 유지되는거 같아서.. 다른 방법을 찾아봐야 될 듯하다..
간단히 실습하기엔 정말 좋은거 같다 !