파이썬으로 GPT API 사용하기 (코드예시)

GPT API를 이용한 “키워드” 기반 답변 프로그램 만들기

최근 챗GPT를 포함한 AI 기술이 세상을 빠르게 변화시키고 있죠. 오늘은 파이썬을 사용해서 OpenAI의 GPT-4 모델 API와 소통하는 간단한 프로그램을 함께 만들어보겠습니다.

필수 준비사항

OpenAI API를 사용하려면 먼저 회원가입 후 API 사용 신청을 해야 합니다. API 호출에는 비용이 발생할 수 있으니, 현재 정책(무료 사용량 등)은 OpenAI 공식 사이트에서 확인하시는 것이 좋습니다. 우선 API 키를 발급받아주세요.

https://platform.openai.com/ 사이트에서 ‘User’ → ‘API keys’ → ‘Create new secret key’를 선택하면 파이썬 코드에서 API를 호출할 때 사용할 수 있는 키 값을 발급받을 수 있습니다. 이제 파이썬 개발 환경에서 작업을 진행합니다. 기본적인 파이썬 설치 및 환경 설정 방법은 다른 자료를 참고해주세요.

OpenAI 플랫폼 사이트 안내

챗GPT 홈페이지 및 사용법 안내

1. 파이썬에서 필요한 GPT 라이브러리 설치

API 사용에 필요한 라이브러리를 먼저 설치합니다.

pip install openai

2. 코드 작성

API 키 설정

다시 한번 강조하지만, OpenAI API를 사용하려면 반드시 API 키가 필요합니다. 위에서 안내된 절차에 따라 OpenAI 웹사이트에서 키를 발급받아주세요.

발급받은 API 키를 다음 코드에 설정합니다.

import openai

# 여기에 본인의 API 키를 입력하세요.
api_key = “

YOUR_API_KEY

“

openai.api_key = api_key

“키워드” 기반 질문 함수 작성

이제 사용자가 입력한 ‘키워드’를 바탕으로 GPT-4에 질문하고 답변을 받아오는 함수를 작성해 보겠습니다.

def ask_gpt(keyword):
response = openai.Completion.create(
model=”gpt-4.0-turbo”,
prompt=f”Please explain about {keyword}.”,
max_tokens=150
)

answer = response.choices[0].text.strip()
return answer

사용자 인터페이스 구현

프로그램 실행 시, 사용자가 ‘키워드’를 입력하면 GPT-4에 해당 질문을 전달하고 받은 답변을 화면에 보여주는 간단한 사용자 인터페이스를 구현합니다.

def main():
while True:
keyword = input(“Please enter a keyword (or ‘exit’ to quit): “)

if keyword.lower() == ‘exit’:
print(“Exiting the program. Goodbye!”)
break

answer = ask_gpt(keyword)
print(f”\nAbout {keyword}: {answer}\n”)

if __name__ == “__main__”:
main()

3. 실행

위에서 작성한 코드를 ‘gpt4_keyword.py’와 같은 이름으로 저장합니다. 이제 이 파이썬 스크립트를 실행하여 프로그램을 사용해 볼 수 있습니다.

python gpt4_keyword.py

4. 전체코드

지금까지 설명된 모든 코드를 합친 전체 코드는 다음과 같습니다. ‘YOUR_API_KEY’ 부분을 반드시 본인이 발급받은 실제 API 키로 바꿔주세요.

import openai

# 1. API 설정
api_key = “YOUR_API_KEY”
openai.api_key = api_key

# 2. GPT-4에게 키워드 기반 질문을 하고 답변을 받는 함수
def ask_gpt(keyword):
response = openai.Completion.create(
model=”gpt-4.0-turbo”,
prompt=f”Please explain about {keyword}.”,
max_tokens=150
)

answer = response.choices[0].text.strip()
return answer

# 3. 사용자 인터페이스 구현
def main():
while True:
keyword = input(“Please enter a keyword (or ‘exit’ to quit): “)

if keyword.lower() == ‘exit’:
print(“Exiting the program. Goodbye!”)
break

answer = ask_gpt(keyword)
print(f”\nAbout {keyword}: {answer}\n”)

if __name__ == “__main__”:
main()

5. 마무리

오늘은 파이썬을 이용해 OpenAI의 GPT-4 API에 명령어를 전달하고 응답을 받는 기본적인 방법을 알아보았습니다. 참고로, ‘키워드’는 한글로 입력해도 잘 동작합니다. 이 코드를 바탕으로 여러분의 멋진 아이디어를 더해 더욱 유용한 프로그램을 만들어 보세요.

자주 묻는 질문