init

.env

@@ -0,0 +1,11 @@
+# OpenAI 接口代理地址，（可选配置）
+OPENAI_API_BASE="https://api.openai-hk.com/v1"
+
+# * 配置你的 OpenAI key（建议使用GPT-4正式账号，临时账号有每分钟3次请求限制）
+OPENAI_API_KEY="hk-218wghmq4zcf4x42crjof0zlrpl4p22r96d44ola4wmyw92o"
+
+# * 生成小说封面，具体可访问 https://platform.stability.ai/ 地址申请key
+STABILITY_API_KEY="sk-iWYWb4PNfkIAv193eRuAoHBkk22FzkWzFSap62YydnUFKEcl"
+
+# 配置你的 Claude2 API Key（可选配置）
+ANTHROPIC_API_KEY="your-anthropic-api-key"

.env.example

@@ -0,0 +1,11 @@
+# OpenAI 接口代理地址，（可选配置）
+OPENAI_API_BASE="your-openai-proxy-url"
+
+# * 配置你的 OpenAI key（建议使用GPT-4正式账号，临时账号有每分钟3次请求限制）
+OPENAI_API_KEY="your-openai-api-key"
+
+# * 生成小说封面，具体可访问 https://platform.stability.ai/ 地址申请key
+STABILITY_API_KEY="your-stablity-api-key"
+
+# 配置你的 Claude2 API Key（可选配置）
+ANTHROPIC_API_KEY="your-anthropic-api-key"

.gitignore

@@ -0,0 +1,93 @@
+.DS_Store
+git.properties
+*.userlibraries
+*.log.20*
+*.log
+log.*
+### Intellij ###
+# Covers JetBrains IDEs: IntelliJ, RubyMine, PhpStorm, AppCode, PyCharm
+
+*.iml
+*.eml
+
+## Directory-based project format:
+.idea/
+
+## File-based project format:
+*.ipr
+*.iws
+
+## Plugin-specific files:
+
+# IntelliJ
+/out/
+
+# mpeltonen/sbt-idea plugin
+.idea_modules/
+
+# JIRA plugin
+atlassian-ide-plugin.xml
+
+# Python
+__pycache__
+*.pyc
+
+# Crashlytics plugin (for Android Studio and IntelliJ)
+com_crashlytics_export_strings.xml
+crashlytics.properties
+crashlytics-build.properties
+
+# java build files
+target
+node_modules
+gulpfile.js
+
+# debug scripts.
+deploy.sh
+ftpsync.settings
+
+*.swp
+.DS_Store
+
+# Xcode
+build/*
+*.pbxuser
+!default.pbxuser
+*.mode1v3
+!default.mode1v3
+*.mode2v3
+!default.mode2v3
+*.perspectivev3
+!default.perspectivev3
+*.xcworkspace
+!default.xcworkspace
+xcuserdata
+profile
+*.moved-aside
+*.cer
+*.p12
+*.mobileprovision
+
+# VSCode
+.vscode
+# AppCode
+.idea
+
+# Tern plugin
+**/.tern-project
+
+#emacs temp file
+*~
+
+# log files
+batch_loader/hadoop_mapred*
+batch_loader/kafka_consumer*
+
+# Maven Repository
+.repository
+
+# UT generated files
+swagger.json
+swagger-model.json
+
+/dependencies/

LICENSE

