본문 바로가기

Python AI 자동 소설 생성기 만들기

디지털 백수 2025. 3. 24.
반응형

 

AI 기술이 발전하면서 이제 소설도 자동으로 생성할 수 있습니다. 이번 프로젝트에서는 Python과 OpenAI ChatGPT API를 활용하여 AI가 자동으로 이야기를 만들어내는 프로그램을 개발합니다.

📌 1. AI 자동 소설 생성기란?

이 프로그램은 사용자가 원하는 소설 장르, 등장인물, 배경 등을 입력하면, AI가 이를 바탕으로 짧은 소설을 자동 생성합니다.

📌 2. 필요한 패키지 설치

Python 환경에서 OpenAI API를 사용하기 위해 다음 패키지를 설치해야 합니다.

pip install openai

📌 3. Python 코드: AI 자동 소설 생성기

아래 코드를 실행하면 AI가 사용자가 원하는 스타일의 소설을 자동으로 생성해 줍니다.

import openai  

# OpenAI API 키 설정
openai.api_key = "YOUR_API_KEY"  # 여기에 본인의 API 키 입력

def generate_story(genre, characters, setting):
    prompt = f"다음 조건을 만족하는 짧은 소설을 작성해줘.\n\n장르: {genre}\n등장인물: {characters}\n배경: {setting}\n\n소설:"
    
    response = openai.ChatCompletion.create(
        model="gpt-3.5-turbo",
        messages=[{"role": "user", "content": prompt}]
    )

    story = response["choices"][0]["message"]["content"]
    return story.strip()

# 사용자의 입력값
genre = "판타지"
characters = "용사, 마법사, 드래곤"
setting = "중세 시대 마법 세계"

print("📖 AI 생성 소설:\n", generate_story(genre, characters, setting))

📌 4. 코드 설명

  • 사용자가 원하는 장르, 캐릭터, 배경을 입력
  • ✅ AI가 자동으로 해당 조건에 맞는 소설을 생성
  • ✅ 명령어(prompt)를 활용해 보다 창의적인 이야기 생성

📌 5. 실행 예시

📖 AI 생성 소설:

용사는 고대의 마법사가 지키고 있는 드래곤의 동굴을 찾아갔다. 
전설에 따르면, 그곳에는 세상을 구할 수 있는 황금의 보물이 숨겨져 있다고 한다.
그러나 용사가 도착했을 때, 드래곤은 이미 그를 기다리고 있었다...

📌 6. 확장 아이디어 🚀

  • 웹 UI 추가 → Flask/Django로 웹에서 실행 가능
  • 챕터별 생성 → 1장에서 5장까지 AI가 연속적인 이야기 작성
  • 소설 스타일 선택 → "로맨스", "SF", "공포" 등 스타일 변경 가능

📌 7. 마무리

이번 프로젝트에서는 Python을 활용하여 AI 자동 소설 생성기를 만드는 방법을 배웠습니다.

이 기술을 활용하면 창작의 한계를 넘어 AI와 협업하여 창의적인 이야기를 만들 수 있습니다. 🚀

반응형

댓글