自动小说创作工具gpt-author

程序笔记 2024年09月17日09:03

GPT-Author 是一个基于 GPT-4 语言模型的自动小说创作工具。它可以根据用户提供的初始提示和章节数生成一整本奇幻小说，并自动打包为电子书格式。

GPT-Author Github地址：https://github.com/mshumer/gpt-author

该项目利用一系列 GPT-4、Stable Diffusion 和 Anthropic API 调用来生成原创奇幻小说。用户可以提供初始提示并输入他们想要的章节数，然后人工智能会生成整本小说，输出与电子书阅读器兼容的 EPUB 文件。

一本 15 章的小说的制作成本只需 4 美元，而且只需几分钟即可完成。

本存储库中提供了一些输出新颖的示例。要阅读其中一篇，您可以下载其文件并在 https://www.fviewer.com/view-epub 上查看，或将其安装在 Kindle 等上。

GPT-Author 运行原理

人工智能被要求根据给定的提示生成潜在绘图的列表。然后，它会选择最吸引人的情节，对其进行改进，并提取标题。之后，它会生成具有指定章节数的详细故事情节，然后尝试改进该故事情节。然后人工智能根据情节并考虑到前几章的内容单独编写每一章。最后，生成设计封面的提示，并创建封面。最后，将所有内容整合在一起，将小说编译成 EPUB 文件。

GPT-Author 用法

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

在 Google Colab 中，只需打开笔记本，添加 API 密钥，然后按顺序运行单元即可。

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

pip install openai ebooklib requests

在笔记本的最后一个单元格中，您可以自定义小说的提示和章节数。例如：

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)

这将根据给定的提示生成一本 20 章的小说。注意——少于 7 章的提示往往会导致问题。