@@ -0,0 +1,201 @@
+                                 Apache License
+                           Version 2.0, January 2004
+                        http://www.apache.org/licenses/
+
+   TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
+
+   1. Definitions.
+
+      "License" shall mean the terms and conditions for use, reproduction,
+      and distribution as defined by Sections 1 through 9 of this document.
+
+      "Licensor" shall mean the copyright owner or entity authorized by
+      the copyright owner that is granting the License.
+
+      "Legal Entity" shall mean the union of the acting entity and all
+      other entities that control, are controlled by, or are under common
+      control with that entity. For the purposes of this definition,
+      "control" means (i) the power, direct or indirect, to cause the
+      direction or management of such entity, whether by contract or
+      otherwise, or (ii) ownership of fifty percent (50%) or more of the
+      outstanding shares, or (iii) beneficial ownership of such entity.
+
+      "You" (or "Your") shall mean an individual or Legal Entity
+      exercising permissions granted by this License.
+
+      "Source" form shall mean the preferred form for making modifications,
+      including but not limited to software source code, documentation
+      source, and configuration files.
+
+      "Object" form shall mean any form resulting from mechanical
+      transformation or translation of a Source form, including but
+      not limited to compiled object code, generated documentation,
+      and conversions to other media types.
+
+      "Work" shall mean the work of authorship, whether in Source or
+      Object form, made available under the License, as indicated by a
+      copyright notice that is included in or attached to the work
+      (an example is provided in the Appendix below).
+
+      "Derivative Works" shall mean any work, whether in Source or Object
+      form, that is based on (or derived from) the Work and for which the
+      editorial revisions, annotations, elaborations, or other modifications
+      represent, as a whole, an original work of authorship. For the purposes
+      of this License, Derivative Works shall not include works that remain
+      separable from, or merely link (or bind by name) to the interfaces of,
+      the Work and Derivative Works thereof.
+
+      "Contribution" shall mean any work of authorship, including
+      the original version of the Work and any modifications or additions
+      to that Work or Derivative Works thereof, that is intentionally
+      submitted to Licensor for inclusion in the Work by the copyright owner
+      or by an individual or Legal Entity authorized to submit on behalf of
+      the copyright owner. For the purposes of this definition, "submitted"
+      means any form of electronic, verbal, or written communication sent
+      to the Licensor or its representatives, including but not limited to
+      communication on electronic mailing lists, source code control systems,
+      and issue tracking systems that are managed by, or on behalf of, the
+      Licensor for the purpose of discussing and improving the Work, but
+      excluding communication that is conspicuously marked or otherwise
+      designated in writing by the copyright owner as "Not a Contribution."
+
+      "Contributor" shall mean Licensor and any individual or Legal Entity
+      on behalf of whom a Contribution has been received by Licensor and
+      subsequently incorporated within the Work.
+
+   2. Grant of Copyright License. Subject to the terms and conditions of
+      this License, each Contributor hereby grants to You a perpetual,
+      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+      copyright license to reproduce, prepare Derivative Works of,
+      publicly display, publicly perform, sublicense, and distribute the
+      Work and such Derivative Works in Source or Object form.
+
+   3. Grant of Patent License. Subject to the terms and conditions of
+      this License, each Contributor hereby grants to You a perpetual,
+      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+      (except as stated in this section) patent license to make, have made,
+      use, offer to sell, sell, import, and otherwise transfer the Work,
+      where such license applies only to those patent claims licensable
+      by such Contributor that are necessarily infringed by their
+      Contribution(s) alone or by combination of their Contribution(s)
+      with the Work to which such Contribution(s) was submitted. If You
+      institute patent litigation against any entity (including a
+      cross-claim or counterclaim in a lawsuit) alleging that the Work
+      or a Contribution incorporated within the Work constitutes direct
+      or contributory patent infringement, then any patent licenses
+      granted to You under this License for that Work shall terminate
+      as of the date such litigation is filed.
+
+   4. Redistribution. You may reproduce and distribute copies of the
+      Work or Derivative Works thereof in any medium, with or without
+      modifications, and in Source or Object form, provided that You
+      meet the following conditions:
+
+      (a) You must give any other recipients of the Work or
+          Derivative Works a copy of this License; and
+
+      (b) You must cause any modified files to carry prominent notices
+          stating that You changed the files; and
+
+      (c) You must retain, in the Source form of any Derivative Works
+          that You distribute, all copyright, patent, trademark, and
+          attribution notices from the Source form of the Work,
+          excluding those notices that do not pertain to any part of
+          the Derivative Works; and
+
+      (d) If the Work includes a "NOTICE" text file as part of its
+          distribution, then any Derivative Works that You distribute must
+          include a readable copy of the attribution notices contained
+          within such NOTICE file, excluding those notices that do not
+          pertain to any part of the Derivative Works, in at least one
+          of the following places: within a NOTICE text file distributed
+          as part of the Derivative Works; within the Source form or
+          documentation, if provided along with the Derivative Works; or,
+          within a display generated by the Derivative Works, if and
+          wherever such third-party notices normally appear. The contents
+          of the NOTICE file are for informational purposes only and
+          do not modify the License. You may add Your own attribution
+          notices within Derivative Works that You distribute, alongside
+          or as an addendum to the NOTICE text from the Work, provided
+          that such additional attribution notices cannot be construed
+          as modifying the License.
+
+      You may add Your own copyright statement to Your modifications and
+      may provide additional or different license terms and conditions
+      for use, reproduction, or distribution of Your modifications, or
+      for any such Derivative Works as a whole, provided Your use,
+      reproduction, and distribution of the Work otherwise complies with
+      the conditions stated in this License.
+
+   5. Submission of Contributions. Unless You explicitly state otherwise,
+      any Contribution intentionally submitted for inclusion in the Work
+      by You to the Licensor shall be under the terms and conditions of
+      this License, without any additional terms or conditions.
+      Notwithstanding the above, nothing herein shall supersede or modify
+      the terms of any separate license agreement you may have executed
+      with Licensor regarding such Contributions.
+
+   6. Trademarks. This License does not grant permission to use the trade
+      names, trademarks, service marks, or product names of the Licensor,
+      except as required for reasonable and customary use in describing the
+      origin of the Work and reproducing the content of the NOTICE file.
+
+   7. Disclaimer of Warranty. Unless required by applicable law or
+      agreed to in writing, Licensor provides the Work (and each
+      Contributor provides its Contributions) on an "AS IS" BASIS,
+      WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
+      implied, including, without limitation, any warranties or conditions
+      of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
+      PARTICULAR PURPOSE. You are solely responsible for determining the
+      appropriateness of using or redistributing the Work and assume any
+      risks associated with Your exercise of permissions under this License.
+
+   8. Limitation of Liability. In no event and under no legal theory,
+      whether in tort (including negligence), contract, or otherwise,
+      unless required by applicable law (such as deliberate and grossly
+      negligent acts) or agreed to in writing, shall any Contributor be
+      liable to You for damages, including any direct, indirect, special,
+      incidental, or consequential damages of any character arising as a
+      result of this License or out of the use or inability to use the
+      Work (including but not limited to damages for loss of goodwill,
+      work stoppage, computer failure or malfunction, or any and all
+      other commercial damages or losses), even if such Contributor
+      has been advised of the possibility of such damages.
+
+   9. Accepting Warranty or Additional Liability. While redistributing
+      the Work or Derivative Works thereof, You may choose to offer,
+      and charge a fee for, acceptance of support, warranty, indemnity,
+      or other liability obligations and/or rights consistent with this
+      License. However, in accepting such obligations, You may act only
+      on Your own behalf and on Your sole responsibility, not on behalf
+      of any other Contributor, and only if You agree to indemnify,
+      defend, and hold each Contributor harmless for any liability
+      incurred by, or claims asserted against, such Contributor by reason
+      of your accepting any such warranty or additional liability.
+
+   END OF TERMS AND CONDITIONS
+
+   APPENDIX: How to apply the Apache License to your work.
+
+      To apply the Apache License to your work, attach the following
+      boilerplate notice, with the fields enclosed by brackets "[]"
+      replaced with your own identifying information. (Don't include
+      the brackets!)  The text should be enclosed in the appropriate
+      comment syntax for the file format. We also recommend that a
+      file or class name and description of purpose be included on the
+      same "printed page" as the copyright notice for easier
+      identification within third-party archives.
+
+   Copyright [yyyy] [name of copyright owner]
+
+   Licensed under the Apache License, Version 2.0 (the "License");
+   you may not use this file except in compliance with the License.
+   You may obtain a copy of the License at
+
+       http://www.apache.org/licenses/LICENSE-2.0
+
+   Unless required by applicable law or agreed to in writing, software
+   distributed under the License is distributed on an "AS IS" BASIS,
+   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+   See the License for the specific language governing permissions and
+   limitations under the License.

README.md

