full text experience including ending the story

README.md

@@ -71,8 +71,8 @@ It puts the user in charge of a how the story is going to develop.
 - [x] Story object setup, schema, logic
 - [x] Set up flow management
 - [x] Add SQlite DB and save stories
+- [x] GPT-4 story generation in a gradio interface
 - [ ] Do the evaluator (if it's time to end)
-- [ ] GPT-4 story generation in a gradio interface
 - [ ] Inerchange text output for play.ht voice generation
 - [ ] Interchange text input for whisper
 - [ ] Dockerfile and deploy (including magic word for access control)

helpers/__init__.py

@@ -10,9 +10,6 @@ def gen_unique_id():
 def check_magic_word(w):
     return magic_w in w
 
-def its_time_to_end(story):
-    return False
-
 def get_fixed_msg(msg_type):
 
     fixed_sayings = {
@@ -24,7 +21,7 @@ def get_fixed_msg(msg_type):
         "ask_ending": ["What kind of ending would you like? A happy one? A tragic one? Something else?"],
         "ask_style": ["What kind of storytelling style do you wnat? Funny? Poetic?"],
         "its_the_end": ["And this is the end of our story. Thank you for listening."],
-        "no_more_story": ["I'm sorry, this story has ended."]
+        "no_more_story": ["I'm sorry, this story has ended. Reload the page to do another story."]
     }
 
     if msg_type in fixed_sayings:

homeros.py

@@ -7,6 +7,7 @@ from helpers import gen_unique_id
 import prompts
 from adaptors.llm import answer
 
+MAX_STORY_LEN = 2 #after how many chunks we force the story to end
 
 '''
 initiates a new story and saves in DB
@@ -69,18 +70,18 @@ def continue_story(user_input, story_data):
 '''
 generates the last part of the story and changes status to "finished"
 '''
-def finish_story(story_data):
+def finish_story(user_input, story_data):
     story = get_story(story_data["uuid"])
     chunks = json.loads(story.chunks)
     messages = json.loads(story.messages)
 
-    user_input = "Please finish the story now."
+    user_input = user_input + "\n\nPlease finish the story now."
 
     messages.append({
         "role":"user",
         "content": user_input
     })
-    next_chunk_text = create_next_chunk_text(story)
+    next_chunk_text = create_next_chunk_text(user_input, story)
     next_chunk_audio = create_next_chunk_audio(next_chunk_text)
     chunks.append({
         "text" : next_chunk_text,
@@ -127,6 +128,18 @@ def create_next_chunk_text(user_input, story):
     return next_chunk
 
 
+'''
+evaluates the story up until now and returns a dict with the result of the evaluation
+'''
+def evaluate_story(story):
+    evaluation = {}
+    story_len = len(story["chunks"])
+    logger.debug(story_len)
+    evaluation["is_time_to_end"] = story_len > MAX_STORY_LEN
+
+    return evaluation
+
+
 '''
 turns next story chunk into audio and returns a URL
 '''

main.py

@@ -6,7 +6,7 @@ from dotenv import load_dotenv
 load_dotenv()
 import helpers
 
-from homeros import init_story, start_story, continue_story, finish_story, define_metadata
+from homeros import init_story, start_story, continue_story, finish_story, define_metadata, evaluate_story
 
 '''
 Here we manage the flow and state of the story
@@ -71,9 +71,9 @@ def do_homeros(user_input, story):
 
     # we are in the middle of the story - evaluate if time to end, or continue
     elif story["status"] == "ongoing":
-        if helpers.its_time_to_end(story):
+        if evaluate_story(story)["is_time_to_end"]:
             logger.debug("status: activating story finish")
-            story = finish_story(story)
+            story = finish_story(user_input, story)
             story["status"] = "finished"
         else:
             story = continue_story(user_input, story)
@@ -82,7 +82,7 @@ def do_homeros(user_input, story):
         next_message = story["chunks"][-1]["text"]
 
     # story has ended, but the user still inputting. tell them it's over
-    elif story["status"] == "ended":
+    elif story["status"] == "finished":
         next_message = helpers.get_fixed_msg("no_more_story")
         story["status"] = "finished"