ChatGPT API 연동

소스코드는 제가 짠게 아닙니다. 번역만 한글로 했는데 제대로 됐는지 모르겠습니다. (파파고 ㅎ)

원소스)

https://teddylee777.github.io/python/chatgpt-blog-automation/

import openai

# https://platform.openai.com/account/api-keys

openai.api_key = "*** 키는 바꾸세요 ***"

city = "런던"

topic = f"당신이 {city}이라는 도시를 여행할때 반드시 방문해야 할 상위 10개의 레스토랑"

category = "travel"

print(topic)

# 프롬프트 (내용 수정 가능)

prompt = f'''

한글로 작성해 주세요.

마크다운 포맷으로 블로그 글을 작성하세요.

위의 영어를 번역해줘

{topic} 같은 당신 블로그의 주제를 작성하세요.

구문의 중요한 단어는 강조, 굵게, 이탤릭체로 표시하세요.

레스토랑의 주소를 포함해 주세요. 메뉴 추천과 다른 유용한 정보(오픈, 클로징 시간)를 리스트 스타일로 포함하세요.

블로그 전체를 10분 이내로 만들어 주세요.

이 글의 독자는 20~40세 입니다.

몇개의 해시태그를 만들어서 라인의 끝에 추가하세요.

전체글의 요약본을 블로그 포스트의 시작에 추가하세요.

'''

def generate_blog(topic, prompt):

    # 모델 엔진 선택

    model_engine = "text-davinci-003"

    # 맥스 토큰

    max_tokens = 2048

    # 블로그 생성

    completion = openai.Completion.create(

        engine=model_engine,

        prompt=prompt,

        max_tokens=max_tokens,

        temperature=0.3,      # creativity

        top_p=1,

        frequency_penalty=0,

        presence_penalty=0

    )

    return completion

response = generate_blog(topic, prompt)

# 생성된 글 출력

print(response.choices[0].text)