@@ -1,13 +1,87 @@
----
-title: Test
-emoji: 📊
-colorFrom: green
-colorTo: red
-sdk: gradio
-sdk_version: 4.0.2
-app_file: app.py
-pinned: false
-license: apache-2.0
----
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+<h1 align="center">● gpt-story-genius</h1>
+
+<p align="center">
+    <br>
+    <b>StoryGenius：一款AI自动创作小说工具。</b><br><br>
+    根据小说的提示词、写作风格和章节数量几分钟即可快速生成奇幻小说。并自动打包为电子书格式。<br>
+</p>
+<br>
+
+![poster](https://github.com/Crossme0809/frenzy_repo/blob/main/assets/story_genius.png)
+<br>
+
+
+**gpt-story-genius** 是一个自动创作小说的AI，它可以在几分钟内根据用户提供的初始提示和章节数生成一整本奇幻小说，并自动打包为电子书格式。
+该项目利用 **GPT-4**、**Stable Diffusion API** 和 **Anthropic API** 等一系列大模型调用组成的链来生成原创奇幻小说。<br>
+
+此外，它还可以根据这本书创建一个原创封面，并将整本作品一次性转换为PDF或电子书格式，并且`制作成本低廉，制作一本15章的小说仅需4美元成本`，并且该工具是开源的，可以免费使用。
+<br>
+
+## 快速使用
+
+在 Google Colab 中，只需打开笔记本，添加 API 密钥，然后按顺序运行单元即可。 </br></br>
+[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/Crossme0809/frenzyTechAI/blob/main/gpt-author/gpt_author_v2.ipynb)
+
+在笔记本的最后一个单元格中，您可以自定义小说的提示和章节数。例如：
+
+```python
+prompt = "一个被遗忘的小岛，上面有一座古老的灯塔。当灯塔亮起时，岛上的生物就会发生奇异的变化。"
+num_chapters = 10
+writing_style = "紧张刺激，类似于青少年恐怖小说。有很多对话和内心独白"
+novel, title, chapters, chapter_titles = write_fantasy_novel(prompt, num_chapters, writing_style)
+```
+
+这将根据给定的提示生成一本 10 章的小说。**注意——少于 7 章的提示往往会导致问题。**
+
+## 本地部署
+
+- 下载项目代码
+```bash
+git clone https://github.com/Crossme0809/gpt-story-genius.git
+```
+
+- 复制项目配置文件
+```bash
+cp .env.example .env
+```
+
+- 配置GPT Key
+```bash
+# OpenAI 接口代理地址，（可选配置）
+OPENAI_API_BASE="your-openai-proxy-url"
+
+# * 配置你的 OpenAI key（建议使用GPT-4正式账号，临时账号有每分钟3次请求限制）
+OPENAI_API_KEY="your-openai-api-key"
+
+# * 生成小说封面，具体可访问 https://platform.stability.ai/ 地址申请key
+STABILITY_API_KEY="your-stablity-api-key"
+sk-iWYWb4PNfkIAv193eRuAoHBkk22FzkWzFSap62YydnUFKEcl
+# 配置你的 Claude2 API Key（可选配置）
+ANTHROPIC_API_KEY="your-anthropic-api-key"
+```
+
+- 安装项目依赖
+```bash
+pip install -r requirements.txt
+```
+
+- 启动项目
+```bash
+gradio app.py
+```
+
+- 后台进程启动
+```bash
+nohup gradio app.py > log.txt 2>&1 &
+```
+
+启动成功后，访问8000端口即可打开 **StoryGenius** 项目主页，如需要修改端口，只需要编辑 `run.py` 最后一行中的 `server_port` 即可。
+<br><br>
+<p>生成完的小说Epub文可以下载其文件并在 https://www.fviewer.com/view-epub 上查看，或将其安装在 Kindle 等上。（Mac上直接预览）</p>
+
+![poster](https://github.com/Crossme0809/frenzy_repo/blob/main/assets/novel_epub.png)
+
+## 特别说明
+
+本项目基于[gpt-author](https://github.com/mshumer/gpt-author)开发，由[@mattshumer_](https://twitter.com/mattshumer_)创建。
+

app.py

@@ -0,0 +1,76 @@
+# coding: utf8
+import gradio as gr
+from write_story import write_fantasy_novel
+from author import create_cover_image
+from author import create_epub
+from config import anthropic_api_key
+
+if anthropic_api_key and anthropic_api_key != "your-anthropic-api-key":
+    print(">>>>>> ========= use anthropic ========")
+    claude_true = True
+else:
+    print(">>>>>> ========= claude_true = False ========")
+    claude_true = False
+
+
+def generate_novel(prompt, num_chapters, writing_style, model_name):
+    # 调用GPT和Claude API，生成小说结果
+    # prompt = "A kingdom hidden deep in the forest, where every tree is a portal to another world."
+    # num_chapters = 2
+    # writing_style = "Clear and easily understandable, similar to a young adult novel. Lots of dialogue."
+    # model_name = "gpt-3.5-turbo-16k"
+    if not prompt or not writing_style:
+        raise gr.Error("提示词和写作风格是必填项")
+    if num_chapters < 1:
+        raise gr.Error("章节数必须大于等于1")
+
+    num_chapters = int(num_chapters)
+    novel, title, chapters, chapter_titles = write_fantasy_novel(prompt,
+                                                                 num_chapters, writing_style, claude_true, model_name)
+
+    # 用chapter_titles中的正文取代章节说明
+    for i, chapter in enumerate(chapters):
+        chapter_number_and_title = list(chapter_titles[i].keys())[0]
+        chapter_titles[i] = {chapter_number_and_title: chapter}
+
+    # 生成小说的封面
+    image_url = create_cover_image(str(chapter_titles))
+    print(f"Image URL: {image_url}")
+
+    # 生成小说 EPUB 文件
+    file_url = create_epub(title, 'AI', chapter_titles, image_url)
+    print(f"Novel URL: {file_url}")
+
+    # novel, file_path = write_fantasy_novel(prompt, num_chapters, writing_style)
+    return { "image_url": image_url, "file_url": file_url }
+
+
+def generate_output(prompt, num_chapters, writing_style, model_name):
+    try:
+        output = generate_novel(prompt, num_chapters, writing_style, model_name)
+        print(output)
+        return (output["image_url"], output["file_url"])
+    except Exception as e:
+        raise gr.Error({str(e)})
+        return f"An error occurred: {str(e)}"
+
+
+inputs = [
+    gr.Textbox(value="一个被遗忘的小岛，上面有一座古老的灯塔。当灯塔亮起时，岛上的生物就会发生奇异的变化。", lines=2, placeholder="Usage：一个被遗忘的小岛，上面有一座古老的灯塔。当灯塔亮起时，岛上的生物就会发生奇异的变化。", label="小说提示词"),
+    gr.Number(value=1, label="小说章节数"),
+    gr.Textbox(value="紧张刺激，类似于青少年恐怖小说。有很多对话和内心独白", lines=2, placeholder="Usage：紧张刺激，类似于青少年恐怖小说。有很多对话和内心独白", label="AI写作风格"),
+    gr.Dropdown(["gpt-3.5-turbo-16k", "gpt-3.5-turbo", "gpt-4", "gpt-4-32k"], label="选择GPT模型", value="gpt-3.5-turbo")
+]
+
+
+outputs = [
+    gr.Image(label="封面图片", width=1028, height=300),
+    gr.File(label="EPUB文件")
+]
+
+title = "StoryGenius：一款AI自动创作小说工具"
+description = "根据小说的提示词、写作风格和章节数量几分钟即可快速生成奇幻小说。并自动打包为电子书格式。"
+
+
+iface = gr.Interface(fn=generate_output, inputs=inputs, outputs=outputs, title=title, description=description)
+iface.launch(server_name="127.0.0.1", server_port=8000)

author.py

@@ -0,0 +1,145 @@
+# coding: utf8
+from ebooklib import epub
+import base64
+import os
+import requests
+from config import completion_with_backoff
+from config import stability_api_key
+
+
+def generate_cover_prompt(plot):
+    response = completion_with_backoff(
+        model="gpt-3.5-turbo-16k",
+        messages=[
+            {"role": "system",
+             "content": "You are a creative assistant that writes a spec for the cover art of a book, based on the book's plot."},
+            {"role": "user",
+             "content": f"Plot: {plot}\n\n--\n\nDescribe the cover we should create, based on the plot. This should be two sentences long, maximum."}
+        ]
+    )
+    return response['choices'][0]['message']['content']
+
+
+def create_cover_image(plot):
+    plot = str(generate_cover_prompt(plot))
+
+    engine_id = "stable-diffusion-xl-beta-v2-2-2"
+    api_host = os.getenv('API_HOST', 'https://api.stability.ai')
+    api_key = stability_api_key
+
+    if api_key is None:
+        raise Exception("Missing Stability API key.")
+
+    response = requests.post(
+        f"{api_host}/v1/generation/{engine_id}/text-to-image",
+        headers={
+            "Content-Type": "application/json",
+            "Accept": "application/json",
+            "Authorization": f"Bearer {api_key}"
+        },
+        json={
+            "text_prompts": [
+                {
+                    "text": plot
+                }
+            ],
+            "cfg_scale": 7,
+            "clip_guidance_preset": "FAST_BLUE",
+            "height": 768,
+            "width": 512,
+            "samples": 1,
+            "steps": 30,
+        },
+    )
+
+    if response.status_code != 200:
+        raise Exception("Non-200 response: " + str(response.text))
+
+    data = response.json()
+
+    # 检查目录是否存在，如果不存在则创建
+    directory = "./cover/"
+    if not os.path.exists(directory):
+        os.makedirs(directory)
+
+    for i, image in enumerate(data["artifacts"]):
+        image_bytes = base64.b64decode(image["base64"])
+        file_path = f"./cover/cover_{i}.png"  # 修改为您想要的文件路径
+        with open(file_path, "wb") as f:
+            f.write(image_bytes)
+
+        return file_path
+
+
+def create_epub(title, author, chapters, cover_image_path='cover.png'):
+    book = epub.EpubBook()
+
+    # Set metadata
+    book.set_identifier('id123456')
+    book.set_title(title)
+    book.set_language('zh-cn')
+    book.add_author(author)
+
+    # Add cover image
+    with open(cover_image_path, 'rb') as cover_file:
+        cover_image = cover_file.read()
+    book.set_cover('cover.png', cover_image)
+
+    # Create chapters and add them to the book
+    epub_chapters = []
+    for i, chapter_dict in enumerate(chapters):
+        full_chapter_title = list(chapter_dict.keys())[0]
+        chapter_content = list(chapter_dict.values())[0]
+        if ' - ' in full_chapter_title:
+            chapter_title = full_chapter_title.split(' - ')[1]
+        else:
+            chapter_title = full_chapter_title
+
+        chapter_file_name = f'chapter_{i + 1}.xhtml'
+        epub_chapter = epub.EpubHtml(title=chapter_title, file_name=chapter_file_name, lang='zh-cn')
+
+        # Add paragraph breaks
+        formatted_content = ''.join(
+            f'<p>{paragraph.strip()}</p>' for paragraph in chapter_content.split('\n') if paragraph.strip())
+
+        epub_chapter.content = f'<h1>{chapter_title}</h1>{formatted_content}'
+        book.add_item(epub_chapter)
+        epub_chapters.append(epub_chapter)
+
+    # Define Table of Contents
+    book.toc = (epub_chapters)
+
+    # Add default NCX and Nav files
+    book.add_item(epub.EpubNcx())
+    book.add_item(epub.EpubNav())
+
+    # Define CSS style
+    style = '''
+    @namespace epub "http://www.idpf.org/2007/ops";
+    body {
+        font-family: Cambria, Liberation Serif, serif;
+    }
+    h1 {
+        text-align: left;
+        text-transform: uppercase;
+        font-weight: 200;
+    }
+    '''
+
+    # Add CSS file
+    nav_css = epub.EpubItem(uid="style_nav", file_name="style/nav.css", media_type="text/css", content=style)
+    book.add_item(nav_css)
+
+    # Create spine
+    book.spine = ['nav'] + epub_chapters
+
+    # 检查目录是否存在，如果不存在则创建
+    directory = "./epub/"
+    if not os.path.exists(directory):
+        os.makedirs(directory)
+
+    # 保存 EPUB 文件
+    file_path = f"./epub/{title}.epub"  # 修改为您想要的文件路径
+    epub.write_epub(file_path, book)
+
+    return file_path

config.py

@@ -0,0 +1,53 @@
+import openai
+import os
+import uuid
+from dotenv import load_dotenv
+
+from tenacity import (
+    retry,
+    stop_after_attempt,
+    wait_random_exponential,
+)  # for exponential backoff
+
+
+# 加载.env文件中的环境变量
+load_dotenv()
+
+llm_model_name = "gpt-3.5-turbo-16k"
+
+
+openai.api_key = os.getenv("OPENAI_API_KEY")  # get it at https://platform.openai.com/
+if os.getenv("OPENAI_API_BASE"):
+    openai.api_base = os.getenv("OPENAI_API_BASE")
+stability_api_key = os.getenv("STABILITY_API_KEY")  # get it at https://beta.dreamstudio.ai/
+anthropic_api_key = os.getenv("ANTHROPIC_API_KEY")  # optional, if you don't add it, keep it as "YOUR ANTHROPIC API KEY"
+
+print("============ config info ===============\n")
+print("OPENAI_API_KEY:" + openai.api_key +"\n")
+print("OPENAI_API_BASE:" + openai.api_base +"\n")
+print("STABILITY_API_KEY:" + str(stability_api_key) +"\n")
+print("ANTHROPIC_API_KEY:" + str(anthropic_api_key) +"\n")
+
+@retry(wait=wait_random_exponential(min=1, max=60), stop=stop_after_attempt(6))
+def completion_with_backoff(**kwargs):
+    return openai.ChatCompletion.create(**kwargs)
+
+
+# 生成32位唯一的uuid
+def generate_uuid():
+    # 生成UUID
+    id = uuid.uuid4().hex
+    return id
+
+
+# 保存小说每章节的内容
+def save_novel_chapter(novel_id, chapter_index, file_name, file_content):
+    # 创建章节文件目录
+    chapter_folder = os.path.join(os.getcwd(), f"story/{novel_id}/chapter_{chapter_index + 1}")
+    if not os.path.exists(chapter_folder):
+        os.makedirs(chapter_folder)
+
+    # 写入章节内容到文件
+    file_path = os.path.join(chapter_folder, f"{file_name}.txt")
+    with open(file_path, "w") as file:
+        file.write(file_content)

flagged/log.csv

@@ -0,0 +1,2 @@
+小说提示词,小说章节数,AI写作风格,选择GPT模型,封面图片,EPUB文件,flag,username,timestamp
+萍儿是一个职场新人，她和自己的好闺蜜佩佩是同事，她们之间无话不谈，但在一次职位竞选中被自己的闺蜜陷害，萍儿厌恶了职场的勾心斗角，有一天从梦中醒来发现自己从现代穿越到了唐朝，凭借自己的能力从庶女一路逆袭成为太后，在逆袭的过程中受尽了各种暗算，有一次遭人陷害，偶然遇到了一个大明王爷，大明王爷使出计谋出手相救，最终萍儿成为皇后，大明成为皇上。梦醒了，但是萍儿从大明身上学到很多谋略，最终在现实的职场中成为一名精英，叱诧风云。,7,紧张刺激，小说主要讲述现代的一名职场新人穿越到古代变成庶女后的故事，类似《知否知否应是绿肥红瘦》小说。有很多对话和内心独白，描述主人公萍儿坎坷的逆袭之路。,gpt-4,,,,,2023-10-31 16:36:04.485217

keyTest.py

@@ -0,0 +1,32 @@
+import requests
+import json
+
+url = "https://api.openai-hk.com/v1/chat/completions"
+
+headers = {
+    "Content-Type": "application/json",
+    "Authorization": "Bearer hk-218wghmq4zcf4x42crjof0zlrpl4p22r96d44ola4wmyw92o"
+}
+
+data = {
+    "max_tokens": 1200,
+    "model": "gpt-3.5-turbo",
+    "temperature": 0.8,
+    "top_p": 1,
+    "presence_penalty": 1,
+    "messages": [
+        {
+            "role": "system",
+            "content": "You are ChatGPT, a large language model trained by OpenAI. Answer as concisely as possible."
+        },
+        {
+            "role": "user",
+            "content": "你是chatGPT多少？"
+        }
+    ]
+}
+
+response = requests.post(url, headers=headers, data=json.dumps(data).encode('utf-8') )
+result = response.content.decode("utf-8")
+
+print(result)

requirements.txt

@@ -0,0 +1,6 @@
+openai==0.28.0
+EbookLib==0.18
+anthropic==0.3.11
+python-dotenv==1.0.0
+gradio==3.42.0
+tenacity==8.2.3

simpleTest.py

@@ -0,0 +1,42 @@
+# coding: utf8
+import ast
+import unicodedata
+
+def fullwidth_to_halfwidth(text):
+    result = ''
+    for char in text:
+        char_width = unicodedata.east_asian_width(char)
+        if char_width == 'F' or char_width == 'W':
+            # 全角字符的宽度标记为 'F' 或 'W'
+            halfwidth_char = unicodedata.normalize('NFKC', char)
+            # 使用 NFKC 形式将全角字符转换为半角字符
+            result += halfwidth_char
+        else:
+            result += char
+    return result
+
+def strQ2B(ustring):
+    """把字符串全角转半角"""
+    ss = []
+    for s in ustring:
+        rstring = ""
+        for uchar in s:
+            inside_code = ord(uchar)
+            if inside_code == 12288:  # 全角空格直接转换
+                inside_code = 32
+            elif (inside_code >= 65281 and inside_code <= 65374):  # 全角字符（除空格）根据关系转化
+                inside_code -= 65248
+            rstring += chr(inside_code)
+        ss.append(rstring)
+    return ''.join(ss)
+
+name = "[{\"Chapter 1 - 骗局的蔓延\":\"平儿和佩佩是两名才华横溢但缺乏经验的年轻职员，她们的友谊因为争夺晋升机会而面临考验。佩佩因嫉妒和欲望而背叛了平儿，导致平儿不仅失去了晋升的机会，也失去了对办公室政治的信任。然而，平儿并不是一个轻易被击倒的人。她决心揭开佩佩的阴谋并证明自己的价值。\"},{\"Chapter 2 - 时间之门\":\"平儿被背叛的伤害使她陷入沉睡，当她醒来时，她发现自己来到了一个神秘的花园。花园里有一扇通往不同时空的门，平儿勇敢地穿越了时空。她来到了一个名为“时光之门”的地方，那里是通往古代的大门。平儿在这个充满危险和诱惑的新世界中，发现自己拥有操控时间的特殊能力。\"},{\"Chapter 3 - 宫廷的险棋\":\"平儿发现自己来到了唐朝的宫廷中，成为了一个卑微的妃子。尽管身份低微，但平儿依靠她的聪明才智和前世记忆，巧妙地在宫廷中游刃有余。她与朝臣和其他妃子之间的斗争越来越激烈，而平儿则不断改变和正义的转变引起了皇帝的注意。平儿必须面对宫廷的阴谋和挑战，才能够达到她的目标。\"},{\"Chapter 4 - 诡谲的盟约\":\"当平儿以为自己已经确立了皇后地位时，一个强大的阴谋威胁到了她的统治。在最黑暗的时刻，一个神秘的明朝王子以宫廷学者的身份，显示出他的真正身份。平儿与这位王子结盟，共同对抗阴谋，保护自己的地位。他们的智慧与战略才能，将会决定唐朝的命运和平儿的未来。\"},{\"Chapter 5 - 觉醒与蓄势\":\"当平儿从梦中醒来时，她带着前世的智慧和力量回到了现代世界。她发现自己变成了一位职场精英，凭借她对历史和背叛的了解，她在职业生涯中崭露头角，并留下了持久的遗产。平儿开始运用她在古代所学到的技巧，在现代社会中迎接新的挑战和冲突。\"},{\"Chapter 6 - 背叛与复仇\":\"《时间之门：从背叛到复仇的传奇》是一个充满惊险和冒险的奇幻故事。它探索了友谊、背叛和自我成长的主题。平儿在古代和现代两个世界中的经历，让她成为一个强大且无法被忽视的力量。故事深入探讨了权力、勇气和复仇的概念，塑造了一个令人着迷的女主角与命运抗争的故事。\"},{\"Chapter 7 - 荣耀与遗留\":\"为了让故事更加吸引人，我增加了对平儿和佩佩关系的深入探索。这突出了友谊的力量和背叛的伤害。在唐朝宫廷中，我加入了更多展示平儿成长和发展的情节，强调她在危险的宫廷游戏中的独到能力。我还引入了更多复杂的宫廷角色，为平儿提供更多机遇和挑战，以推动她的角色发展。通过增加阴谋情节和紧张气氛，增加了读者的悬念和猜测。最后，我更加流畅地将古代与现代世界联系起来，以展现平儿在两个环境中的相似与联系。\"}]"
+
+name1 = "[{\"Chapter 6 - 背叛与复仇\": \"《时间之门：从背叛到复仇的传奇》是一个充满惊险和冒险的奇幻故事它探索了友谊背叛和自我成长的主题平儿在古代和现代两个世界中的经历，让她成为一个强大且无法被忽视的力量故事深入探讨了权力勇气和复仇的概念，塑造了一个令人着迷的女主角与命运抗争的故事\"},{\"Chapter 7 - 荣耀与遗留\": \"为了让故事更加吸引人，我增加了对平儿和佩佩关系的深入探索。这突出了友谊的力量和背叛的伤害。在唐朝宫廷中，我加入了更多展示平儿成长和发展的情节，强调她在危险的宫廷游戏中的独到能力。我还引入了更多复杂的宫廷角色，为平儿提供更多机遇和挑战，以推动她的角色发展。通过增加阴谋情节和紧张气氛，增加了读者的悬念和猜测。最后，我更加流畅地将古代与现代世界��系起来，以展现平儿在两个环境中的相似与联系。\"}]"
+
+print(f'【storyline】: {name}\n\n')
+
+chapter_titles = ast.literal_eval(name)
+print(f'【chapter_titles】: {chapter_titles}\n\n')
+
+

story/0e5d9be3d7f84ac28961947d5f3f942f/chapter_1/Chapter 1 - The Betrayal.txt

@@ -0,0 +1,25 @@
+第一章 - 背叛
+
+年轻的企业精英平儿，才华出众且野心勃勃，卷入了与她信任的朋友佩佩之间的恶意办公室竞争中。在无情商业世界的背叛和失望之中，平儿一觉醒来，发现自己穿越到唐朝成为一个卑微的宫女身体寄生的灵魂。
+
+一片黑暗中，平儿缓缓睁开眼睛。疲惫的身体透露出一种淡淡的不安，仿佛来自另一个时空。纷乱的思绪渐渐清晰起来，她发现自己置身于华丽的宫殿之中，房间充满着沉重的气息。她的头脑里回想起了刚才的一幕：佩佩对她的背叛，朋友之间的狼狈一面。
+
+"怎么会这样？我的未来本该美好。" 平儿默默想道。她曾以为在商业世界中，能够发展自己的梦想，与佩佩一起开创一片新的天地。然而，现实却残酷无情。平儿被背叛和失望深深地伤害了。这个世界让她感到疲惫，决绝地迫使着她面对不可避免的挫折。
+
+无尽的黑暗中，她的脑海里依稀浮现出和佩佩密切相关的画面，商业报表和办公室的暗斗。佩佩的一句顺手牵羊的话让平儿崩溃了。如今，她在这个古代宫廷中，变成了一个庶女的灵魂寄生体，命运嘲弄着她的一切抱负和努力。
+
+躺在床上，把手掌紧握成拳头，平儿咬紧牙关。她发誓，无论在哪个时代，她都要找回自己失去的尊严和力量。她深呼吸，舒展凝固的身体，让思绪回到当下。
+
+"我是平儿，一个坚定追求目标的女人。这个时代或许不理解我的雄心，但我要利用我的才智，战胜这个陌生的世界，重铸我的命运！" 平儿发誓着。
+
+慢慢地，平儿挣脱沉重的惆怅，不再让失去束缚住她的灵魂。她站起身，目光坚毅地扫视宅子的四周。她知道自己需要找到适应这个新世界的方法。
+
+唐朝的宫廷毫不留情，她必须敏锐地观察并学会扭转局势。平儿决定利用她无与伦比的智慧和机智，赢得这个挑战。她的目标是晋升，成为这个宫廷中引起有影响力人物注意的存在。
+
+未来的道路虽然漫长而充满荆棘，平儿知道她已有了解决困难的内在力量。她要站稳脚跟，融入这个新的世界，同时保护自己免受险恶阴谋的伤害。这意味着她必须学会心机和耐心，提高自己的智慧，以此超越那些企图阻止她崛起的人。
+
+平儿看向窗外，安慰自己的决心。一场新的冒险即将展开，她将不再是一个被背叛的受害者，而是一名坚强而富有智慧的女性，为重塑她的命运而奋斗。她知道，只有这样，她才能回到现代世界，并彻底改变自己的未来。
+
+在这个陌生的古代天地中，平儿决定成为那个无人忽略的存在，无论付出多大代价。她相信背叛与失望只是人生中的一小部分，她一定能赢得她应得的荣光和自由。
+
+平儿开始紧锁眉头，考

story/0e5d9be3d7f84ac28961947d5f3f942f/chapter_2/Chapter 2 - The Rise.txt

@@ -0,0 +1,9 @@
+第二章 - 崛起
+
+为了改变自己的命运，平儿运用她非凡的智慧和机智，逐渐适应了险恶的宫廷环境。她坚定决心地晋升，凭借着她的策略头脑吸引起朝廷中有影响力的人物的注意。
+
+平儿逐渐适应了这个陌生而危险的宫廷世界。她深知在这里晋升是一条漫长而需要精明策略的道路。她要抓住每个机会，争取每一次相见的机会。她不断观察并分析各个人物之间的互动，找到他们的弱点和喜好。
+
+平儿用她的聪慧和洞察力，悄悄地积累起宫廷中的力量。她学会了如何在政治斗争的泥潭中游刃有余地舞动。她发现自己的每一步都引起了朝廷中有影响力的人物的注意。
+
+她以巧妙的方式与那些有势力的人建立起联系，并展示出她的价值。她通过认真执着的工作，成为皇帝赏识的宫

story/13927aa500284de3a924b54fbb3cd97c/chapter_1/第一章 - 幻影的脱困.txt

@@ -0,0 +1,23 @@
+第一章 - 幻影的脱困
+
+珍妮佩恩是一位才智过人的年轻女士，在一个充满压力和竞争的企业中辛勤工作。每天，她穿过一栋巨大而冷酷的办公楼，度过无尽的工作时间，承受着高压和残酷的竞争。然而，在她看似平凡的外表下，隐藏着一颗渴望冒险和探索的梦想家的心。
+
+这一天，珍妮佩恩按部就班地完成了一连串繁重的任务，却感到自己又一次沉浸在绝望和无望之中。她渴望一种解脱，渴望逃离现实的枷锁，再次找到那个能给她带来兴奋和激情的地方。
+
+她双手颤抖地打开了她钟爱的虚拟现实游戏《幻影》。眼前展现出了一个神奇而又充满魔法的世界。这个世界充满了各种奇异的生物、令人叹为观止的景色和潜藏着无尽秘密的地方。这是一个让珍妮佩恩可以成为真正无所不能的英雄，并与其他游戏玩家建立联系的地方。
+
+当她穿越游戏门户进入《幻影》的时刻，她完全沉浸在了奇幻世界的美妙之中。她的外貌转变，她变成了一名崭露头角的战士，凭借卓越的技巧和无尽的勇气赢得了众人的尊敬和仰慕。
+
+然而，她并不知道，《幻影》隐藏着一个永久改变她生活的魔法秘密。在她击败怪物、探索未知领域的过程中，一股神秘的力量渗透进她的身体，赋予她非凡的能力。她的眼睛闪烁着星星般的光芒，她的手指蕴含着无尽的魔力。
+
+然而，幻影的影响力并不仅限于游戏。当珍妮佩恩回到现实世界时，她开始在日常生活中出现奇怪的现象。她的手指能释放出闪电，她的眼睛能看到隐藏的真相。这让她震惊不已，也让她感到困惑和恐慌。
+
+虚拟和现实的界限开始模糊起来，珍妮佩恩陷入了困惑和恐慌之中。她不知道自己身处何处，也不知道如何控制这些力量。于是，她开始探寻《幻影》背后的秘密，希望找到答案，同时追求自己内心的平静与和谐。
+
+通过在在线游戏社区的交流和探寻，《幻影》揭示了一个隐藏的游戏玩家社群。这群玩家发现了连接现实世界和虚拟世界的门户。尽管他们来自不同的背景和经历，但他们团结一致，展开任务，保护两个世界免受即将降临的邪恶混乱的威胁。
+
+对珍妮佩恩来说，《幻影》不再只是一款游戏，它变成了她的命运。她必须面对自己内心的恐惧和挑战，接受自己身为领导者和英雄的真正潜力。
+
+《幻影的脱困》讲述了一个关于勇气、发现真正自我的旅程。在珍妮佩恩的冒险中，她不仅找到了自己，也找到了一群真挚的盟友和隐藏的力量。这个故事将带你进入一个充满魔法、想象力和意想不到的转变的世界，唤醒你内在的勇气和梦想。
+
+（注：通过这个章节，预示着珍妮佩恩即将进入她作为主人公的冒险旅程，将她的命运与《幻影》紧密联系在一起。章节的语气紧张刺激，小说主要讲述现代的一名职场新人穿越到古代变成庶女后的故事，类似《知否知否应是绿肥红瘦》小说。有很多对话和内心独白，描述主人公萍儿坎坷的逆袭之路。）

write_story.py

@@ -0,0 +1,385 @@
+# coding: utf8
+import random
+import os
+import requests
+import ast
+import time
+from anthropic import Anthropic
+from config import anthropic_api_key
+from config import llm_model_name
+from config import completion_with_backoff
+from config import save_novel_chapter
+from config import generate_uuid
+
+def print_step_costs(response, model):
+    input = response['usage']['prompt_tokens']
+    output = response['usage']['completion_tokens']
+
+    if model == "gpt-4" or model == "gpt-4":
+        input_per_token = 0.00003
+        output_per_token = 0.00006
+    if model == "gpt-3.5-turbo-16k":
+        input_per_token = 0.000003
+        output_per_token = 0.000004
+    if model == "gpt-4-32k" or model == "gpt-4-32k":
+        input_per_token = 0.00006
+        output_per_token = 0.00012
+    if model == "gpt-3.5-turbo" or model == "gpt-3.5-turbo":
+        input_per_token = 0.0000015
+        output_per_token = 0.000002
+    if model == "claude-2":
+        input_per_token = 0.00001102
+        output_per_token = 0.00003268
+
+    input_cost = int(input) * input_per_token
+    output_cost = int(output) * output_per_token
+
+    total_cost = input_cost + output_cost
+    print('Step Cost (OpenAI):', total_cost)
+
+
+def print_step_costs_anthropic(prompt, response):
+    client = Anthropic()
+    in_tokens = client.count_tokens(prompt)
+    out_tokens = client.count_tokens(response)
+
+    input_cost = 0.00001102 * in_tokens
+    output_cost = 0.00003268 * out_tokens
+
+    total_cost = input_cost + output_cost
+    print('Step Cost (Anthropic):', total_cost)
+
+
+def generate_plots(prompt):
+    response = completion_with_backoff(
+        model=llm_model_name,
+        messages=[
+            {"role": "system", "content": "You are a creative assistant that generates engaging fantasy novel plots."},
+            {"role": "user", "content": f"Generate 10 fantasy novel plots based on this prompt: {prompt}"}
+        ]
+    )
+
+    print_step_costs(response, llm_model_name)
+
+    return response['choices'][0]['message']['content'].split('\n')
+
+
+def select_most_engaging(plots):
+    response = completion_with_backoff(
+        model=llm_model_name,
+        messages=[
+            {"role": "system", "content": "You are an expert in writing fantastic fantasy novel plots."},
+            {"role": "user",
+             "content": f"Here are a number of possible plots for a new novel: "
+                        f"{plots}\n\n--\n\nNow, write the final plot that we will go with. "
+                        f"It can be one of these, a mix of the best elements of multiple, "
+                        f"or something completely new and better. "
+                        f"The most important thing is the plot should be fantastic, unique, and engaging."}
+        ]
+    )
+
+    print_step_costs(response, llm_model_name)
+
+    return response['choices'][0]['message']['content']
+
+
+def improve_plot(plot):
+    response = completion_with_backoff(
+        model=llm_model_name,
+        messages=[
+            {"role": "system", "content": "You are an expert in improving and refining story plots."},
+            {"role": "user", "content": f"Improve this plot: {plot}"}
+        ]
+    )
+
+    print_step_costs(response, llm_model_name)
+
+    return response['choices'][0]['message']['content']
+
+
+def get_title(plot):
+    response = completion_with_backoff(
+        model=llm_model_name,
+        messages=[
+            {"role": "system", "content": "You are an expert writer."},
+            {"role": "user",
+             "content": f"Here is the plot: {plot}\n\nWhat is the title of this book? "
+                        f"Just respond with the title, do nothing else. Please respond in Chinese."}
+        ]
+    )
+
+    print_step_costs(response, llm_model_name)
+
+    return response['choices'][0]['message']['content']
+
+
+def write_first_chapter(plot, first_chapter_title, writing_style, claude=True):
+    if claude:
+        url = "https://api.anthropic.com/v1/complete"
+
+        headers = {
+            "anthropic-version": "2023-06-01",
+            "content-type": "application/json",
+            "x-api-key": anthropic_api_key,
+        }
+
+        prompt_one = f"\n\nHuman: You are a world-class fantasy writer. " \
+                     f"I will give you the title of a novel, a high-level plot to follow, " \
+                     f"and a desired writing style to use. " \
+                     f"From the title, plot, and writing style, write the first chapter of the novel. " \
+                     f"Make it incredibly unique, engaging, and well-written. " \
+                     f"Start it off with a bang, and include dialogue. " \
+                     f"Include only the chapter text, and no surrounding explanations or text. " \
+                     f"Do you understand?\n\nAssistant: Yes, I understand. " \
+                     f"Please provide the title, plot, and writing style, " \
+                     f"and I will write a fantastic opening chapter with dialogue that will hook the reader." \
+                     f"\n\nHuman: Here is the high-level plot to follow: {plot}" \
+                     f"\n\nThe title of the novel is: `{first_chapter_title}`.\n\n" \
+                     f"Here is a description of the writing style you should use: `{writing_style}`" \
+                     f"\n\nWrite the first chapter please!\n\nAssistant: " \
+                     f"Okay, I've got a really exciting first chapter for you. " \
+                     f"It's twenty paragraphs long and very well-written. " \
+                     f"As you can see, the language I use is very understandable — " \
+                     f"I avoided using overly complex words and phrases. Please respond in Chinese:" \
+                     f"\n\nTitle: {first_chapter_title}\n\nChapter #1 Text```"
+
+        data = {
+            "model": "claude-2",
+            "prompt": prompt_one,
+            "max_tokens_to_sample": 5000,
+        }
+
+        response = requests.post(url, headers=headers, json=data)
+        print(response)
+        initial_first_chapter = response.json()['completion'].strip().split('```')[0].strip()
+
+        print_step_costs_anthropic(prompt_one, response.json()['completion'])
+
+        prompt_two = f"\n\nHuman: You are a world-class fantasy writer. " \
+                     f"Your job is to take your student's rough initial draft of the first chapter of their fantasy novel, " \
+                     f"and rewrite it to be significantly better, with much more detail. " \
+                     f"Do you understand?\n\nAssistant: Yes, I understand. " \
+                     f"Please provide the plot and the student-written chapter, " \
+                     f"and I will rewrite the chapter in a far superior way.\n\nHuman: " \
+                     f"Here is the high-level plot you asked your student to follow: " \
+                     f"{plot}\n\nHere is the first chapter they wrote: {initial_first_chapter}\n\n" \
+                     f"Now, rewrite the first chapter of this novel, in a way that is far superior to your student's chapter. " \
+                     f"It should still follow the exact same plot, but it should be far more detailed, much longer, and more engaging. " \
+                     f"Here is a description of the writing style you should use: `{writing_style}`\n\nAssistant: Okay, I've rewritten the first chapter. " \
+                     f"I took great care to improve it. While the plot is the same, " \
+                     f"you can see that my version is noticeably longer, easier to read, " \
+                     f"and more exciting. Also, the language I used is far more accessible to a broader audience." \
+                     f"Please respond in Chinese.\n\n```"
+        data = {
+            "model": "claude-2",
+            "prompt": prompt_two,
+            "max_tokens_to_sample": 5000,
+        }
+
+        response_improved = requests.post(url, headers=headers, json=data)
+
+        print_step_costs_anthropic(prompt_two, response_improved.json()['completion'])
+
+        return response_improved.json()['completion'].strip().split('```')[0].strip()
+
+
+    else:
+        response = completion_with_backoff(
+            model=llm_model_name,
+            messages=[
+                {"role": "system", "content": "You are a world-class fantasy writer."},
+                {"role": "user",
+                 "content": f"Here is the high-level plot to follow: {plot}\n\nWrite the first chapter of this novel: `{first_chapter_title}`.\n\nMake it incredibly unique, engaging, and well-written.\n\nHere is a description of the writing style you should use: `{writing_style}`\n\nInclude only the chapter text. There is no need to rewrite the chapter name. Please respond in Chinese."}
+            ]
+        )
+
+        print_step_costs(response, llm_model_name)
+
+        improved_response = completion_with_backoff(
+            model=llm_model_name,
+            messages=[
+                {"role": "system",
+                 "content": "You are a world-class fantasy writer. Your job is to take your student's rough initial draft of the first chapter of their fantasy novel, and rewrite it to be significantly better, with much more detail."},
+                {"role": "user",
+                 "content": f"Here is the high-level plot you asked your student to follow: {plot}\n\nHere is the first chapter they wrote: {response['choices'][0]['message']['content']}\n\nNow, rewrite the first chapter of this novel, in a way that is far superior to your student's chapter. It should still follow the exact same plot, but it should be far more detailed, much longer, and more engaging. Here is a description of the writing style you should use: `{writing_style}`.Please respond in Chinese."}
+            ]
+        )
+
+        print_step_costs(response, llm_model_name)
+
+        return improved_response['choices'][0]['message']['content']
+
+
+def write_chapter(previous_chapters, plot, chapter_title, claude=True):
+    if claude:
+        url = "https://api.anthropic.com/v1/complete"
+
+        headers = {
+            "anthropic-version": "2023-06-01",
+            "content-type": "application/json",
+            "x-api-key": anthropic_api_key,
+        }
+
+        prompt = f"\n\nHuman: You are a world-class fantasy writer. I will provide you with the plot of the novel, the previous chapters, and the plan for the next chapter. Your task is to write the next chapter of the novel in Chinese, following the plot and taking in the previous chapters as context. Do you understand?\n\nAssistant: Yes, I understand. You want me to write the next chapter of a novel, using the plot you provide, the previous chapters for context, and a specific plan for the next chapter. I will ensure the chapter is beautifully written and I will not rewrite the chapter name.\n\nHuman: That's correct. Here is the plot: {plot}\n\nHere are the previous chapters: {previous_chapters}\n\nHere is the plan for the next chapter: {chapter_title}\n\nWrite it beautifully. Include only the chapter text. There is no need to rewrite the chapter name.\n\nAssistant: Here is the next chapter. As you can see, it's around the same length as the previous chapters, and contains witty dialogue:\n```Chapter"
+
+        data = {
+            "model": "claude-2",
+            "prompt": prompt,
+            "max_tokens_to_sample": 5000,
+        }
+
+        response = requests.post(url, headers=headers, json=data)
+
+        print_step_costs_anthropic(prompt, response.json()['completion'])
+
+        return 'Chapter ' + response.json()['completion'].strip().split('```')[0].strip()
+    else:
+        try:
+            i = random.randint(1, 2242)
+            response = completion_with_backoff(
+                model=llm_model_name,
+                messages=[
+                    {"role": "system", "content": "You are a world-class fantasy writer. "},
+                    {"role": "user",
+                     "content": f"Plot: {plot}, Previous Chapters: {previous_chapters}\n\n--\n\nWrite the next chapter of this novel in Chinese, following the plot and taking in the previous chapters as context. Here is the plan for this chapter: {chapter_title}\n\nWrite it beautifully. Include only the chapter text. There is no need to rewrite the chapter name."}
+                ]
+            )
+
+            print_step_costs(response, llm_model_name)
+
+            return response['choices'][0]['message']['content']
+        except:
+            response = completion_with_backoff(
+                model=llm_model_name,
+                messages=[
+                    {"role": "system", "content": "You are a world-class fantasy writer."},
+                    {"role": "user",
+                     "content": f"Plot: {plot}, Previous Chapters: {previous_chapters}\n\n--\n\nWrite the next chapter of this novel in Chinese, following the plot and taking in the previous chapters as context. Here is the plan for this chapter: {chapter_title}\n\nWrite it beautifully. Include only the chapter text. There is no need to rewrite the chapter name."}
+                ]
+            )
+
+            print_step_costs(response, llm_model_name)
+
+            return response['choices'][0]['message']['content']
+
+
+def generate_storyline(prompt, num_chapters):
+    print("Generating storyline with chapters and high-level details...")
+    json_format = """[{"Chapter CHAPTER_NUMBER_HERE - CHAPTER_TITLE_GOES_HERE": 
+    "CHAPTER_OVERVIEW_AND_DETAILS_GOES_HERE"}, ...]"""
+    response = completion_with_backoff(
+        model=llm_model_name,
+        messages=[
+            {"role": "system",
+             "content": "You are a world-class fantasy writer. Your job is to write a detailed storyline,"
+                        " complete with chapters, for a fantasy novel. "
+                        "Don't be flowery -- you want to get the message across in as few words as possible. "
+                        "But those words should contain lots of information. Please respond in Chinese"},
+            {"role": "user",
+             "content": f'Write a fantastic storyline with {num_chapters} chapters and high-level details based on this plot:'
+                        f' {prompt}.\n\nDo it in this list of dictionaries format {json_format}.'
+                        f' And Please respond in Chinese. The response content must be in standard JSON format, without any prefixes and special symbols.'}
+        ]
+    )
+
+    print_step_costs(response, llm_model_name)
+
+    improved_response = completion_with_backoff(
+        model=llm_model_name,
+        messages=[
+            {"role": "system",
+             "content": "You are a world-class fantasy writer. "
+                        "Your job is to take your student's rough initial draft of the storyline of a fantasy novel in Chinese, "
+                        "and rewrite it to be significantly better. Please respond in Chinese"},
+            {"role": "user",
+             "content": f"Here is the draft storyline they wrote: {response['choices'][0]['message']['content']}\n\nNow, "
+                        f"rewrite the storyline in Chinese, in a way that is far superior to your student's version. "
+                        f"It should have the same number of chapters, "
+                        f"but it should be much improved in as many ways as possible. "
+                        f"Remember to do it in this list of dictionaries format {json_format}. please respond in Chinese"
+                        f' And please, only return the JSON content without any prefix.'}
+        ]
+    )
+
+    print_step_costs(improved_response, llm_model_name)
+
+    return improved_response['choices'][0]['message']['content']
+
+
+def write_to_file(prompt, content):
+    # Create a directory for the prompts if it doesn't exist
+    if not os.path.exists('prompts'):
+        os.mkdir('prompts')
+
+    # Replace invalid characters for filenames
+    valid_filename = ''.join(c for c in prompt if c.isalnum() or c in (' ', '.', '_')).rstrip()
+    file_path = f'prompts/{valid_filename}.txt'
+
+    with open(file_path, 'w', encoding='utf-8') as f:
+        f.write(content)
+
+    print(f'Output for prompt "{prompt}" has been written to {file_path}\n')
+
+
+def write_fantasy_novel(prompt, num_chapters, writing_style, claude_true=False, model_name="gpt-3.5-turbo-16k"):
+    global llm_model_name
+    llm_model_name = model_name
+
+    # 每本小说生成一个唯一的uuid
+    novel_id = generate_uuid()
+
+    plots = generate_plots(prompt)
+    print('generated plots')
+    print(f'【plots】: {plots}\n\n')
+
+    best_plot = select_most_engaging(plots)
+    print('selected best plot')
+    print(f'【best_plot】: {best_plot}\n\n')
+
+    improved_plot = improve_plot(best_plot)
+    print('plot improved')
+    print(f'【improved_plot】: {improved_plot}\n\n')
+    time.sleep(20)
+
+    title = get_title(improved_plot)
+    print('title generated')
+    print(f'【title】: {title}\n\n')
+
+    storyline = generate_storyline(improved_plot, num_chapters)
+    print('storyline generated')
+    print(f'【storyline】: {storyline}\n\n')
+
+    chapter_titles = ast.literal_eval(storyline)
+    print(f'【chapter_titles】: {chapter_titles}\n\n')
+
+    novel = f"Storyline:\n{storyline}\n\n"
+
+    first_chapter = write_first_chapter(storyline, chapter_titles[0], writing_style.strip(), claude_true)
+    print('first chapter written')
+    save_novel_chapter(novel_id, 0, list(chapter_titles[0])[0], first_chapter)
+    print(f'【first_chapter】: {first_chapter}\n\n')
+    time.sleep(20)
+
+    novel += f"Chapter 1:\n{first_chapter}\n"
+    chapters = [first_chapter]
+
+    for i in range(num_chapters - 1):
+        print(f"Writing chapter {i + 2}...")  # + 2 because the first chapter was already added
+        time.sleep(30)
+
+        chapter = write_chapter(novel, storyline, chapter_titles[i + 1], claude_true)
+        try:
+            if len(str(chapter)) < 100:
+                time.sleep(30)
+                print('Length minimum not hit. Trying again.')
+                chapter = write_chapter(novel, storyline, chapter_titles[i + 1], claude_true)
+        except:
+            time.sleep(20)
+            chapter = write_chapter(novel, storyline, chapter_titles[i + 1], claude_true)
+
+        novel += f"Chapter {i + 2}:\n{chapter}\n"
+        chapters.append(chapter)
+        print(f'【Chapter_{i + 2}】: {chapter}\n\n')
+        save_novel_chapter(novel_id, (i+1), list(chapter_titles[i + 1])[0], chapter)
+
+    return novel, title, chapters, chapter_titles