Python과 OpenAI API를 사용하여 기사 작성 도구를 만들려면 여러 단계가 필요합니다.

환경 설정, 필요한 라이브러리 설치, 기사 생성을 위한 코드 작성 과정을 진행하겠습니다.

전제 조건

시작하기 전에 다음 사항을 확인하세요.

• 시스템에 Python이 설치되어 있습니다(Python 3.6 권장).
• OpenAI API 키입니다. OpenAI 웹사이트에 가입하시면 받으실 수 있습니다.

1단계: 환경 설정

먼저 가상 환경을 생성하고 필요한 라이브러리를 설치해야 합니다. 터미널을 열고 다음 명령을 실행하세요:

# Create a virtual environment
python -m venv myenv

# Activate the virtual environment
# On Windows
myenv\Scripts\activate
# On macOS/Linux
source myenv/bin/activate

# Install necessary libraries
pip install openai

2단계: 코드 작성

article_writer.py와 같은 Python 파일을 만들고 원하는 텍스트 편집기에서 엽니다. 코드를 여러 섹션으로 나누겠습니다.

필수 라이브러리 가져오기

import openai
import os

OpenAI API 키 설정

'your-api-key'를 실제 OpenAI API 키로 바꿔야 합니다.

# Set up the OpenAI API key
openai.api_key = 'your-api-key'

기사 생성 기능

OpenAI의 GPT 모델을 사용하여 주제를 입력으로 받아 기사를 반환하는 함수를 작성하겠습니다.

def generate_article(topic):
    response = openai.Completion.create(
        engine="text-davinci-003",
        prompt=f"Write an article about {topic}.",
        max_tokens=1024,
        n=1,
        stop=None,
        temperature=0.7,
    )
    return response.choices[0].text.strip()

도구를 실행하는 주요 기능

def main():
    print("Welcome to the Article Writing Tool!")
    topic = input("Enter the topic for your article: ")
    print("\nGenerating article...\n")
    article = generate_article(topic)
    print(article)

if __name__ == "__main__":
    main()

3단계: 도구 실행

article_writer.py 파일을 저장하고 터미널에서 실행하세요.

python article_writer.py

주제를 입력하라는 메시지가 표시되고 도구는 해당 주제를 기반으로 기사를 생성합니다.

4단계: 개선 및 사용자 정의

이것은 기사 작성 도구의 기본 버전이지만 고려할 수 있는 몇 가지 향상된 기능이 있습니다.

오류 처리 추가

도구를 더욱 강력하게 만들려면 API 오류 또는 잘못된 입력을 관리하는 오류 처리를 추가하세요.

def generate_article(topic):
    try:
        response = openai.Completion.create(
            engine="text-davinci-003",
            prompt=f"Write an article about {topic}.",
            max_tokens=1024,
            n=1,
            stop=None,
            temperature=0.7,
        )
        return response.choices[0].text.strip()
    except openai.error.OpenAIError as e:
        return f"An error occurred: {str(e)}"

프롬프트 사용자 정의

뉴스 기사, 블로그 게시물, 연구 논문 등 보다 구체적인 유형의 기사를 얻으려면 프롬프트를 맞춤설정하세요.

def generate_article(topic, style="blog post"):
    prompt = f"Write a {style} about {topic}."
    try:
        response = openai.Completion.create(
            engine="text-davinci-003",
            prompt=prompt,
            max_tokens=1024,
            n=1,
            stop=None,
            temperature=0.7,
        )
        return response.choices[0].text.strip()
    except openai.error.OpenAIError as e:
        return f"An error occurred: {str(e)}"

주 함수에서 스타일을 포함하도록 입력을 수정합니다.

def main():
    print("Welcome to the Article Writing Tool!")
    topic = input("Enter the topic for your article: ")
    style = input("Enter the style of the article (e.g., blog post, news article, research paper): ")
    print("\nGenerating article...\n")
    article = generate_article(topic, style)
    print(article)

마무리

이 단계를 따르면 Python과 OpenAI API를 사용하여 기본 기사 작성 도구를 만들 수 있습니다.

이 도구는 기사를 파일에 저장하거나, 웹 인터페이스와 통합하거나, 생성된 콘텐츠에 대한 더 많은 사용자 정의 옵션을 제공하는 등의 추가 기능을 통해 더욱 향상될 수 있습니다.

자세히 알아보고 싶으신가요? ZeroByteCode.