gpt-author

该项目利用 GPT-4、Stable Diffusion 和 Anthropic API 的串联调用，生成一部原创奇幻小说。用户只需提供初始提示，并指定希望的小说章节数，AI 就会自动生成整部小说，并导出为兼容电子书阅读器的 EPUB 文件。

一部 15 章的小说，制作成本低至 4 美元，且仅需几分钟即可完成写作。

新增 2024 年 4 月 2 日：Claude 3 版本

我添加了一个全新的 gpt-author 版本，充分利用了 Anthropic 的 Claude 3 模型。这个版本代码和规划都更加简单，但写出的小说质量却远超以往。请尝试使用仓库中的 Claude_Author.ipynb 笔记本来体验吧！

---

本仓库中提供了一些由原始（已过时）系统生成的小说示例。要阅读其中任意一本，您可以下载其文件，在 https://www.fviewer.com/view-epub 上查看，或将它安装到您的 Kindle 等设备上。

工作原理

首先，AI 会根据给定的提示生成一系列潜在的情节大纲。随后，它会选择最具吸引力的情节，对其进行优化，并提取一个标题。接着，AI 会基于选定的情节生成包含指定章节数的详细故事线，并进一步优化该故事线。之后，AI 会按照情节发展，结合前几章的内容，逐章撰写每一章节的内容。最后，AI 会生成封面设计的提示并创作封面图。所有内容整合完成后，整部小说将被编译成 EPUB 格式文件。

使用方法

您可以在 Google Colab 中运行该项目的最新版本，或在本地 Jupyter 笔记本中运行（建议使用 gpt_author_v2.ipynb 以获得最佳效果）！

在 Google Colab 中，只需打开笔记本，填入您的 API 密钥，然后按顺序运行各个单元格即可。

如果您使用的是本地 Jupyter 笔记本，则需要先安装必要的依赖项。您可以通过在终端中运行以下命令来完成安装：

pip install openai ebooklib requests

在笔记本的最后一部分，您可以自定义小说的提示词和章节数。例如：

prompt = "氛围类似于《波西·杰克逊》或《哈利·波特》，但剧情完全不同。故事设定在现代。加入一些科技元素。"
num_chapters = 20
writing_style = "清晰易懂，类似青少年小说风格。描写细腻，有时略显冗长。"
novel, title, chapters, chapter_titles = write_fantasy_novel(prompt, num_chapters, writing_style)

这将根据给定的提示生成一部包含 20 章的小说。请注意——如果指定的章节数少于 7 章，可能会出现一些问题。

贡献

欢迎提交贡献、报告问题或提出功能需求！

一些初步的想法：

• 修改代码，使其仅使用 GPT-3.5-Turbo、GPT-3.5-Turbo-16k 或 Claude Instant 模型运行（可能需要对早期章节进行一定程度的压缩或摘要，以避免在生成后续章节时耗尽 token）。
• 改进首章的生成机制——首章写得越好，整部小说的质量就越高。
• 优化提示词的设计，因为当前的提示词编写较为仓促。
• 在每个步骤中增加更多的检查和优化迭代，进一步提升生成效果。
• 在生成优化内容之前，先让模型识别出可以添加到提示词中的改进点，这样有望显著提升整体表现。
• 扩展功能，不仅限于奇幻题材，允许用户生成其他类型的小说。
• 修复导致部分章节提前截断的问题。

许可证

本项目采用 MIT 许可证 开放源代码。

联系方式

Matt Shumer - @mattshumer_

项目链接：https://github.com/mshumer/gpt-author/

gpt-author 快速上手指南

gpt-author 是一个利用 GPT-4、Stable Diffusion 和 Anthropic API 自动生成原创奇幻小说的开源项目。用户只需提供初始提示和章节数量，AI 即可在几分钟内生成完整小说并输出为 EPUB 格式电子书。

环境准备

• 系统要求：支持 Python 3.8+ 的操作系统（Windows/macOS/Linux）
• 前置依赖：
  • Google 账号（用于运行 Google Colab 版本，推荐新手使用）
  • 或本地安装 Jupyter Notebook 环境
  • API Keys：需准备 OpenAI API Key 和/或 Anthropic API Key

安装步骤

方案一：使用 Google Colab（推荐）

无需本地安装，直接在浏览器运行：

1. 打开 gpt-author Colab 笔记本
2. 点击“在 Colab 中打开”
3. 按照单元格顺序执行，填入你的 API Keys 即可

方案二：本地 Jupyter Notebook

若需在本地运行，请执行以下命令安装依赖：

pip install openai ebooklib requests

注：国内用户可使用清华镜像源加速安装：

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple openai ebooklib requests

基本使用

在 Notebook 的最后一个代码单元格中，自定义你的小说参数并运行：

prompt = "Similar to Percy Jackson or Harry Potter in terms of vibes, but a different plot entirely. Set in modern day. Add some element of technology to it."
num_chapters = 20
writing_style = "Clear and easily understandable, similar to a young adult novel. Highly descriptive and sometimes long-winded."
novel, title, chapters, chapter_titles = write_fantasy_novel(prompt, num_chapters, writing_style)

说明：

• prompt：小说主题描述（英文效果更佳）
• num_chapters：章节数量（建议不少于 7 章，否则可能出错）
• writing_style：写作风格描述
• 运行完成后将自动生成 EPUB 文件，可直接导入 Kindle 或使用在线阅读器查看。

💡 提示：项目已更新支持 Claude 3 模型（Claude_Author.ipynb），生成的小说质量更高，代码更简洁，强烈建议尝试。