AI 에이전트 만들기

from openai import OpenAI
client = OpenAI()   # 키는 .env에서 자동으로

def ask_ai(prompt):
    res = client.chat.completions.create(
        model="gpt-4o-mini",
        messages=[{"role": "user", "content": prompt}])
    return res.choices[0].message.content

# 긴 기사는 삼중 따옴표로 붙여넣기
news = """삼성전자가 청소년 10만 명을 대상으로 인공지능 교육을 지원하기로 했다.
이번 프로그램은 전국 중·고등학생을 대상으로 하며, AI 기초 개념부터 실습까지 배울 수 있는 커리큘럼으로 구성된다.
교육은 온라인과 오프라인을 병행해 지역과 관계없이 참여할 수 있도록 운영된다.
삼성전자는 디지털 격차를 줄이고 미래 인재를 양성하기 위해 이 프로그램을 마련했다고 밝혔다.
참여 학생에게는 수료증과 함께 우수자 대상 추가 교육 기회도 제공된다.
회사 측은 이번 지원을 시작으로 AI 교육 대상을 점차 확대해 나갈 계획이라고 설명했다."""

summary = ask_ai(f"다음 기사를 3줄로 요약해줘. 각 줄은 한 문장으로:\n{news}")
print(summary)

# 이번엔 짧은 기사로 테스트
news = """오늘 우리 동네에 새 도서관이 문을 열었다. 주말에는 어린이 책 읽기 행사도 열린다고 한다."""

print("기사 길이:", len(news))

if len(news) > 500:
    print(ask_ai(f"다음 기사를 3줄로 요약해줘:\n{news}"))
else:
    print("기사가 짧아요. 그대로 읽어보세요:")
    print(news)

history = []   # 요약 결과를 모아두는 상자

summary = ask_ai(f"다음 기사를 3줄로 요약해줘:\n{news}")
history.append(summary)
print(f"오늘 {len(history)}번째 요약입니다.")
print(summary)

# 1번 봇: 기사를 주제로 분류
category = ask_ai(f"다음 기사의 주제를 한 단어로 알려줘(예: 교육, 경제, 스포츠):\n{news}")

# 2번 봇: 분류 결과를 받아 요약
summary = ask_ai(f"다음은 '{category}' 주제 기사야. 3줄로 요약해줘:\n{news}")

print("주제:", category)
print("요약:", summary)