extract plot information from the summary text

Signed-off-by: Xe Iaso <me@xeiaso.net>

index.ts

@@ -20,14 +20,15 @@ const configuration = new Configuration({
 });
 const openai = new OpenAIApi(configuration);
 
-plot.cast.forEach(async (ch) => {});
-
 const promptBase = `Write me the following about the following plot summary for a novel:
 
-- A two word title for the novel starting with "Title: " and followed by two newlines.
+- A two to five word title for the novel starting with "Title: " and followed by two newlines. For example: "Fresh Beginnings" or "Jared's Adventure through Crime".
-- A detailed plot summary for the story starting with "Plot Summary: " and followed by two newlines.
+- A detailed plot summary for the story starting with "Plot Summary: " and followed by two newlines. The plot summary should be on the same line as the prefix.
 - The string "Chapter Summaries" followed by two newlines.
-- A markdown list of detailed chapter summaries in at least 3 sentences and titles for each of the 10 chapters that a novel based on the plot summary would have. Surround each chapter title in quotes and put a dash after the name like this: "Chapter name" - Chapter summary goes here.`;
+- A markdown list of detailed chapter summaries in at least 3 sentences and titles for each of the 10 chapters that a novel based on the plot summary would have. Surround each chapter title in quotes and put a dash after the name like this:
+
+- Chapter name - Chapter summary goes here. More words in the summary go here.
+- Second chapter name - Second chapter summary goes here.`;
 
 const summary = await openai.createChatCompletion({
   model: "gpt-3.5-turbo",
@@ -46,5 +47,43 @@ if (!!summary.data.usage) {
   );
 }
 
-console.log(summary.data.choices[0].message?.content);
+const summaryText = summary.data.choices[0].message?.content;
-await fs.writeFile(`${dirName}/summary.txt`, summary.data.choices[0].message?.content as string);
+
+console.log(summaryText);
+await fs.writeFile(`${dirName}/summary.txt`, summaryText as string);
+
+const titleRegex = /^Title: (.+)$/gm;
+const plotSummaryRegex = /^Plot Summary: (.+)$/gm;
+const chapterSummaryRegex = /^- (.+) - (.+)$/gm;
+
+let title = summaryText?.split("\n", 2)[0].split(titleRegex)[1] as string;
+
+if (title[0] === '"') {
+  title = title.slice(1, -1);
+}
+
+const chapterList = summaryText
+  ?.split("\n\n")
+  .slice(-1)[0]
+  .split("\n")
+  .map((line) => {
+    return line.split(chapterSummaryRegex);
+  })
+  .map((ch) => {
+    ch.shift();
+    ch.pop();
+    return { title: ch[0].slice(1, -1), summary: ch[1] };
+  });
+
+const plotSummary = summaryText?.split("\n\n", 3)[1].split(plotSummaryRegex)[1];
+
+plot.cast.forEach(async (ch) => {});
+
+const bookInfo = {
+  title,
+  chapterList,
+  plotSummary,
+};
+
+console.log(bookInfo);
+await fs.writeFile(`${dirName}/summary.json`, JSON.stringify(bookInfo));