[Cowork#002] ChatGPT API 연동하기.

<GPT의 주니어 Human 비서 요한 👨‍💻>

논문 쉽게 읽기 프로젝트 #2: ChatGPT API 연동하기.

이번에는 논문의 텍스트를 자동으로 번역, 요약해주는 프로그램을 만들기 위한 사전 작업으로 프롬프트를 입력 받아 GPT에게 넣어주는 Python 프로그램을 만들어 보려고 한다. API 연동을 위해서는 Key 같은게 필요할 것 같아, 저물어가는 구글신에게 물어봐 Open AI 사이트에서 Account API Keys를 받아왔다. (아직 죽지 않았어, 구글)

(Prompt#1) I would like to create a python program that can save the output of the prompt input to a txt file using the ChatGPT API. How can I do this?

ChatGPT에게 물어보니 동작하는 코드를 던져주었지만, 모델명이 'text-davinci-002'로 GPT3 모델을 안내해 주었다. 몇 가지 프롬프트를 입력해 봤는데 출력이 영 시원치 않다. 그래서 GPT4에게 같은 내용을 물어보니 https://beta.openai.com/signup 라는 최신 링크를 안내해 주면서 프로그램 설치 과정부터 더 상세히 안내해 주었다 (다정해😍). 링크에 들어가 GPT-4 API waitlist도 신청하고, model 페이지에서 ChatGPT에 해당하는 'gpt-3.5-turbo' 모델명도 알게 되었다. 나름 문제 해결 능력이 있는 Human 비서😎. 해당 모델을 입력 후 몇 번의 디버깅을 거쳐 ChatGPT와 유사한 수준의 대답을 하는 프로그램을 작성할 수 있었다.

import openai
import os

# Replace YOUR_API_KEY with your actual API key
openai.api_key = "YOUR_API_KEY"

def generate_text(prompt):
    response = openai.ChatCompletion.create(
        model="gpt-3.5-turbo",
        messages=[
            {"role": "system", "content": "You are a helpful assistant."},
            {"role": "user", "content": prompt},
        ],
        max_tokens=2048,
        n=1,
        stop=None,
        temperature=0.7,
    )
    message = response.choices[0].message['content'].strip()
    return message

def save_to_file(content, filename):
    with open(filename, 'w') as f:
        f.write(content)

if __name__ == "__main__":
    user_prompt = input("Enter your prompt: ")
    generated_text = generate_text(user_prompt)
    print("Generated text:\n", generated_text)

    output_filename = "output.txt"
    save_to_file(generated_text, output_filename)
    print(f"Generated text saved to {output_filename}")

GPT3.5님도 돌아가는 코드를 던져 주긴했지만, 돈을 지불하는 서비스는 역시 다르다는 것을 알려주신 갓GPT4님👍 이번에는 지난 번보다 익숙해져 돌아가는 코드를 얻는데까지는 꽤 빨리 도달했지만, 원하는 최신 모델이 아니었기 때문에 Open AI 사이트를 배회하는데 시간을 많이 사용했다. 생각해 보니 다 GPT님께 물어볼 걸 괜히 주제넘게 내가 검색해서 시간을 낭비한 걸까?

✅ 총 작업 소요 시간: 61분 26초