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

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

안녕하세요. 오픈AI에서 등장한 챗GPT로 인해서 엄청난 변화가 있었습니다. 당연히 현재까지도 인기는 어마어마한데요. 오늘은 파이썬을 이용하여 OpenAI의 GPT-4 모델에 명령어를 입력하면 답변을 받는 간단한 프로그램을 만들어보겠습니다.

필수 준비사항

기본적으로 GPT API를 사용하시기 위해서는 회원가입 후 API 사용신청을 하셔야 합니다. 제가 가입할때는 초기에 사용할 수 있는 테스트 용량을 제공했는데요. 현시점은 어떤지 확인이 안되는 군요. 우선 API가입 후 API Keys를 발급 받아주세요.

GPT-API-Keys-발급

https://platform.openai.com/ 사이트에서 위와 같이 “User -> API keys -> Create new secret key”를 선택하시면 파이썬에서 호출시 사용할 수 있는 키값을 발급 받을 수 있습니다. 이제 파이썬에서 진행합니다. 기본적인 파이썬 구성은 검색해보시면 쉽게 찾을 수 있으니 패스합니다.

ChatGPT 홈페이지 안내

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

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

pip install openai

2. 코드 작성

API 키 설정

다시 한번 말씀 드리지만 OpenAI의 API를 사용하려면 먼저 OpenAI에서 제공하는 API 키가 필요합니다. 위에서 설명한 부분을 참고해서 OpenAI 웹사이트에서 계정을 생성하고 API 키를 발급 받아주세요.

발급 받은 GPT 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
GPT-API-CODE-안내

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. 마무리

오늘은 간단하게 파이썬을 사용해서 GPT-4모델에 명령어를 입력해서 사용하는 방법에 대해서 알아보았습니다. 아시겠지만 실제 키워드는 한글로 입력하셔도 동작합니다. 참고하시고 코드를 활용해서 더 멋진 아이디어를 집어넣어서 프로그램을 만들어보세요.

Leave a Comment