본문 바로가기

프로그램 개발

Open AI API 사용 방법 - 2

4. 기사 작성 인공지능

이전에 증권사에서 뉴스를 발표하면서 AI가 만든 뉴스라고 발표하는 뉴스가 있었다. 형태도 비슷하고, 나오는 내용도 비슷해서 자동으로 생성되는 것 같다는 생각은 했지만 그때는 어떻게 만드는지 궁금했었다.

 

OpenAI를 사용해서 뉴스를 자동으로 생성하는 프로그램을 가벼와서 어떻게 반응을 하는지 수정하면서 테스트해 봤다. 먼저 아래와 같은 함수를 만든다.

 

def assist_journalist(
        facts: list[str], tone: str = str, length_words: int = 100, style: str = "journalistic"
):
    response = client.chat.completions.create(
        model="gpt-3.5-turbo",
        messages=[
            {"role": "system", "content": "You are a helpful assistant."},
            {"role": "user", "content": "Write a news article based on the following facts: " + ", ".join(facts)},
            {"role": "user", "content": "Tone: " + tone},
            {"role": "user", "content": "Length: " + str(length_words) + " words"},
            {"role": "user", "content": "Style: " + style},
        ],
    )

    return response.choices[0].message.content

 

정의된 함수를 사용해서 뉴스를 만드는 테스트를 해 보면,

 

facts = [
    "The stock market is down 5% today.",
    "The president is visiting France.",
    "A new study shows that coffee is good for you."
]
print(assist_journalist(facts, tone="neutral", length_words=200, style="journalistic"))

 

AI가 만든 뉴스 내용을 복사해서 자세히 보면 다음과 같다.

 

Stock Market Drops 5% as President Visits France; Study Reveals Health Benefits of Coffee

The stock market took a hit today, with a 5% drop in major indices, 
leaving investors on edge as uncertainty looms over the economy. 
Analysts attributed this decline to concerns over global trade tensions 
and slowing economic growth. Despite the dip, experts remain cautiously 
optimistic about the market's resilience.

Meanwhile, President Smith is currently on an official visit to France, 
where discussions on trade, security, and other bilateral issues are 
expected to take center stage. The visit is viewed as an opportunity 
to strengthen diplomatic ties between the two nations.

In health news, a new study has revealed surprising benefits of coffee consumption. 
Researchers found that moderate coffee intake can have various health benefits, 
including improved alertness, metabolism, and even a reduced risk of certain diseases. 
This study has sparked interest among coffee lovers and health enthusiasts alike, 
encouraging further research into the potential advantages of this popular beverage.

As events unfold, market fluctuations, international diplomacy, and health 
breakthroughs continue to shape the news cycle, capturing the public's attention 
around the globe.

 

실제 기사에서 보는 듯한 제법 그럴듯한 뉴스를 생성해 내는 것을 확인할 수 있었다. 이 정도면 데이터를 넣으면 기사는 자동으로 작성해 줄 것 같다. 자동으로 블로그를 작성하는 프로그램도 한번 만들어 봐야겠다.

5. 번역하는 인공지능

만약 작성된 뉴스를 한글로 번역하도록 만들려면 아래와 같이 함수를 작성하고 위에서 작성된 뉴스를 함수에 놓으면 한글로 번역된 기사가 만들어진다.

 

def translate_text(text: str, target_language: str):
    content = f"Translate the following text to {target_language}: {text}"
    response = client.chat.completions.create(
        model="gpt-3.5-turbo",
        messages=[
            {"role": "system", "content": "You are a helpful assistant."},
            {"role": "user", "content": content},
        ]
    )
    return response.choices[0].message.content

 

두 함수를 연결해서 기사를 만들도록 해 보면,

 

facts = [
    "The stock market is down 5% today.",
    "The president is visiting France.",
    "A new study shows that coffee is good for you."
]
news = assist_journalist(facts, tone="neutral", length_words=200, style="journalistic")

translated_text = translate_text(news, "korean")
print("Translated Text:\n", translated_text)

 

실제 나온 내용을 복사해 놓은 텍스트는 다음과 같다.

 

 주식 시장이 5% 하락하며 대통령이 프랑스를 방문하는 가운데, 커피의 건강 상 이점을 밝혀내는 새로운 연구 발표

오늘 주식 시장은 중요 지수들이 5% 하락하는 중요한 하락을 경험하였는데, 
이는 세계 경제 안정성에 대한 우려가 높아져 있는 상황에서 발생했습니다. 
투자자들은 무역 긴장과 지리학적 요인들이 시장 성과에 영향을 계속 
주고 있는 가운데 상황을 밀접하게 지켜보고 있습니다.

다른 소식으로, 대통령은 현재 프랑스를 외교 방문 중이며, 이는 프랑스 리더들과 
양자간 관계를 강화하고 핵심 문제를 논의하기 위한 것입니다. 이 방문은 현재의 
복잡한 지리학적 환경 속에서 국제적 협력과 협조가 중요한 시점에 일어납니다.

한편, 새로운 연구가 커피 섭취의 놀라운 건강 이점을 밝혀냈습니다. 연구자들은 
적당량의 커피 섭취가 심혈관 질환이나 신경퇴행성 장애를 비롯한 특정 질병의 
위험 감소와 관련이 있다는 것을 발견했습니다. 이 최신 연구는 적당한 양의 
커피를 섭취하는 것이 건강에 이익을 줄 수 있는 잠재적 이점을 강조하는 
연구들의 늘어나는 근거에 더해졌습니다.

금융 시장, 국제 관계, 건강 연구 등에서 발생하는 발전들에 따라 이해관계자들은 
더 많은 통찰과 함의를 주시할 계획입니다.