📖 AI 小说全自动生成器 V2

项目概述

本项目是一个基于 Python Tkinter + requests 构建的 AI 小说全自动生成器（V2版）。支持 DeepSeek 和阿里云百炼（通义千问）双模型自由切换，采用多 Tab 页签分组配置参数，右侧实时流式显示 AI 生成的小说内容（打字机效果），所有配置参数可一键保存到本地 JSON 文件，下次启动自动加载。

相比 V1 版本，V2 的核心升级：

1. 左侧参数由单列滚动改为 5 个 Tab 页签（模型/写作/世界观/人物/任务），分组清晰不拥挤
2. AI 返回内容采用 SSE 流式读取，逐字追加到文本框，实时可见生成进度
3. 所有配置 持久化保存：保存为 novel_config.json，启动时自动加载
4. 支持 停止生成：中途可中断，已生成内容保留
5. 生成前自动静默保存配置，防止意外丢失

项目仅依赖 requests 一个第三方库（标准 HTTP 请求），Tkinter 为 Python 内置 GUI，无需额外安装。适合网文作者、AI 创作爱好者作为辅助写作工具。

技术架构

组件	技术	说明
GUI	Tkinter + ttk.Notebook	多Tab页签布局
AI接口	requests (stream=True)	SSE流式读取
配置持久化	JSON 文件	自动保存/加载
多线程	threading.Thread	后台生成不阻塞UI
支持模型	DeepSeek / 通义千问	OpenAI兼容接口

功能特性

功能	说明
🤖 双模型	DeepSeek / 阿里云百炼一键切换
📝 完整设定	书名/章节/字数/风格/温度
🌍 世界观	自定义世界设定和故事大纲
👤 人物	主角7维设定 + 配角自由编写
🎯 任务	本章核心剧情 + 创作禁忌
⚡ 流式输出	逐字显示，打字机效果
⏹ 可停止	随时中断，保留已生成内容
💾 配置保存	一键保存JSON，启动自动加载
📊 字数统计	实时更新当前字数
📄 保存TXT	时间戳命名，带书名章节

Tab 页签说明

Tab	内容
🤖 模型	AI选择、API Key、Model名
📝 写作	书名、章节、字数、风格、温度、语言
🌍 世界	世界观设定、故事大纲
👤 人物	主角7项设定、配角描述
🎯 任务	剧情禁忌、本章核心任务

快速开始

pip install requestspython 0721/novel_generator_v2.py

使用步骤

1. 在「🤖 模型」Tab 填入 API Key，选择模型
2. 在「📝 写作」Tab 设置书名、章节、字数
3. 在「🌍 世界」「👤 人物」「🎯 任务」Tab 完善设定
4. 点击左下方「💾 保存所有配置」
5. 点击右侧「🚀 开始生成」
6. 观看流式输出，生成完毕后点「💾 保存」导出TXT

配置文件

保存路径：同目录下 novel_config.json

• 点击「💾 保存所有配置」按钮手动保存
• 每次生成前自动静默保存
• 下次启动自动加载

项目结构

0721/├── novel_generator_v2.py   # V2 主程序├── novel_config.json       # 配置文件（自动生成）└── README_NOVEL_V2.md      # 本文档

---

完整源码

novel_generator_v2.py

