Refactoring and regenerate name

gradio_server.py

@@ -27,18 +27,24 @@ def validate_inputs(model_state):
         raise gr.Error("Please set the API key!")
 
 
-def generate_plan(novel_type, description, model_state):
+def generate_name(state, model_state):
+    writer = RecurrentGPT(model_state)
+    state.name = writer.generate_name(state)
+    return (state, state.name)
+
+
+def generate_meta(novel_type, description, model_state):
     validate_inputs(model_state)
     writer = RecurrentGPT(model_state)
-    state = writer.generate_plan(novel_type=novel_type, description=description)
-    return (state, state.name, state.language, state.synopsis, state.plan)
+    state = writer.generate_meta(novel_type=novel_type, description=description)
+    return (state, state.name, state.language, state.synopsis, state.outline)
 
 
-def generate_first_paragraphs(state, model_state):
+def generate_first_step(state, model_state):
     assert state is not None
     validate_inputs(model_state)
     writer = RecurrentGPT(model_state)
-    state = writer.generate_first_paragraphs(state)
+    state = writer.generate_first_step(state)
     return (
         state,
         state.short_memory,
@@ -96,7 +102,7 @@ def load(file_name):
         state,
         state.name,
         state.synopsis,
-        state.plan,
+        state.outline,
         state.short_memory,
         "\n\n".join(state.paragraphs),
         state.next_instructions[0],
@@ -193,6 +199,10 @@ def on_prompt_template_name_select(model_state, prompt_template, evt: gr.SelectD
                     max_lines=1,
                     lines=1
                 )
+                btn_gen_name = gr.Button(
+                    "Regenerate",
+                    variant="primary"
+                )
                 language = gr.Textbox(
                     label="Language (editable)",
                     max_lines=1,
@@ -204,7 +214,7 @@ def on_prompt_template_name_select(model_state, prompt_template, evt: gr.SelectD
                     lines=8
                 )
             with gr.Column(scale=9):
-                plan = gr.Textbox(
+                outline = gr.Textbox(
                     label="Outline (editable)",
                     lines=17,
                     max_lines=17
@@ -336,7 +346,7 @@ def on_prompt_template_name_select(model_state, prompt_template, evt: gr.SelectD
                         value=model_state.value.generation_params.repetition_penalty,
                         step=0.05,
                         interactive=True,
-                        label="Rep"
+                        label="Repetition penalty"
                     )
                 with gr.Column(scale=1, min_width=200):
                     top_p = gr.Slider(
@@ -367,7 +377,7 @@ def set_field(state, field, value):
         "description": description,
         "novel_type": novel_type,
         "synopsis": synopsis,
-        "plan": plan,
+        "outline": outline,
         "instruction": instruction,
         "short_memory": short_memory
     }
@@ -404,11 +414,11 @@ def set_param(state, field, value):
 
     # Main events
     btn_init.click(
-        generate_plan,
+        generate_meta,
         inputs=[novel_type, description, model_state],
-        outputs=[state, name, language, synopsis, plan]
+        outputs=[state, name, language, synopsis, outline]
     ).success(
-        generate_first_paragraphs,
+        generate_first_step,
         inputs=[state, model_state],
         outputs=[state, short_memory, paragraphs, instruction1, instruction2, instruction3]
     )
@@ -426,6 +436,11 @@ def set_param(state, field, value):
             instruction
         ]
     )
+    btn_gen_name.click(
+        generate_name,
+        inputs=[state, model_state],
+        outputs=[state, name]
+    )
 
     # Save/Load
     btn_save.click(
@@ -461,7 +476,7 @@ def set_param(state, field, value):
             state,
             name,
             synopsis,
-            plan,
+            outline,
             short_memory,
             paragraphs,
             instruction1,
@@ -480,7 +495,7 @@ def set_param(state, field, value):
             state,
             name,
             synopsis,
-            plan,
+            outline,
             short_memory,
             paragraphs,
             instruction1,

tale_studio/prompts/begin.jinja → tale_studio/prompts/first_paragraphs.jinja

@@ -10,7 +10,7 @@ Synopsis:
 {{synopsis}}
 
 Outline:
-{{plan}}
+{{outline}}
 
 Remember, you are an author narrating events. Events should be narrated in the third person limited perspective and contain dialogues between the characters present. Do not conclude your output, leave it open for continuation. Write in a novelistic style and take your time to set the scene.
 Write 3 paragraphs below:

tale_studio/prompts/process.jinja → tale_studio/prompts/first_summary.jinja

@@ -2,7 +2,7 @@ Write a summary that captures the critical information of the paragraphs.
 
 Then write 3 different instructions for what to write next, each containing around 5 sentences.
 
-Each instruction should present a possible, exciting story continuation that fits the global plan presented below.
+Each instruction should present a possible, exciting story continuation that fits the global outline presented below.
 
 The output should be in JSON with the following fields:
 {
@@ -17,7 +17,7 @@ Synopsis:
 {{synopsis}}
 
 Outline:
-{{plan}}
+{{outline}}
 
 Paragraphs:
 {{paragraphs}}

tale_studio/prompts/instruct.jinja

@@ -1,5 +1,5 @@
 You should output 3 different instructions, each is a possible interesting continuation of the story.
-Each output instruction should contain around 5 sentences and fit the global plan.
+Each output instruction should contain around 5 sentences and fit the global outline.
 Think about what plot can be attractive for common readers when writing output instructions.
 
 Write and organize your output by strictly following the format below using JSON.
@@ -14,7 +14,7 @@ Use strictly this language: {{language}}
 
 Here is the outline:
 ####
-{{plan}}
+{{outline}}
 ####
 
 Previous short summary:

tale_studio/prompts/plan.jinja → tale_studio/prompts/meta.jinja

@@ -3,9 +3,9 @@ Please write a detailed outline for a {% if not novel_type %}Science Fiction{% e
 
 Begin with a catchy name of the novel and determine its main language.
 Then, generate a summary of the whole novel containing its main idea.
-Then, plan all 6 or 7 chapters of the novel.
-The story should have an ending.
+Then, write an outline with all 6 or 7 chapters of the novel.
 Chapter summaries should contain at least 2 sentences.
+The story should have an ending.
 Write in a novelistic style.
 Write all the fields in the selected langauge.
 
@@ -14,9 +14,17 @@ The output should be in JSON with the following fields:
     "language": <language of the novel>,
     "name": <name of the novel>,
     "synopsis": <synopsis of the whole novel>,
-    "chapter_summaries": [
-        "Chapter 1. <summary of chapter 1>",
-        ...,
-        "Chapter 6. <summary of chapter 6>"
+    "outline": [
+        {
+            "index": 1,
+            "chapter_name": <chapter_name>,
+            "chapter_summary": <chapter_summary>,
+        },
+        ...
+        {
+            "index": 6,
+            "chapter_name": <chapter_name>,
+            "chapter_summary": <chapter_summary>,
+        }
     ]
 }

tale_studio/prompts/name.jinja

@@ -0,0 +1,12 @@
+Generate a name for the novel.
+
+Idea: {{description}}
+Novel type: {{novel_type}}
+Synopsis: {{synopsis}}
+Outline: {{outline}}
+Language: {{language}}
+
+Format output as a JSON with one key, "name":
+{
+    "name": "..."
+}

tale_studio/prompts/output.jinja

@@ -9,7 +9,7 @@ Use strictly this language: {{language}}. Do not switch to another language.
 Do not write any chapter numbers.
 
 Outline of possible past and future events:
-{{plan}}
+{{outline}}
 
 Summary of previous events:
 {{short_memory}}