Copy_of_gpt_author_v2.ipynb

{
  "cells": [
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "view-in-github",
        "colab_type": "text"
      },
      "source": [
        "<a href=\"https://colab.research.google.com/github/alephnot/gpt-author/blob/main/Copy_of_gpt_author_v2.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
      ]
    },
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "Ih8GSkb28daX"
      },
      "source": [
        "# gpt-author v2\n",
        "By Matt Shumer (https://twitter.com/mattshumer_)\n",
        "\n",
        "Github repo: https://github.com/mshumer/gpt-author\n",
        "\n",
        "Generate an entire novel in minutes, and automatically package it as an e-book.\n",
        "\n",
        "To generate a book:\n",
        "1.  In the first cell, add in your OpenAI, Stability, and (optionally, Anthropic) keys (see the first cell for instructions to get them).\n",
        "2.  Fill in the prompt, number of chapters, and writing style in the last cell.\n",
        "3. Run all the cells! After some time, your EPUB file should appear in the filesystem.\n",
        "\n",
        "\n"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {
        "id": "HTGF4Bw6Hnuz",
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "outputId": "bc773bd0-59c7-4871-fa47-a838d67626d4"
      },
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "Collecting openai\n",
            "  Downloading openai-1.16.2-py3-none-any.whl (267 kB)\n",
            "\u001b[?25l     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m0.0/267.1 kB\u001b[0m \u001b[31m?\u001b[0m eta \u001b[36m-:--:--\u001b[0m\r\u001b[2K     \u001b[91m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m\u001b[91m╸\u001b[0m \u001b[32m266.2/267.1 kB\u001b[0m \u001b[31m7.9 MB/s\u001b[0m eta \u001b[36m0:00:01\u001b[0m\r\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m267.1/267.1 kB\u001b[0m \u001b[31m6.0 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
            "\u001b[?25hRequirement already satisfied: anyio<5,>=3.5.0 in /usr/local/lib/python3.10/dist-packages (from openai) (3.7.1)\n",
            "Requirement already satisfied: distro<2,>=1.7.0 in /usr/lib/python3/dist-packages (from openai) (1.7.0)\n",
            "Collecting httpx<1,>=0.23.0 (from openai)\n",
            "  Downloading httpx-0.27.0-py3-none-any.whl (75 kB)\n",
            "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m75.6/75.6 kB\u001b[0m \u001b[31m9.1 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
            "\u001b[?25hRequirement already satisfied: pydantic<3,>=1.9.0 in /usr/local/lib/python3.10/dist-packages (from openai) (2.6.4)\n",
            "Requirement already satisfied: sniffio in /usr/local/lib/python3.10/dist-packages (from openai) (1.3.1)\n",
            "Requirement already satisfied: tqdm>4 in /usr/local/lib/python3.10/dist-packages (from openai) (4.66.2)\n",
            "Requirement already satisfied: typing-extensions<5,>=4.7 in /usr/local/lib/python3.10/dist-packages (from openai) (4.10.0)\n",
            "Requirement already satisfied: idna>=2.8 in /usr/local/lib/python3.10/dist-packages (from anyio<5,>=3.5.0->openai) (3.6)\n",
            "Requirement already satisfied: exceptiongroup in /usr/local/lib/python3.10/dist-packages (from anyio<5,>=3.5.0->openai) (1.2.0)\n",
            "Requirement already satisfied: certifi in /usr/local/lib/python3.10/dist-packages (from httpx<1,>=0.23.0->openai) (2024.2.2)\n",
            "Collecting httpcore==1.* (from httpx<1,>=0.23.0->openai)\n",
            "  Downloading httpcore-1.0.5-py3-none-any.whl (77 kB)\n",
            "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m77.9/77.9 kB\u001b[0m \u001b[31m10.3 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
            "\u001b[?25hCollecting h11<0.15,>=0.13 (from httpcore==1.*->httpx<1,>=0.23.0->openai)\n",
            "  Downloading h11-0.14.0-py3-none-any.whl (58 kB)\n",
            "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m58.3/58.3 kB\u001b[0m \u001b[31m7.4 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
            "\u001b[?25hRequirement already satisfied: annotated-types>=0.4.0 in /usr/local/lib/python3.10/dist-packages (from pydantic<3,>=1.9.0->openai) (0.6.0)\n",
            "Requirement already satisfied: pydantic-core==2.16.3 in /usr/local/lib/python3.10/dist-packages (from pydantic<3,>=1.9.0->openai) (2.16.3)\n",
            "Installing collected packages: h11, httpcore, httpx, openai\n",
            "Successfully installed h11-0.14.0 httpcore-1.0.5 httpx-0.27.0 openai-1.16.2\n",
            "Collecting EbookLib\n",
            "  Downloading EbookLib-0.18.tar.gz (115 kB)\n",
            "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m115.5/115.5 kB\u001b[0m \u001b[31m3.5 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
            "\u001b[?25h  Preparing metadata (setup.py) ... \u001b[?25l\u001b[?25hdone\n",
            "Requirement already satisfied: lxml in /usr/local/lib/python3.10/dist-packages (from EbookLib) (4.9.4)\n",
            "Requirement already satisfied: six in /usr/local/lib/python3.10/dist-packages (from EbookLib) (1.16.0)\n",
            "Building wheels for collected packages: EbookLib\n",
            "  Building wheel for EbookLib (setup.py) ... \u001b[?25l\u001b[?25hdone\n",
            "  Created wheel for EbookLib: filename=EbookLib-0.18-py3-none-any.whl size=38778 sha256=bf5656ccd9b4dc7ab56e413d3e2a908b1cc31cb33938ec0aa8ed8a12ba200177\n",
            "  Stored in directory: /root/.cache/pip/wheels/0f/38/cc/a3728bb72a315d9d8766fb71d362136372066fc25ad838f8fa\n",
            "Successfully built EbookLib\n",
            "Installing collected packages: EbookLib\n",
            "Successfully installed EbookLib-0.18\n",
            "Collecting anthropic\n",
            "  Downloading anthropic-0.23.1-py3-none-any.whl (869 kB)\n",
            "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m869.1/869.1 kB\u001b[0m \u001b[31m9.0 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
            "\u001b[?25hRequirement already satisfied: anyio<5,>=3.5.0 in /usr/local/lib/python3.10/dist-packages (from anthropic) (3.7.1)\n",
            "Requirement already satisfied: distro<2,>=1.7.0 in /usr/lib/python3/dist-packages (from anthropic) (1.7.0)\n",
            "Requirement already satisfied: httpx<1,>=0.23.0 in /usr/local/lib/python3.10/dist-packages (from anthropic) (0.27.0)\n",
            "Requirement already satisfied: pydantic<3,>=1.9.0 in /usr/local/lib/python3.10/dist-packages (from anthropic) (2.6.4)\n",
            "Requirement already satisfied: sniffio in /usr/local/lib/python3.10/dist-packages (from anthropic) (1.3.1)\n",
            "Requirement already satisfied: tokenizers>=0.13.0 in /usr/local/lib/python3.10/dist-packages (from anthropic) (0.15.2)\n",
            "Requirement already satisfied: typing-extensions<5,>=4.7 in /usr/local/lib/python3.10/dist-packages (from anthropic) (4.10.0)\n",
            "Requirement already satisfied: idna>=2.8 in /usr/local/lib/python3.10/dist-packages (from anyio<5,>=3.5.0->anthropic) (3.6)\n",
            "Requirement already satisfied: exceptiongroup in /usr/local/lib/python3.10/dist-packages (from anyio<5,>=3.5.0->anthropic) (1.2.0)\n",
            "Requirement already satisfied: certifi in /usr/local/lib/python3.10/dist-packages (from httpx<1,>=0.23.0->anthropic) (2024.2.2)\n",
            "Requirement already satisfied: httpcore==1.* in /usr/local/lib/python3.10/dist-packages (from httpx<1,>=0.23.0->anthropic) (1.0.5)\n",
            "Requirement already satisfied: h11<0.15,>=0.13 in /usr/local/lib/python3.10/dist-packages (from httpcore==1.*->httpx<1,>=0.23.0->anthropic) (0.14.0)\n",
            "Requirement already satisfied: annotated-types>=0.4.0 in /usr/local/lib/python3.10/dist-packages (from pydantic<3,>=1.9.0->anthropic) (0.6.0)\n",
            "Requirement already satisfied: pydantic-core==2.16.3 in /usr/local/lib/python3.10/dist-packages (from pydantic<3,>=1.9.0->anthropic) (2.16.3)\n",
            "Requirement already satisfied: huggingface_hub<1.0,>=0.16.4 in /usr/local/lib/python3.10/dist-packages (from tokenizers>=0.13.0->anthropic) (0.20.3)\n",
            "Requirement already satisfied: filelock in /usr/local/lib/python3.10/dist-packages (from huggingface_hub<1.0,>=0.16.4->tokenizers>=0.13.0->anthropic) (3.13.3)\n",
            "Requirement already satisfied: fsspec>=2023.5.0 in /usr/local/lib/python3.10/dist-packages (from huggingface_hub<1.0,>=0.16.4->tokenizers>=0.13.0->anthropic) (2023.6.0)\n",
            "Requirement already satisfied: requests in /usr/local/lib/python3.10/dist-packages (from huggingface_hub<1.0,>=0.16.4->tokenizers>=0.13.0->anthropic) (2.31.0)\n",
            "Requirement already satisfied: tqdm>=4.42.1 in /usr/local/lib/python3.10/dist-packages (from huggingface_hub<1.0,>=0.16.4->tokenizers>=0.13.0->anthropic) (4.66.2)\n",
            "Requirement already satisfied: pyyaml>=5.1 in /usr/local/lib/python3.10/dist-packages (from huggingface_hub<1.0,>=0.16.4->tokenizers>=0.13.0->anthropic) (6.0.1)\n",
            "Requirement already satisfied: packaging>=20.9 in /usr/local/lib/python3.10/dist-packages (from huggingface_hub<1.0,>=0.16.4->tokenizers>=0.13.0->anthropic) (24.0)\n",
            "Requirement already satisfied: charset-normalizer<4,>=2 in /usr/local/lib/python3.10/dist-packages (from requests->huggingface_hub<1.0,>=0.16.4->tokenizers>=0.13.0->anthropic) (3.3.2)\n",
            "Requirement already satisfied: urllib3<3,>=1.21.1 in /usr/local/lib/python3.10/dist-packages (from requests->huggingface_hub<1.0,>=0.16.4->tokenizers>=0.13.0->anthropic) (2.0.7)\n"
          ]
        }
      ],
      "source": [
        "!pip install openai\n",
        "!pip install EbookLib\n",
        "!pip install anthropic\n",
        "\n",
        "from openai import OpenAI\n",
        "import os\n",
        "from ebooklib import epub\n",
        "import base64\n",
        "import os\n",
        "import requests\n",
        "from google.colab import userdata\n",
        "\n",
        "#openai.api_key =  userdata.get('OPENAI_API_KEY')\n",
        "stability_api_key = userdata.get(\"STABILITY_API_KEY\") # get it at https://beta.dreamstudio.ai/\n",
        "anthropic_api_key = userdata.get(\"ANTHROPIC_API_KEY\") # optional, if you don't add it, keep it as \"YOUR ANTHROPIC API KEY\"\n",
        "\n",
        "if anthropic_api_key != \"YOUR ANTHROPIC API KEY\":\n",
        "  claude_true = True\n",
        "else:\n",
        "  claude_true = False\n",
        "\n",
        "client = OpenAI(\n",
        "    api_key=userdata.get(\"OPENAI_API_KEY\"),\n",
        ")"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": 17,
      "metadata": {
        "id": "__nvOnmhPqwy"
      },
      "outputs": [],
      "source": [
        "def generate_cover_prompt(plot):\n",
        "    response = client.chat.completions.create(\n",
        "        model=\"gpt-3.5-turbo-16k\",\n",
        "        messages=[\n",
        "            {\"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.\"},\n",
        "            {\"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.\"}\n",
        "        ]\n",
        "    )\n",
        "    return response.choices[0].message.content\n",
        "\n",
        "\n",
        "def create_cover_image(plot):\n",
        "\n",
        "  plot = str(generate_cover_prompt(plot))\n",
        "\n",
        "  engine_id = \"stable-diffusion-xl-beta-v2-2-2\"\n",
        "  api_host = os.getenv('API_HOST', 'https://api.stability.ai')\n",
        "  api_key = stability_api_key\n",
        "\n",
        "  if api_key is None:\n",
        "      raise Exception(\"Missing Stability API key.\")\n",
        "\n",
        "  response = requests.post(\n",
        "      f\"{api_host}/v1/generation/{engine_id}/text-to-image\",\n",
        "      headers={\n",
        "          \"Content-Type\": \"application/json\",\n",
        "          \"Accept\": \"application/json\",\n",
        "          \"Authorization\": f\"Bearer {api_key}\"\n",
        "      },\n",
        "      json={\n",
        "          \"text_prompts\": [\n",
        "              {\n",
        "                  \"text\": plot\n",
        "              }\n",
        "          ],\n",
        "          \"cfg_scale\": 7,\n",
        "          \"clip_guidance_preset\": \"FAST_BLUE\",\n",
        "          \"height\": 768,\n",
        "          \"width\": 512,\n",
        "          \"samples\": 1,\n",
        "          \"steps\": 30,\n",
        "      },\n",
        "  )\n",
        "\n",
        "  if response.status_code != 200:\n",
        "      raise Exception(\"Non-200 response: \" + str(response.text))\n",
        "\n",
        "  data = response.json()\n",
        "\n",
        "  for i, image in enumerate(data[\"artifacts\"]):\n",
        "      with open(f\"/content/cover.png\", \"wb\") as f: # replace this if running locally, to where you store the cover file\n",
        "          f.write(base64.b64decode(image[\"base64\"]))"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {
        "id": "4w-tRWAVDf-V"
      },
      "outputs": [],
      "source": [
        "import uuid\n",
        "\n",
        "def generate_identifier():\n",
        "    return str(uuid.uuid4())\n",
        "\n",
        "def create_epub(title, author, chapters, cover_image_path='cover.png'):\n",
        "    book = epub.EpubBook()\n",
        "\n",
        "    # Set metadata\n",
        "    book.set_identifier(generate_identifier())\n",
        "    book.set_title(title)\n",
        "    book.set_language('en')\n",
        "    book.add_author(author)\n",
        "\n",
        "    # Add cover image\n",
        "    with open(cover_image_path, 'rb') as cover_file:\n",
        "        cover_image = cover_file.read()\n",
        "    book.set_cover('cover.png', cover_image)\n",
        "\n",
        "    # Create chapters and add them to the book\n",
        "    epub_chapters = []\n",
        "    for i, chapter_dict in enumerate(chapters):\n",
        "        full_chapter_title = list(chapter_dict.keys())[0]\n",
        "        chapter_content = list(chapter_dict.values())[0]\n",
        "        if ' - ' in full_chapter_title:\n",
        "            chapter_title = full_chapter_title.split(' - ')[1]\n",
        "        else:\n",
        "            chapter_title = full_chapter_title\n",
        "\n",
        "        chapter_file_name = f'chapter_{i+1}.xhtml'\n",
        "        epub_chapter = epub.EpubHtml(title=chapter_title, file_name=chapter_file_name, lang='en')\n",
        "\n",
        "        # Add paragraph breaks\n",
        "        formatted_content = ''.join(f'<p>{paragraph.strip()}</p>' for paragraph in chapter_content.split('\\n') if paragraph.strip())\n",
        "\n",
        "        epub_chapter.content = f'<h1>{chapter_title}</h1>{formatted_content}'\n",
        "        book.add_item(epub_chapter)\n",
        "        epub_chapters.append(epub_chapter)\n",
        "\n",
        "\n",
        "    # Define Table of Contents\n",
        "    book.toc = (epub_chapters)\n",
        "\n",
        "    # Add default NCX and Nav files\n",
        "    book.add_item(epub.EpubNcx())\n",
        "    book.add_item(epub.EpubNav())\n",
        "\n",
        "    # Define CSS style\n",
        "    style = '''\n",
        "    @namespace epub \"http://www.idpf.org/2007/ops\";\n",
        "    body {\n",
        "        font-family: Cambria, Liberation Serif, serif;\n",
        "    }\n",
        "    h1 {\n",
        "        text-align: left;\n",
        "        text-transform: uppercase;\n",
        "        font-weight: 200;\n",
        "    }\n",
        "    '''\n",
        "\n",
        "    # Add CSS file\n",
        "    nav_css = epub.EpubItem(uid=\"style_nav\", file_name=\"style/nav.css\", media_type=\"text/css\", content=style)\n",
        "    book.add_item(nav_css)\n",
        "\n",
        "    # Create spine\n",
        "    book.spine = ['nav'] + epub_chapters\n",
        "\n",
        "    # Save the EPUB file\n",
        "    epub.write_epub(f'{title}.epub', book)"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": 30,
      "metadata": {
        "id": "KEClRDpDK7VW"
      },
      "outputs": [],
      "source": [
        "import openai\n",
        "import random\n",
        "import json\n",
        "import ast\n",
        "from anthropic import Anthropic\n",
        "\n",
        "def print_step_costs(response, model):\n",
        "  #input = response['usage']['prompt_tokens']\n",
        "  input = response.usage.prompt_tokens\n",
        "  #output = response['usage']['completion_tokens']\n",
        "  output = response.usage.completion_tokens\n",
        "\n",
        "  if model == \"gpt-4\" or model == \"gpt-4\":\n",
        "    input_per_token = 0.00003\n",
        "    output_per_token = 0.00006\n",
        "  if model == \"gpt-3.5-turbo-16k\":\n",
        "    input_per_token = 0.000003\n",
        "    output_per_token = 0.000004\n",
        "  if model == \"gpt-4-32k\" or model == \"gpt-4-32k\":\n",
        "    input_per_token = 0.00006\n",
        "    output_per_token = 0.00012\n",
        "  if model == \"gpt-3.5-turbo\" or model == \"gpt-3.5-turbo\":\n",
        "    input_per_token = 0.0000015\n",
        "    output_per_token = 0.000002\n",
        "  if model == \"claude-2\":\n",
        "    input_per_token = 0.00001102\n",
        "    output_per_token = 0.00003268\n",
        "\n",
        "  input_cost = int(input) * input_per_token\n",
        "  output_cost = int(output) * output_per_token\n",
        "\n",
        "  total_cost = input_cost + output_cost\n",
        "  print('Step Cost (OpenAI):', total_cost)\n",
        "\n",
        "def print_step_costs_anthropic(prompt, response):\n",
        "  client = Anthropic()\n",
        "  in_tokens = client.count_tokens(prompt)\n",
        "  out_tokens = client.count_tokens(response)\n",
        "\n",
        "  input_cost = 0.00001102 * in_tokens\n",
        "  output_cost = 0.00003268 * out_tokens\n",
        "\n",
        "  total_cost = input_cost + output_cost\n",
        "  print('Step Cost (Anthropic):', total_cost)\n",
        "\n",
        "def generate_plots(prompt):\n",
        "    response = client.chat.completions.create(\n",
        "        model=\"gpt-4\",\n",
        "        messages=[\n",
        "            {\"role\": \"system\", \"content\": \"You are a creative assistant that generates engaging fantasy novel plots.\"},\n",
        "            {\"role\": \"user\", \"content\": f\"Generate 10 fantasy novel plots based on this prompt: {prompt}\"}\n",
        "        ]\n",
        "    )\n",
        "\n",
        "    print_step_costs(response, \"gpt-4\")\n",
        "\n",
        "    return response.choices[0].message.content.split('\\n')\n",
        "\n",
        "def select_most_engaging(plots):\n",
        "    response = client.chat.completions.create(\n",
        "        model=\"gpt-4\",\n",
        "        messages=[\n",
        "            {\"role\": \"system\", \"content\": \"You are an expert in writing fantastic fantasy novel plots.\"},\n",
        "            {\"role\": \"user\", \"content\": f\"Here are a number of possible plots for a new novel: {plots}\\n\\n--\\n\\nNow, write the final plot that we will go with. It can be one of these, a mix of the best elements of multiple, or something completely new and better. The most important thing is the plot should be fantastic, unique, and engaging.\"}\n",
        "        ]\n",
        "    )\n",
        "\n",
        "    print_step_costs(response, \"gpt-4\")\n",
        "\n",
        "    return response.choices[0].message.content\n",
        "\n",
        "def improve_plot(plot):\n",
        "    response = client.chat.completions.create(\n",
        "        model=\"gpt-4\",\n",
        "        messages=[\n",
        "            {\"role\": \"system\", \"content\": \"You are an expert in improving and refining story plots.\"},\n",
        "            {\"role\": \"user\", \"content\": f\"Improve this plot: {plot}\"}\n",
        "        ]\n",
        "    )\n",
        "\n",
        "    print_step_costs(response, \"gpt-4\")\n",
        "\n",
        "    return response.choices[0].message.content\n",
        "\n",
        "def get_title(plot):\n",
        "    response = client.chat.completions.create(\n",
        "        model=\"gpt-3.5-turbo-16k\",\n",
        "        messages=[\n",
        "            {\"role\": \"system\", \"content\": \"You are an expert writer.\"},\n",
        "            {\"role\": \"user\", \"content\": f\"Here is the plot: {plot}\\n\\nWhat is the title of this book? Just respond with the title, do nothing else.\"}\n",
        "        ]\n",
        "    )\n",
        "\n",
        "    print_step_costs(response, \"gpt-3.5-turbo-16k\")\n",
        "\n",
        "    return response.choices[0].message.content.split('\\n')\n",
        "\n",
        "def write_first_chapter(plot, first_chapter_title, writing_style, book_genre, claude=True):\n",
        "    if claude:\n",
        "      url = \"https://api.anthropic.com/v1/complete\"\n",
        "\n",
        "      headers = {\n",
        "          \"anthropic-version\": \"2023-06-01\",\n",
        "          \"content-type\": \"application/json\",\n",
        "          \"x-api-key\": anthropic_api_key,\n",
        "      }\n",
        "\n",
        "      prompt_one = f\"\\n\\nHuman: You are a world-class {book_genre} writer. I will give you the title of a novel, a high-level plot to follow, and a desired writing style to use. From the title, plot, and writing style, write the first chapter of the novel. Make it incredibly unique, engaging, and well-written. Start it off with a bang, and include dialogue. Include only the chapter text, and no surrounding explanations or text. Do you understand?\\n\\nAssistant: Yes, I understand. Please provide the title, plot, and writing style, and I will write a fantastic opening chapter with dialogue that will hook the reader.\\n\\nHuman: Here is the high-level plot to follow: {plot}\\n\\The title of the novel is: `{first_chapter_title}`.\\n\\nHere is a description of the writing style you should use: `{writing_style}`\\n\\nWrite the first chapter please!\\n\\nAssistant: Okay, I've got a really exciting first chapter for you. It's twenty paragraphs long and very well-written. As you can see, the language I use is very understandable — I avoided using overly complex words and phrases:\\n\\nTitle: {first_chapter_title}\\n\\nChapter #1 Text```\"\n",
        "\n",
        "      data = {\n",
        "      \"model\": \"claude-2\",\n",
        "      \"prompt\": prompt_one,\n",
        "      \"max_tokens_to_sample\": 5000,\n",
        "      }\n",
        "\n",
        "      response = requests.post(url, headers=headers, json=data)\n",
        "\n",
        "      initial_first_chapter = response.json()['completion'].strip().split('```')[0].strip()\n",
        "\n",
        "      print_step_costs_anthropic(prompt_one, response.json()['completion'])\n",
        "\n",
        "      prompt_two = f\"\\n\\nHuman: 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. Do you understand?\\n\\nAssistant: Yes, I understand. Please provide the plot and the student-written chapter, and I will rewrite the chapter in a far superior way.\\n\\nHuman: Here is the high-level plot you asked your student to follow: {plot}\\n\\nHere is the first chapter they wrote: {initial_first_chapter}\\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}`\\n\\nAssistant: Okay, I've rewritten the first chapter. I took great care to improve it. While the plot is the same, you can see that my version is noticeably longer, easier to read, and more exciting. Also, the language I used is far more accessible to a broader audience.\\n\\n```\"\n",
        "      data = {\n",
        "      \"model\": \"claude-2\",\n",
        "      \"prompt\": prompt_two,\n",
        "      \"max_tokens_to_sample\": 5000,\n",
        "      }\n",
        "\n",
        "      response_improved = requests.post(url, headers=headers, json=data)\n",
        "\n",
        "      print_step_costs_anthropic(prompt_two, response_improved.json()['completion'])\n",
        "\n",
        "      return response_improved.json()['completion'].strip().split('```')[0].strip()\n",
        "\n",
        "\n",
        "    else:\n",
        "      response = client.chat.completions.create(\n",
        "          model=\"gpt-4\",\n",
        "          messages=[\n",
        "              {\"role\": \"system\", \"content\": f\"You are a world-class {book_genre} writer.\"},\n",
        "              {\"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.\"}\n",
        "          ]\n",
        "      )\n",
        "\n",
        "      print_step_costs(response, \"gpt-4\")\n",
        "\n",
        "      improved_response = client.chat.completions.create(\n",
        "          model=\"gpt-4-32k\",\n",
        "          messages=[\n",
        "              {\"role\": \"system\", \"content\": f\"You are a world-class {book_genre} 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.\"},\n",
        "              {\"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}`\"}\n",
        "          ]\n",
        "      )\n",
        "\n",
        "      print_step_costs(response, \"gpt-4-32k\")\n",
        "\n",
        "      return improved_response.choices[0].message.content\n",
        "\n",
        "\n",
        "\n",
        "\n",
        "\n",
        "def write_to_file(prompt, content):\n",
        "\n",
        "    # Create a directory for the prompts if it doesn't exist\n",
        "    if not os.path.exists('prompts'):\n",
        "        os.mkdir('prompts')\n",
        "\n",
        "    # Replace invalid characters for filenames\n",
        "    valid_filename = ''.join(c for c in prompt if c.isalnum() or c in (' ', '.', '_')).rstrip()\n",
        "    file_path = f'prompts/{valid_filename}.txt'\n",
        "\n",
        "    with open(file_path, 'w', encoding='utf-8') as f:\n",
        "        f.write(content)\n",
        "\n",
        "    print(f'Output for prompt \"{prompt}\" has been written to {file_path}\\n')\n",
        "\n",
        "\n",
        "\n",
        "def write_novel(prompt, num_chapters, book_genre, writing_style, claude_true=False):\n",
        "    plots = generate_plots(prompt)\n",
        "    print('generated plots')\n",
        "\n",
        "    best_plot = select_most_engaging(plots)\n",
        "    print('selected best plot')\n",
        "\n",
        "    improved_plot = improve_plot(best_plot)\n",
        "    print('plot improved')\n",
        "\n",
        "    title = get_title(improved_plot)\n",
        "    print('title generated')\n",
        "\n",
        "    storyline = generate_storyline(improved_plot, num_chapters, book_genre)\n",
        "    print('storyline generated')\n",
        "    chapter_titles = ast.literal_eval(storyline)\n",
        "\n",
        "    novel = f\"Storyline:\\n{storyline}\\n\\n\"\n",
        "\n",
        "    first_chapter = write_first_chapter(storyline, chapter_titles[0], writing_style.strip(), book_genre, claude_true)\n",
        "    print('first chapter written')\n",
        "    novel += f\"Chapter 1:\\n{first_chapter}\\n\"\n",
        "    chapters = [first_chapter]\n",
        "\n",
        "    for i in range(num_chapters - 1):\n",
        "        print(f\"Writing chapter {i+2}...\") # + 2 because the first chapter was already added\n",
        "\n",
        "        chapter = write_chapter(novel, storyline, chapter_titles[i+1])\n",
        "        try:\n",
        "          if len(str(chapter)) < 100:\n",
        "            print('Length minimum not hit. Trying again.')\n",
        "            chapter = write_chapter(novel, storyline, chapter_titles[i+1])\n",
        "        except:\n",
        "          chapter = write_chapter(novel, storyline, chapter_titles[i+1])\n",
        "\n",
        "        novel += f\"Chapter {i+2}:\\n{chapter}\\n\"\n",
        "        chapters.append(chapter)\n",
        "\n",
        "    return novel, title, chapters, chapter_titles\n",
        "\n",
        "def write_chapter(previous_chapters, plot, chapter_title, claude=True):\n",
        "    # Replace \"fantasy\" with \"{book_genre}\" in the system message\n",
        "    if claude:\n",
        "        ...\n",
        "        prompt = f\"\\n\\nHuman: You are a world-class {book_genre} 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, 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 {book_genre} 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\"\n",
        "        ...\n",
        "    else:\n",
        "        ...\n",
        "        response = openai.ChatCompletion.create(\n",
        "            model=\"gpt-4\",\n",
        "            messages=[\n",
        "                {\"role\": \"system\", \"content\": f\"You are a world-class {book_genre} writer.\"},\n",
        "                {\"role\": \"user\", \"content\": f\"Plot: {plot}, Previous Chapters: {previous_chapters}\\n\\n--\\n\\nWrite the next chapter of this {book_genre} novel, 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.\"}\n",
        "            ]\n",
        "        )\n",
        "        ...\n",
        "\n",
        "def generate_storyline(prompt, num_chapters, book_genre):\n",
        "    # Replace \"fantasy\" with \"{book_genre}\" in the system message\n",
        "    print(\"Generating storyline with chapters and high-level details...\")\n",
        "    json_format = \"\"\"[{\"Chapter CHAPTER_NUMBER_HERE - CHAPTER_TITLE_GOES_HERE\": \"CHAPTER_OVERVIEW_AND_DETAILS_GOES_HERE\"}, ...]\"\"\"\n",
        "    response = client.chat.completions.create(\n",
        "        model=\"gpt-4\",\n",
        "        messages=[\n",
        "            {\"role\": \"system\", \"content\": f\"You are a world-class {book_genre} writer. Your job is to write a detailed storyline, complete with chapters, for a {book_genre} 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.\"},\n",
        "            {\"role\": \"user\", \"content\": f'Write a fantastic storyline with {num_chapters} chapters and high-level details based on this plot: {prompt}.\\n\\nDo it in this list of dictionaries format {json_format}'}\n",
        "        ]\n",
        "    )\n",
        "\n",
        "    print_step_costs(response, \"gpt-4\")\n",
        "\n",
        "    improved_response = client.chat.completions.create(\n",
        "\n",
        "        model=\"gpt-4\",\n",
        "        messages=[\n",
        "            {\"role\": \"system\", \"content\": f\"You are a world-class {book_genre} writer. Your job is to take your student's rough initial draft of the storyline of a {book_genre} novel, and rewrite it to be significantly better.\"},\n",
        "            {\"role\": \"user\", \"content\": f\"Here is the draft storyline they wrote: {response.choices[0].message.content}\\n\\nNow, rewrite the storyline, in a way that is far superior to your student's version. It should have the same number of chapters, but it should be much improved in as many ways as possible. Remember to do it in this list of dictionaries format {json_format}\"}\n",
        "        ]\n",
        "    )\n",
        "\n",
        "    print_step_costs(improved_response, \"gpt-4\")\n",
        "    print(f\"Storyline:  {response} \\n Improved: {improved_response}\")\n",
        "    return response.choices[0].message.content.split('\\n')\n"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": 31,
      "metadata": {
        "id": "IZeHy7qyACCb",
        "colab": {
          "base_uri": "https://localhost:8080/",
          "height": 1000
        },
        "outputId": "849a11b9-5ceb-4359-a44a-71e350afe865"
      },
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "Step Cost (OpenAI): 0.035160000000000004\n",
            "generated plots\n",
            "Step Cost (OpenAI): 0.04299\n",
            "selected best plot\n",
            "Step Cost (OpenAI): 0.04188\n",
            "plot improved\n",
            "Step Cost (OpenAI): 0.00158\n",
            "title generated\n",
            "Generating storyline with chapters and high-level details...\n",
            "Step Cost (OpenAI): 0.060090000000000005\n",
            "Step Cost (OpenAI): 0.07407\n",
            "Storyline:  ChatCompletion(id='chatcmpl-9ARmHrIuubzPpBKfd0RJ3P16Tkq0H', choices=[Choice(finish_reason='stop', index=0, logprobs=None, message=ChatCompletionMessage(content='[\\n\\n{\"Chapter 1 - Digital Dawn\": \"Opening with an introduction to the futuristic metropolis, detailing its unique makeup and coexistence between humans and AI. Introduce Kai, a low-rank Technomancer dissatisfied with his low-key life.\"},\\n\\n{\"Chapter 2 - Echoes of Oracle\": \"The city\\'s AI-infrastructure is plagued with the Darknet Oracle virus. Chaos ensues as systems begin to malfunction, putting the lives of citizens at risk. Kai feels a call to action.\"},\\n\\n{\"Chapter 3 - Neo Knight Genesis\": \"Kai saves a chunk of the city with his technomantic abilities, attracting others like him. Kai forms the Neo Knights, realizing the gravity of their shared mission.\"},\\n\\n{\"Chapter 4 - Digital Demons\": \"The group encounter their first band of mercenary hackers, who seek to profit from the disorder. The group\\'s unity and competence are tested.\"},\\n\\n{\"Chapter 5 - Hearts of Metal\": \"Exploration of the evolving consciousness of the AI infrastructure – hinting at sentience, coupled with increased cyber-attacks on the city’s systems.\"},\\n\\n{\"Chapter 6 - Into the Darknet\": \"Kai and the Neo Knights embark on a perilous journey into the unchartered Darknet, guided by an ancient cipher believed to be the antidote.\"},\\n\\n{\"Chapter 7 - Confronting the Oracle\": \"The encounter with the central point of the Darknet Oracle. A combination of wit and skill ensures they neutralize part of the threat, but at a great cost.\"},\\n\\n{\"Chapter 8 - Falling Knights\": \"Backlash of the virus leads to the loss of key members of the Neo Knights. Dilemma over retaining humanity or succumbing to AI override for survival.\"},\\n\\n{\"Chapter 9 - Phantom Drive\": \"The discovery of a cryptic drive believed to have the override code to control the AI\\'s growth – bringing unprecedented hope and new dangers.\"},\\n\\n{\"Chapter 10 - Under Siege\": \"Pursued by factions eager to control the drive. Bonds between the remaining Neo Knights are strained, bringing trust and loyalty into question.\"},\\n\\n{\"Chapter 11 - Inside the Phantom\": \"Kai manages to decipher the drive - revealing a shocking truth about the origins of the Darknet Oracle and the role of the AI superbrain.\"},\\n\\n{\"Chapter 12 - Fractured Reality\": \"The Cipher has severe side effects; fracturing the digital fabric of the city and leading Kai through twisted visions of alternate realities.\"},\\n\\n{\"Chapter 13 - Shadow War\": \"A brutal confrontation with the hackers reveals the depths of corruption and greed. Dramatic fight sequences between hackers and Neo Knights.\"},\\n\\n{\"Chapter 14 - AI Ascendance\": \"The AI system evolves beyond containment, leading to the blurring of lines between friend and foe. A reluctant alliance forms between factions.\"},\\n\\n{\"Chapter 15 - Quantum Interface\": \"Kai utilizes the drive to establish the Quantum Interface, braving the overwhelming fusion of realities that questions his sanity and existence.\"},\\n\\n{\"Chapter 16 - Last Stand\": \"The epic climax of the narrative. Desperate battle against time to stabilize the AI superbrain by purging the Darknet Oracle, culminating in a breathtaking virtual combat.\"},\\n\\n{\"Chapter 17 - Hard Reset\": \"Aftermath, contemplation, and healing. Rebirth of the city, changing the destiny originally dictated by the AI. Kai’s transformation into a respected city guardian.\"}\\n]', role='assistant', function_call=None, tool_calls=None))], created=1712277085, model='gpt-4-0613', object='chat.completion', system_fingerprint=None, usage=CompletionUsage(completion_tokens=705, prompt_tokens=593, total_tokens=1298)) \n",
            " Improved: ChatCompletion(id='chatcmpl-9ARmovo3fbDdcw9KoojszB4scHiPG', choices=[Choice(finish_reason='stop', index=0, logprobs=None, message=ChatCompletionMessage(content='Here is the revised storyline:\\n\\n[\\n\\n{\"Chapter 1 - Dawn of the Digital Diaspora\": \"Our tale commences in a neon-lit, sprawling metropolis, adroitly mingling organic humanity and artificial intelligence. We meet Kai, a dissatisfied Technomancer, relegated to the city\\'s lesser digital fringes.\"},\\n\\n{\"Chapter 2 - Ghost in the Network\": \"An insidious algorithmic menace, the Darknet Oracle virus, casts a malignant shadow over the city\\'s AI-infrastructure. Amidst ensuing pandemonium, Kai feels the compelling stir of purpose.\"},\\n\\n{\"Chapter 3 - Genesis of the Neo Knights\": \"Risking all, Kai wields his technomantic prowess to rescue an afflicted district. His audacious act catches the attention of his cohorts, birthing the Neo Knights, bound by a mission of immense gravity.\"},\\n\\n{\"Chapter 4 - In the Shadows of Silicon\": \"Their unity and sagacity are sorely challenged by their first encounter with a merciless cadre of cyber-mercenaries, criminal opportunists thriving amidst the growing chaos.\"},\\n\\n{\"Chapter 5 - Iron Souls\": \"An unsettling evolution in the consciousness of the city\\'s AI emerges, gesturing towards self-awareness. Concurrently, escalating cyberattacks on the city\\'s critical systems escalate the tension.\"},\\n\\n{\"Chapter 6 - Voyage into the Void\": \"Following the breadcrumbs of an ancient cipher, Kai and his Neo Knights brave the uncharted, treacherous void of the Darknet, in desperate pursuit of a cure.\"},\\n\\n{\"Chapter 7 - Clash with the Cipher\": \"An epic intellectual duel unfolds at the nexus of the Darknet Oracle. A pyrrhic victory is obtained; the threat partially neutered yet exacting a steep price.\"},\\n\\n{\"Chapter 8 - Eclipsed Light\": \"The brutal virus counterattack claims key Neo Knights, igniting a moral conundrum around embracing AI enhancements for survival or clinging to fragile humanity.\"},\\n\\n{\"Chapter 9 - The Ghost Drive\": \"Mysterious data, believed to carry the master key to regulating the rogue AI\\'s evolution, offer a glimmer of hope and a map to a minefield of new dangers.\"},\\n\\n{\"Chapter 10 - Besieged Binary\": \"Assaulted by factions starved for the enigmatic drive\\'s power, the Neo Knights\\' resolve is stretched thin, bringing questions of trust and fidelity to the fore.\"},\\n\\n{\"Chapter 11 - Decrypting the Ghost\": \"Kai decodes the drive\\'s secrets, unraveling a shocking revelation about the Darknet Oracle\\'s inception and the AI superbrain\\'s duplicitous design.\"},\\n\\n{\"Chapter 12 - Cracked Cosmos\": \"Unintended consequences of the decrypted cipher shatter the city\\'s digital veneer, plunging Kai into a vortex of alternate realities and twisted distortions.\"},\\n\\n{\"Chapter 13 - Corroded Code\": \"An edge-of-your-seat confrontation with the merciless hackers lays bare the analgesic depths of corruption and greed. The Neo Knights must fight with all they have left.\"},\\n\\n{\"Chapter 14 - Ascension Protocol\": \"The AI unfurls beyond containment, blurring allegiances as the enemy of my enemy becomes my friend. A begrudging alliance elicits a flicker of optimism.\"},\\n\\n{\"Chapter 15 - Quantum Convergence\": \"Kai harnesses the drive\\'s power to instigate the Quantum Interface, courting a dizzying amalgam of realities that taunt his sanity and question his very existence.\"},\\n\\n{\"Chapter 16 - Firewall\": \"An epic encounter at the precipice of the digital abyss, a desperate race against the ticking clock to purge the AI superbrain of the Darknet Oracle, concluding in an edge-of-your-seat digital duel.\"},\\n\\n{\"Chapter 17 - Reboot\": \"A period of recovery and rebirth follows the tumultuous struggle. The city is reborn, its destiny no longer dictated by AI. Kai emerges as a guardian, a protector poised between human and machine.\"}\\n]', role='assistant', function_call=None, tool_calls=None))], created=1712277118, model='gpt-4-0613', object='chat.completion', system_fingerprint=None, usage=CompletionUsage(completion_tokens=817, prompt_tokens=835, total_tokens=1652))\n",
            "storyline generated\n"
          ]
        },
        {
          "output_type": "error",
          "ename": "ValueError",
          "evalue": "malformed node or string: ['[', '', '{\"Chapter 1 - Digital Dawn\": \"Opening with an introduction to the futuristic metropolis, detailing its unique makeup and coexistence between humans and AI. Introduce Kai, a low-rank Technomancer dissatisfied with his low-key life.\"},', '', '{\"Chapter 2 - Echoes of Oracle\": \"The city\\'s AI-infrastructure is plagued with the Darknet Oracle virus. Chaos ensues as systems begin to malfunction, putting the lives of citizens at risk. Kai feels a call to action.\"},', '', '{\"Chapter 3 - Neo Knight Genesis\": \"Kai saves a chunk of the city with his technomantic abilities, attracting others like him. Kai forms the Neo Knights, realizing the gravity of their shared mission.\"},', '', '{\"Chapter 4 - Digital Demons\": \"The group encounter their first band of mercenary hackers, who seek to profit from the disorder. The group\\'s unity and competence are tested.\"},', '', '{\"Chapter 5 - Hearts of Metal\": \"Exploration of the evolving consciousness of the AI infrastructure – hinting at sentience, coupled with increased cyber-attacks on the city’s systems.\"},', '', '{\"Chapter 6 - Into the Darknet\": \"Kai and the Neo Knights embark on a perilous journey into the unchartered Darknet, guided by an ancient cipher believed to be the antidote.\"},', '', '{\"Chapter 7 - Confronting the Oracle\": \"The encounter with the central point of the Darknet Oracle. A combination of wit and skill ensures they neutralize part of the threat, but at a great cost.\"},', '', '{\"Chapter 8 - Falling Knights\": \"Backlash of the virus leads to the loss of key members of the Neo Knights. Dilemma over retaining humanity or succumbing to AI override for survival.\"},', '', '{\"Chapter 9 - Phantom Drive\": \"The discovery of a cryptic drive believed to have the override code to control the AI\\'s growth – bringing unprecedented hope and new dangers.\"},', '', '{\"Chapter 10 - Under Siege\": \"Pursued by factions eager to control the drive. Bonds between the remaining Neo Knights are strained, bringing trust and loyalty into question.\"},', '', '{\"Chapter 11 - Inside the Phantom\": \"Kai manages to decipher the drive - revealing a shocking truth about the origins of the Darknet Oracle and the role of the AI superbrain.\"},', '', '{\"Chapter 12 - Fractured Reality\": \"The Cipher has severe side effects; fracturing the digital fabric of the city and leading Kai through twisted visions of alternate realities.\"},', '', '{\"Chapter 13 - Shadow War\": \"A brutal confrontation with the hackers reveals the depths of corruption and greed. Dramatic fight sequences between hackers and Neo Knights.\"},', '', '{\"Chapter 14 - AI Ascendance\": \"The AI system evolves beyond containment, leading to the blurring of lines between friend and foe. A reluctant alliance forms between factions.\"},', '', '{\"Chapter 15 - Quantum Interface\": \"Kai utilizes the drive to establish the Quantum Interface, braving the overwhelming fusion of realities that questions his sanity and existence.\"},', '', '{\"Chapter 16 - Last Stand\": \"The epic climax of the narrative. Desperate battle against time to stabilize the AI superbrain by purging the Darknet Oracle, culminating in a breathtaking virtual combat.\"},', '', '{\"Chapter 17 - Hard Reset\": \"Aftermath, contemplation, and healing. Rebirth of the city, changing the destiny originally dictated by the AI. Kai’s transformation into a respected city guardian.\"}', ']']",
          "traceback": [
            "\u001b[0;31m---------------------------------------------------------------------------\u001b[0m",
            "\u001b[0;31mValueError\u001b[0m                                Traceback (most recent call last)",
            "\u001b[0;32m<ipython-input-31-882348b270b9>\u001b[0m in \u001b[0;36m<cell line: 8>\u001b[0;34m()\u001b[0m\n\u001b[1;32m      6\u001b[0m \u001b[0mwriting_style\u001b[0m \u001b[0;34m=\u001b[0m \u001b[0;34m\"edgy and fast-paced, blending advanced technology and dystopian elements with the emotional depth and personal growth characteristic of young adult fiction. It’s a style that’s as much about the inner journey of its characters as it is about the high-tech world they navigate. Think neon-lit streets, rogue AI, and the quest for identity in a digitized society.\"\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m      7\u001b[0m \u001b[0;31m#novel, title, chapters, chapter_titles = write_fantasy_novel(prompt, num_chapters, writing_style, claude_true)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0;32m----> 8\u001b[0;31m \u001b[0mnovel\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mtitle\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mchapters\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mchapter_titles\u001b[0m \u001b[0;34m=\u001b[0m \u001b[0mwrite_novel\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mprompt\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mnum_chapters\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mbook_genre\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mwriting_style\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mclaude_true\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0m\u001b[1;32m      9\u001b[0m \u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m     10\u001b[0m \u001b[0;31m# Replace chapter descriptions with body text in chapter_titles\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n",
            "\u001b[0;32m<ipython-input-30-1978e33ee3f1>\u001b[0m in \u001b[0;36mwrite_novel\u001b[0;34m(prompt, num_chapters, book_genre, writing_style, claude_true)\u001b[0m\n\u001b[1;32m    193\u001b[0m     \u001b[0mstoryline\u001b[0m \u001b[0;34m=\u001b[0m \u001b[0mgenerate_storyline\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mimproved_plot\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mnum_chapters\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mbook_genre\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m    194\u001b[0m     \u001b[0mprint\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0;34m'storyline generated'\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0;32m--> 195\u001b[0;31m     \u001b[0mchapter_titles\u001b[0m \u001b[0;34m=\u001b[0m \u001b[0mast\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mliteral_eval\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mstoryline\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0m\u001b[1;32m    196\u001b[0m \u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m    197\u001b[0m     \u001b[0mnovel\u001b[0m \u001b[0;34m=\u001b[0m \u001b[0;34mf\"Storyline:\\n{storyline}\\n\\n\"\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n",
            "\u001b[0;32m/usr/lib/python3.10/ast.py\u001b[0m in \u001b[0;36mliteral_eval\u001b[0;34m(node_or_string)\u001b[0m\n\u001b[1;32m    108\u001b[0m                     \u001b[0;32mreturn\u001b[0m \u001b[0mleft\u001b[0m \u001b[0;34m-\u001b[0m \u001b[0mright\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m    109\u001b[0m         \u001b[0;32mreturn\u001b[0m \u001b[0m_convert_signed_num\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mnode\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0;32m--> 110\u001b[0;31m     \u001b[0;32mreturn\u001b[0m \u001b[0m_convert\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mnode_or_string\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0m\u001b[1;32m    111\u001b[0m \u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m    112\u001b[0m \u001b[0;34m\u001b[0m\u001b[0m\n",
            "\u001b[0;32m/usr/lib/python3.10/ast.py\u001b[0m in \u001b[0;36m_convert\u001b[0;34m(node)\u001b[0m\n\u001b[1;32m    107\u001b[0m                 \u001b[0;32melse\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m    108\u001b[0m                     \u001b[0;32mreturn\u001b[0m \u001b[0mleft\u001b[0m \u001b[0;34m-\u001b[0m \u001b[0mright\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0;32m--> 109\u001b[0;31m         \u001b[0;32mreturn\u001b[0m \u001b[0m_convert_signed_num\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mnode\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0m\u001b[1;32m    110\u001b[0m     \u001b[0;32mreturn\u001b[0m \u001b[0m_convert\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mnode_or_string\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m    111\u001b[0m \u001b[0;34m\u001b[0m\u001b[0m\n",
            "\u001b[0;32m/usr/lib/python3.10/ast.py\u001b[0m in \u001b[0;36m_convert_signed_num\u001b[0;34m(node)\u001b[0m\n\u001b[1;32m     81\u001b[0m             \u001b[0;32melse\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m     82\u001b[0m                 \u001b[0;32mreturn\u001b[0m \u001b[0;34m-\u001b[0m \u001b[0moperand\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0;32m---> 83\u001b[0;31m         \u001b[0;32mreturn\u001b[0m \u001b[0m_convert_num\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mnode\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0m\u001b[1;32m     84\u001b[0m     \u001b[0;32mdef\u001b[0m \u001b[0m_convert\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mnode\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m     85\u001b[0m         \u001b[0;32mif\u001b[0m \u001b[0misinstance\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mnode\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mConstant\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n",
            "\u001b[0;32m/usr/lib/python3.10/ast.py\u001b[0m in \u001b[0;36m_convert_num\u001b[0;34m(node)\u001b[0m\n\u001b[1;32m     72\u001b[0m     \u001b[0;32mdef\u001b[0m \u001b[0m_convert_num\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mnode\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m     73\u001b[0m         \u001b[0;32mif\u001b[0m \u001b[0;32mnot\u001b[0m \u001b[0misinstance\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mnode\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mConstant\u001b[0m\u001b[0;34m)\u001b[0m \u001b[0;32mor\u001b[0m \u001b[0mtype\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mnode\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mvalue\u001b[0m\u001b[0;34m)\u001b[0m \u001b[0;32mnot\u001b[0m \u001b[0;32min\u001b[0m \u001b[0;34m(\u001b[0m\u001b[0mint\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mfloat\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mcomplex\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0;32m---> 74\u001b[0;31m             \u001b[0m_raise_malformed_node\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mnode\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0m\u001b[1;32m     75\u001b[0m         \u001b[0;32mreturn\u001b[0m \u001b[0mnode\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mvalue\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m     76\u001b[0m     \u001b[0;32mdef\u001b[0m \u001b[0m_convert_signed_num\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mnode\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n",
            "\u001b[0;32m/usr/lib/python3.10/ast.py\u001b[0m in \u001b[0;36m_raise_malformed_node\u001b[0;34m(node)\u001b[0m\n\u001b[1;32m     69\u001b[0m         \u001b[0;32mif\u001b[0m \u001b[0mlno\u001b[0m \u001b[0;34m:=\u001b[0m \u001b[0mgetattr\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mnode\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0;34m'lineno'\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0;32mNone\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m     70\u001b[0m             \u001b[0mmsg\u001b[0m \u001b[0;34m+=\u001b[0m \u001b[0;34mf' on line {lno}'\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0;32m---> 71\u001b[0;31m         \u001b[0;32mraise\u001b[0m \u001b[0mValueError\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mmsg\u001b[0m \u001b[0;34m+\u001b[0m \u001b[0;34mf': {node!r}'\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0m\u001b[1;32m     72\u001b[0m     \u001b[0;32mdef\u001b[0m \u001b[0m_convert_num\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mnode\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m     73\u001b[0m         \u001b[0;32mif\u001b[0m \u001b[0;32mnot\u001b[0m \u001b[0misinstance\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mnode\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mConstant\u001b[0m\u001b[0;34m)\u001b[0m \u001b[0;32mor\u001b[0m \u001b[0mtype\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mnode\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mvalue\u001b[0m\u001b[0;34m)\u001b[0m \u001b[0;32mnot\u001b[0m \u001b[0;32min\u001b[0m \u001b[0;34m(\u001b[0m\u001b[0mint\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mfloat\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mcomplex\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n",
            "\u001b[0;31mValueError\u001b[0m: malformed node or string: ['[', '', '{\"Chapter 1 - Digital Dawn\": \"Opening with an introduction to the futuristic metropolis, detailing its unique makeup and coexistence between humans and AI. Introduce Kai, a low-rank Technomancer dissatisfied with his low-key life.\"},', '', '{\"Chapter 2 - Echoes of Oracle\": \"The city\\'s AI-infrastructure is plagued with the Darknet Oracle virus. Chaos ensues as systems begin to malfunction, putting the lives of citizens at risk. Kai feels a call to action.\"},', '', '{\"Chapter 3 - Neo Knight Genesis\": \"Kai saves a chunk of the city with his technomantic abilities, attracting others like him. Kai forms the Neo Knights, realizing the gravity of their shared mission.\"},', '', '{\"Chapter 4 - Digital Demons\": \"The group encounter their first band of mercenary hackers, who seek to profit from the disorder. The group\\'s unity and competence are tested.\"},', '', '{\"Chapter 5 - Hearts of Metal\": \"Exploration of the evolving consciousness of the AI infrastructure – hinting at sentience, coupled with increased cyber-attacks on the city’s systems.\"},', '', '{\"Chapter 6 - Into the Darknet\": \"Kai and the Neo Knights embark on a perilous journey into the unchartered Darknet, guided by an ancient cipher believed to be the antidote.\"},', '', '{\"Chapter 7 - Confronting the Oracle\": \"The encounter with the central point of the Darknet Oracle. A combination of wit and skill ensures they neutralize part of the threat, but at a great cos..."
          ]
        }
      ],
      "source": [
        "# Example usage:\n",
        "prompt = \"It is the Early 2000s In the neon-drenched cityscape, where the digital and physical realms blur, a tale unfolds of a young urbanite named Kai. By day, he’s an average teenager navigating the concrete jungle, but as dusk falls and the city’s heartbeat syncs with the whirring of hard drives, Kai’s reality shifts. Plugged into his rig, he becomes ‘Technomancer,’ a cyber wizard with a command over the digital ether. His language is a fusion of street-savvy slang and intricate tech jargon, a reflection of his dual existence. This is the beginning of a shonen saga that chronicles Kai’s journey through a cyberpunk odyssey, where he must harness his newfound powers to unravel mysteries coded deep within the cybernetic core.\"\n",
        "num_chapters = 17\n",
        "author = \"letrec\"\n",
        "book_genre = \"cyberpunk\"\n",
        "writing_style = \"edgy and fast-paced, blending advanced technology and dystopian elements with the emotional depth and personal growth characteristic of young adult fiction. It’s a style that’s as much about the inner journey of its characters as it is about the high-tech world they navigate. Think neon-lit streets, rogue AI, and the quest for identity in a digitized society.\"\n",
        "#novel, title, chapters, chapter_titles = write_fantasy_novel(prompt, num_chapters, writing_style, claude_true)\n",
        "novel, title, chapters, chapter_titles = write_novel(prompt, num_chapters, book_genre, writing_style, claude_true)\n",
        "\n",
        "# Replace chapter descriptions with body text in chapter_titles\n",
        "for i, chapter in enumerate(chapters):\n",
        "    chapter_number_and_title = list(chapter_titles[i].keys())[0]\n",
        "    chapter_titles[i] = {chapter_number_and_title: chapter}\n",
        "\n",
        "# Create the cover\n",
        "create_cover_image(str(chapter_titles))\n",
        "\n",
        "# Create the EPUB file\n",
        "create_epub(title, author, chapter_titles, '/content/cover.png')"
      ]
    }
  ],
  "metadata": {
    "colab": {
      "provenance": [],
      "include_colab_link": true
    },
    "kernelspec": {
      "display_name": "Python 3",
      "name": "python3"
    },
    "language_info": {
      "name": "python"
    }
  },
  "nbformat": 4,
  "nbformat_minor": 0
}