"""AI 小说全自动生成器 V2 - 多Tab配置 + 流式显示双模型支持：DeepSeek / 阿里云百炼（通义千问）依赖：pip install requests"""import tkinter as tkfrom tkinter import ttk, messagebox, filedialog, scrolledtextimport requestsimport jsonimport osimport reimport timefrom datetime import datetimefrom threading import ThreadCONFIG_FILE = os.path.join(os.path.dirname(os.path.abspath(__file__)), "novel_config.json")CONFIG = {"ai_model": "deepseek","deepseek_api_key": "sk-your-deepseek-key","deepseek_model": "deepseek-chat","deepseek_endpoint": "https://api.deepseek.com/v1/chat/completions","aliyun_api_key": "sk-your-aliyun-key","aliyun_model": "qwen-plus","aliyun_endpoint": "https://dashscope.aliyuncs.com/compatible-mode/v1/chat/completions","book_name": "凡人修仙传","chapter_title": "第一章 初入仙途","chapter_intro": "少年韩立因机缘巧合踏上修仙之路","target_words": 3000,"language": "中文","style": "玄幻网文","temperature": 0.85,"world_setting": "修仙世界，境界：炼气、筑基、结丹、元婴、化神。门派林立，弱肉强食。","story_outline": "韩立被七玄门收为弟子，偶得神秘小瓶，由此改变命运。","protagonist": {"name":"韩立","appearance":"相貌普通，眼神沉稳","personality":"谨慎多疑、冷静务实","ability":"神秘小瓶催熟灵药","weakness":"灵根资质极差","identity":"七玄门记名弟子","catchphrase":"小心无大错"},"supporting_characters": "墨大夫：心狠手辣。\n厉飞雨：天资聪颖。","forbidden": "不可崩人设\n不可出现现代词汇\n不可无脑升级","chapter_mission": "描写韩立初入七玄门，环境描写+心理活动",}defload_saved_config():if os.path.exists(CONFIG_FILE):try:with open(CONFIG_FILE,"r",encoding="utf-8") as f:                saved=json.load(f)                merged=dict(CONFIG); merged.update(saved)if"protagonist"in saved and isinstance(saved["protagonist"],dict):                    merged["protagonist"]=dict(CONFIG["protagonist"]); merged["protagonist"].update(saved["protagonist"])return mergedexcept: passreturn dict(CONFIG)defsave_config_to_file(cfg):with open(CONFIG_FILE,"w",encoding="utf-8") as f:        json.dump(cfg,f,ensure_ascii=False,indent=2)return CONFIG_FILEdefbuild_prompt(cfg):    pro=cfg["protagonist"]returnf"""你是顶级{cfg['style']}作家。严格按设定撰写小说正文。【输出要求】只输出纯正文，无解释/markdown，字数{cfg['target_words']}字浮动10%，{cfg['language']}撰写【小说】{cfg['book_name']} - {cfg['chapter_title']}简介：{cfg['chapter_intro']}【世界观】{cfg['world_setting']}【大纲】{cfg['story_outline']}【主角】{pro['name']} 外貌:{pro['appearance']} 性格:{pro['personality']} 能力:{pro['ability']} 弱点:{pro['weakness']} 身份:{pro['identity']} 口头禅:{pro['catchphrase']}【配角】{cfg['supporting_characters']}【禁忌】{cfg['forbidden']}【本章任务】{cfg['chapter_mission']}开始撰写："""defstream_generate(cfg, on_chunk, on_done, on_error, stop_flag):    prompt=build_prompt(cfg)if cfg["ai_model"]=="deepseek":        url=cfg["deepseek_endpoint"]        headers={"Authorization":f"Bearer {cfg['deepseek_api_key']}","Content-Type":"application/json"}        payload={"model":cfg["deepseek_model"],"messages":[{"role":"system","content":"你是专业小说作家，只输出正文。"},{"role":"user","content":prompt}],"temperature":cfg["temperature"],"max_tokens":8000,"stream":True}else:        url=cfg["aliyun_endpoint"]        headers={"Authorization":f"Bearer {cfg['aliyun_api_key']}","Content-Type":"application/json"}        payload={"model":cfg["aliyun_model"],"messages":[{"role":"system","content":"你是专业小说作家，只输出正文。"},{"role":"user","content":prompt}],"temperature":cfg["temperature"],"max_tokens":8000,"stream":True}try:        resp=requests.post(url,headers=headers,json=payload,stream=True,timeout=120)        resp.raise_for_status()        full_text=""for line in resp.iter_lines(decode_unicode=True):if stop_flag[0]: on_done(full_text,stopped=True); returnifnot line ornot line.startswith("data: "): continue            data_str=line[6:]if data_str.strip()=="[DONE]": breaktry:                data=json.loads(data_str)                content=data.get("choices",[{}])[0].get("delta",{}).get("content","")if content: full_text+=content; on_chunk(content)except json.JSONDecodeError: continue        on_done(full_text,stopped=False)except requests.exceptions.HTTPError as e:        s=e.response.status_code if e.response else0        on_error("❌ API密钥错误"if s==401elsef"❌ HTTP {s}")except requests.exceptions.Timeout: on_error("⏰ 超时")except requests.exceptions.ConnectionError: on_error("❌ 网络失败")except Exception as e: on_error(f"❌ {e}")classNovelV2GUI:def__init__(self):        self.root=tk.Tk(); self.root.title("📖 AI小说生成器 V2")        self.root.geometry("1020x680"); self.root.resizable(True,True); self.root.configure(bg="#fafafa")        self.generating=False; self.stop_flag=[False]; self.full_text=""        self.saved_config=load_saved_config()        self.build_ui()defbuild_ui(self):        h=tk.Frame(self.root,bg="#37474f",height=38); h.pack(fill="x"); h.pack_propagate(False)        tk.Label(h,text="📖 AI小说生成器 V2",font=("Microsoft YaHei",12,"bold"),bg="#37474f",fg="#fff").pack(side="left",padx=12,pady=4)        tk.Label(h,text="流式输出 · 多Tab配置 · 配置持久化",font=("Microsoft YaHei",8),bg="#37474f",fg="#90a4ae").pack(side="right",padx=12)        body=tk.Frame(self.root,bg="#fafafa"); body.pack(fill="both",expand=True,padx=8,pady=6)        left=tk.Frame(body,bg="#fff",width=420,highlightthickness=1,highlightbackground="#ddd")        left.pack(side="left",fill="y",padx=(0,6)); left.pack_propagate(False)        self.notebook=ttk.Notebook(left); self.notebook.pack(fill="both",expand=True,padx=4,pady=4)        self.build_tabs()        sf=tk.Frame(left,bg="#fff"); sf.pack(fill="x",padx=6,pady=(0,6))        tk.Button(sf,text="💾 保存所有配置",font=("Microsoft YaHei",9,"bold"),bg="#43a047",fg="#fff",relief="flat",padx=12,pady=3,command=self.do_save_config).pack(fill="x")        self.config_status=tk.Label(sf,text="",bg="#fff",fg="#43a047",font=("Microsoft YaHei",7)); self.config_status.pack(fill="x")        right=tk.Frame(body,bg="#fff",highlightthickness=1,highlightbackground="#ddd"); right.pack(side="left",fill="both",expand=True)        tb=tk.Frame(right,bg="#fff"); tb.pack(fill="x",padx=8,pady=(6,2))        tk.Label(tb,text="📝 生成内容",font=("Microsoft YaHei",9,"bold"),bg="#fff",fg="#333").pack(side="left")        self.word_count_label=tk.Label(tb,text="0 字",font=("Microsoft YaHei",8),bg="#fff",fg="#888"); self.word_count_label.pack(side="right")        self.output_text=scrolledtext.ScrolledText(right,font=("Microsoft YaHei",10),bg="#fefefe",fg="#222",wrap="word",spacing1=2,spacing2=4,spacing3=2,insertbackground="#1565c0",relief="flat",bd=0)        self.output_text.pack(fill="both",expand=True,padx=8,pady=4)        bf=tk.Frame(right,bg="#fff"); bf.pack(fill="x",padx=8,pady=6)        self.btn_gen=tk.Button(bf,text="🚀 开始生成",font=("Microsoft YaHei",10,"bold"),bg="#1565c0",fg="#fff",relief="flat",padx=16,pady=4,command=self.do_generate); self.btn_gen.pack(side="left",padx=(0,6))        self.btn_stop=tk.Button(bf,text="⏹ 停止",font=("Microsoft YaHei",9),bg="#e53935",fg="#fff",relief="flat",padx=12,state="disabled",command=self.do_stop); self.btn_stop.pack(side="left",padx=(0,6))        tk.Button(bf,text="🗑️ 清空",font=("Microsoft YaHei",9),bg="#e0e0e0",fg="#333",relief="flat",padx=10,command=self.clear_output).pack(side="left",padx=(0,6))        tk.Button(bf,text="💾 保存",font=("Microsoft YaHei",9),bg="#ff9800",fg="#fff",relief="flat",padx=10,command=self.save_file).pack(side="left")defbuild_tabs(self):        sc=self.saved_config        t1=tk.Frame(self.notebook,bg="#fff"); self.notebook.add(t1,text=" 🤖 模型 ")        self._sec(t1,"AI模型"); self.model_var=tk.StringVar(value=sc["ai_model"])        mf=tk.Frame(t1,bg="#fff");mf.pack(fill="x",padx=10,pady=4)        tk.Radiobutton(mf,text="DeepSeek",variable=self.model_var,value="deepseek",bg="#fff").pack(side="left",padx=6)        tk.Radiobutton(mf,text="阿里云百炼",variable=self.model_var,value="aliyun",bg="#fff").pack(side="left")        self._sec(t1,"DeepSeek"); self.inp_ds_key=self._entry(t1,"API Key",sc["deepseek_api_key"],show="*"); self.inp_ds_model=self._entry(t1,"Model",sc["deepseek_model"])        self._sec(t1,"阿里云百炼"); self.inp_al_key=self._entry(t1,"API Key",sc["aliyun_api_key"],show="*"); self.inp_al_model=self._entry(t1,"Model",sc["aliyun_model"])        t2=tk.Frame(self.notebook,bg="#fff"); self.notebook.add(t2,text=" 📝 写作 ")        self.inp_book=self._entry(t2,"书名",sc["book_name"]); self.inp_chapter=self._entry(t2,"章节",sc["chapter_title"])        self.inp_intro=self._entry(t2,"简介",sc["chapter_intro"]); self.inp_style=self._entry(t2,"风格",sc["style"])        wf=tk.Frame(t2,bg="#fff");wf.pack(fill="x",padx=10,pady=4)        tk.Label(wf,text="字数:",bg="#fff",fg="#555",font=("Microsoft YaHei",9)).pack(side="left")        self.words_var=tk.IntVar(value=sc["target_words"]); tk.Spinbox(wf,from_=500,to=8000,textvariable=self.words_var,width=7,font=("Consolas",9),increment=500).pack(side="left",padx=4)        tk.Label(wf,text="温度:",bg="#fff",fg="#555",font=("Microsoft YaHei",9)).pack(side="left",padx=(12,0))        self.temp_var=tk.DoubleVar(value=sc["temperature"]); tk.Spinbox(wf,from_=0.1,to=1.5,textvariable=self.temp_var,width=5,font=("Consolas",9),increment=0.05,format="%.2f").pack(side="left",padx=4)        tk.Label(t2,text="语言:",bg="#fff",fg="#555",font=("Microsoft YaHei",9)).pack(fill="x",padx=10,pady=(6,0))        self.lang_var=tk.StringVar(value=sc["language"]); lf=tk.Frame(t2,bg="#fff"); lf.pack(fill="x",padx=10)for l in ["中文","繁体","English"]: tk.Radiobutton(lf,text=l,variable=self.lang_var,value=l,bg="#fff").pack(side="left",padx=4)        t3=tk.Frame(self.notebook,bg="#fff"); self.notebook.add(t3,text=" 🌍 世界 ")        tk.Label(t3,text="世界观:",bg="#fff",fg="#333",font=("Microsoft YaHei",9,"bold")).pack(fill="x",padx=10,pady=(8,2))        self.txt_world=tk.Text(t3,font=("Microsoft YaHei",9),height=6,wrap="word",bg="#f8f9fa",relief="solid",bd=1); self.txt_world.pack(fill="x",padx=10,pady=2); self.txt_world.insert("1.0",sc["world_setting"])        tk.Label(t3,text="大纲:",bg="#fff",fg="#333",font=("Microsoft YaHei",9,"bold")).pack(fill="x",padx=10,pady=(8,2))        self.txt_outline=tk.Text(t3,font=("Microsoft YaHei",9),height=6,wrap="word",bg="#f8f9fa",relief="solid",bd=1); self.txt_outline.pack(fill="x",padx=10,pady=2); self.txt_outline.insert("1.0",sc["story_outline"])        t4=tk.Frame(self.notebook,bg="#fff"); self.notebook.add(t4,text=" 👤 人物 ")        pro=sc["protagonist"]; self._sec(t4,"主角")        self.inp_pro_name=self._entry(t4,"姓名",pro["name"]); self.inp_pro_look=self._entry(t4,"外貌",pro["appearance"])        self.inp_pro_char=self._entry(t4,"性格",pro["personality"]); self.inp_pro_ability=self._entry(t4,"能力",pro["ability"])        self.inp_pro_weak=self._entry(t4,"弱点",pro["weakness"]); self.inp_pro_id=self._entry(t4,"身份",pro["identity"])        self.inp_pro_say=self._entry(t4,"口头禅",pro["catchphrase"])        self._sec(t4,"配角"); self.txt_support=tk.Text(t4,font=("Microsoft YaHei",9),height=4,wrap="word",bg="#f8f9fa",relief="solid",bd=1); self.txt_support.pack(fill="x",padx=10,pady=2); self.txt_support.insert("1.0",sc["supporting_characters"])        t5=tk.Frame(self.notebook,bg="#fff"); self.notebook.add(t5,text=" 🎯 任务 ")        tk.Label(t5,text="禁忌:",bg="#fff",fg="#333",font=("Microsoft YaHei",9,"bold")).pack(fill="x",padx=10,pady=(8,2))        self.txt_forbidden=tk.Text(t5,font=("Microsoft YaHei",9),height=5,wrap="word",bg="#f8f9fa",relief="solid",bd=1); self.txt_forbidden.pack(fill="x",padx=10,pady=2); self.txt_forbidden.insert("1.0",sc["forbidden"])        tk.Label(t5,text="本章任务:",bg="#fff",fg="#333",font=("Microsoft YaHei",9,"bold")).pack(fill="x",padx=10,pady=(8,2))        self.inp_mission=self._entry(t5,"",sc["chapter_mission"])def_sec(self,p,t): tk.Label(p,text=t,bg="#fff",fg="#1565c0",font=("Microsoft YaHei",9,"bold")).pack(fill="x",padx=10,pady=(10,2))def_entry(self,p,label,default,show=None):        f=tk.Frame(p,bg="#fff");f.pack(fill="x",padx=10,pady=2)if label: tk.Label(f,text=label+":",bg="#fff",fg="#555",font=("Microsoft YaHei",8),width=7,anchor="e").pack(side="left")        e=tk.Entry(f,font=("Microsoft YaHei",9),show=show);e.pack(side="left",fill="x",expand=True,padx=2);e.insert(0,default);return edefdo_save_config(self,silent=False):        cfg=self.collect_config(); filepath=save_config_to_file(cfg)        self.config_status.config(text=f"✅ 已保存 ({datetime.now().strftime('%H:%M:%S')})")ifnot silent: messagebox.showinfo("保存成功",f"配置已保存到:\n{filepath}")defcollect_config(self):        cfg=dict(CONFIG); cfg["ai_model"]=self.model_var.get()        cfg["deepseek_api_key"]=self.inp_ds_key.get().strip(); cfg["deepseek_model"]=self.inp_ds_model.get().strip()        cfg["aliyun_api_key"]=self.inp_al_key.get().strip(); cfg["aliyun_model"]=self.inp_al_model.get().strip()        cfg["book_name"]=self.inp_book.get().strip(); cfg["chapter_title"]=self.inp_chapter.get().strip()        cfg["chapter_intro"]=self.inp_intro.get().strip(); cfg["style"]=self.inp_style.get().strip()        cfg["target_words"]=self.words_var.get(); cfg["temperature"]=self.temp_var.get(); cfg["language"]=self.lang_var.get()        cfg["world_setting"]=self.txt_world.get("1.0","end").strip(); cfg["story_outline"]=self.txt_outline.get("1.0","end").strip()        cfg["protagonist"]={"name":self.inp_pro_name.get().strip(),"appearance":self.inp_pro_look.get().strip(),"personality":self.inp_pro_char.get().strip(),"ability":self.inp_pro_ability.get().strip(),"weakness":self.inp_pro_weak.get().strip(),"identity":self.inp_pro_id.get().strip(),"catchphrase":self.inp_pro_say.get().strip()}        cfg["supporting_characters"]=self.txt_support.get("1.0","end").strip()        cfg["forbidden"]=self.txt_forbidden.get("1.0","end").strip(); cfg["chapter_mission"]=self.inp_mission.get().strip()return cfgdefdo_generate(self):        self.do_save_config(silent=True)if self.generating: return        self.generating=True; self.stop_flag[0]=False; self.full_text=""        self.btn_gen.config(state="disabled"); self.btn_stop.config(state="normal")        self.output_text.delete("1.0","end"); self.word_count_label.config(text="生成中...")        cfg=self.collect_config()defon_chunk(t): self.root.after(0,self._append,t)defon_done(f,stopped=False): self.full_text=f; self.root.after(0,self._done,stopped)defon_error(m): self.root.after(0,self._err,m)        Thread(target=stream_generate,args=(cfg,on_chunk,on_done,on_error,self.stop_flag),daemon=True).start()defdo_stop(self): self.stop_flag[0]=Truedef_append(self,t):        self.output_text.insert("end",t); self.output_text.see("end")        self.word_count_label.config(text=f"{len(self.output_text.get('1.0','end').strip())} 字")def_done(self,stopped):        self.generating=False; self.btn_gen.config(state="normal"); self.btn_stop.config(state="disabled")        c=len(self.output_text.get("1.0","end").strip())        self.word_count_label.config(text=f"{'（停止）'if stopped else'✅'}{c} 字")def_err(self,m):        self.generating=False; self.btn_gen.config(state="normal"); self.btn_stop.config(state="disabled")        self.output_text.insert("end",f"\n\n{m}"); self.word_count_label.config(text="错误")defclear_output(self): self.output_text.delete("1.0","end"); self.word_count_label.config(text="0 字"); self.full_text=""defsave_file(self):        content=self.output_text.get("1.0","end").strip()ifnot content: messagebox.showinfo("提示","无内容"); return        cfg=self.collect_config(); ts=datetime.now().strftime("%Y%m%d%H%M%S")        ch=cfg["chapter_title"].replace(" ",""); nm=f"{ts}_{cfg['book_name']}_{ch}.txt"        path=filedialog.asksaveasfilename(title="保存",defaultextension=".txt",initialfile=nm,filetypes=[("文本","*.txt")])if path:with open(path,"w",encoding="utf-8") as f: f.write(f"《{cfg['book_name']}》\n{cfg['chapter_title']}\n{'='*40}\n\n{content}")            messagebox.showinfo("成功",f"已保存:\n{path}\n字数:{len(content)}")defrun(self): self.root.mainloop()if __name__=="__main__": NovelV2GUI().run()

许可证

MIT License - 可自由使用、修改和分发。