Skip to content

尝试使用大模型进行应用开发来辅助小说写作

License

Notifications You must be signed in to change notification settings

IRandonation/AI-Novel

Repository files navigation

尝试使用AI辅助写网文

简介

尝试使用大模型用来对网文进行辅助写作

项目功能

大纲完善:

输入:用户提供的小说总体大纲或章节情节大纲。 输出:更加完善的大纲,包括情节结构、章节分布、人物发展等。 实现:利用大语言模型生成和完善大纲,可能包括分章节的建议。

记忆系:

人物记忆:追踪主要人物的状态(例如,性格特征、发展、经历)以及他们之间的关系。(尚未加入) 事件记忆:追踪小说中发生的关键事件,确保在后续章节中保持一致。 关系网络:维护人物之间的关系图,跟踪关系的变化,例如朋友、敌人、亲属等。(尚未加入)

章节扩充与润色:

输入:用户提供的每一章的主要大纲和情节要点。 输出:扩充并润色后的章节内容。 实现:使用大语言模型,根据大纲进行扩充,包括场景描写、对话、情感等细节的丰富。

开发日志

------ 2024.9.19 ------ 基本实现了通过章节大纲与段落大纲来进行扩写的功能,并可以对段落进行润色要求并进行润色。 新增了剧情记忆,可以总结以往的剧情进行记忆

------ 2024.9.21 ------ 新增了参考大纲的生成,帮助考虑小说内容 新增了小说剧情记忆功能,下次使用可以加载小说目前总结的剧情梗概帮助生成有逻辑的小说

下一步需要解决的问题

  1. 使用大模型的记忆问题,是否可以进行多开以及书本记忆存档 2024.9.19(finished)
  2. 根据用户提供的情节大纲来编写更细化的大纲进行参考,但是不参与文字的生成 2024.9.19(finished)
  3. 加入小说存档,这一次写完后可以将情节概述给保存下来,下一次可以输入使用 2024.9.21(finished)

快速上手指南

步骤1: 环境配置

首先,安装所需的依赖项:

pip install -r requirements.txt

步骤2: 实现LLM接口

项目依赖一个大语言模型。你需要实现LLM.py中的chatLLM函数。

步骤3: 运行项目

  • 直接运行demo.py,将自动创作小说,并将结果保存在novel_record.md文件中。

  • 运行app.py启动一个基于gradio的应用,通过打开显示的链接,你可以体验到AI小说生成的可视化过程。

About

尝试使用大模型进行应用开发来辅助小说写作

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages