Python 및 OpenAI API를 사용하여 기본 기사 작성 도구를 만드는 방법-파이썬 튜토리얼-php.cn

Python 및 OpenAI API를 사용하여 기본 기사 작성 도구를 만드는 방법

PHPz

풀어 주다： 2024-07-23 18:47:44

원래의

773명이 탐색했습니다.

How to Create a Basic Article Writing Tool with Python and OpenAI API

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)

로그인 후 복사