Update archives/modified_version.py

archives/modified_version.py

@@ -6,6 +6,10 @@ from sentence_transformers import SentenceTransformer
 from utils import get_init, parse_instructions
 import re
 
+###################################################################################################################################################################
+###################################################################################################################################################################
+# TODO: add user database
+
 # from urllib.parse import quote_plus
 # from pymongo import MongoClient
 
@@ -15,27 +19,31 @@ import re
 # db = client.recurrentGPT_db
 # log = db.log
 
-_CACHE = {}
+###################################################################################################################################################################
+###################################################################################################################################################################
 
+_CACHE = {}
 
-# Build the semantic search model
+# TODO : try with different embedding models
+## Build the semantic search model
 embedder = SentenceTransformer('multi-qa-mpnet-base-cos-v1')
 
-def init_prompt(novel_type, description):
+# TODO: make a prompt class
+def init_prompt(text_type, description, structure, units, tone):
     if description == "":
         description = ""
     else:
         description = " about " + description
     return f"""
-Please write a {novel_type} novel{description} with 50 chapters. Follow the format below precisely:
-Begin with the name of the novel.
-Next, write an outline for the first chapter. The outline should describe the background and the beginning of the novel.
-Write the first three paragraphs with their indication of the novel based on your outline. Write in a novelistic style and take your time to set the scene.
+Please write a {text_type} about {description} with {units} {structure}. Follow the format below precisely:
+Begin with the title of the {text_type}.
+Next, write an outline for the first {structure}. The outline should describe the {description} and the beginning of the {text_type}.
+Write the first three paragraphs with their indication of the {text_type} based on your outline. Write in a {tone} style and take your time to set the context.
 Write a summary that captures the key information of the three paragraphs.
-Finally, write three different instructions for what to write next, each containing around five sentences. Each instruction should present a possible, interesting continuation of the story.
+Finally, write three different instructions for what to write next, each containing around five sentences. Each instruction should present a possible, interesting continuation of the {text_type}.
 The output format should follow these guidelines:
-Name: <name of the novel>
-Outline: <outline for the first chapter>
+Name: <name of the {text_type}>
+Outline: <outline for the first {structure}>
 Paragraph 1: <content for paragraph 1>
 Paragraph 2: <content for paragraph 2>
 Paragraph 3: <content for paragraph 3>
@@ -46,14 +54,14 @@ Instruction 3: <content for instruction 3>
 Make sure to be precise and follow the output format strictly.
 """
 
-def init(novel_type, description, request: gr.Request):
-    if novel_type == "":
-        novel_type = "Science Fiction"
+def init(text_type, description, request: gr.Request):
+    if tex_type == "":
+        text_type = "Linkedin post"
     global _CACHE
     cookie = request.headers['cookie']
     cookie = cookie.split('; _gat_gtag')[0]
     # prepare first init
-    init_paragraphs = get_init(text=init_prompt(novel_type,description))
+    init_paragraphs = get_init(text=init_prompt(text_type, description, structure, units, tone))
     # print(init_paragraphs)
     start_input_to_human = {
         'output_paragraph': init_paragraphs['Paragraph 3'],
@@ -64,7 +72,7 @@ def init(novel_type, description, request: gr.Request):
 
     _CACHE[cookie] = {"start_input_to_human": start_input_to_human,
                       "init_paragraphs": init_paragraphs}
-    written_paras = f"""Title: {init_paragraphs['name']}
+    written_paras = f"""Title: {init_paragraphs['name']} 
 Outline: {init_paragraphs['Outline']}
 Paragraphs:
 {start_input_to_human['input_paragraph']}"""