progress on defining the main story state flow

README.md

@@ -34,7 +34,7 @@ It puts the user in charge of a how the story is going to develop.
 ## Story schema
 
 - STRING `uuid` = uuid of this story
-- STRING `status` = 'not_started' / 'generating_metadata' / ongoing' / 'finished'
+- STRING `status` = 'not_started' / 'checking_magic_word' / 'defining_metadata' / 'ongoing' / 'finished' ...etc.
 - TEXT `world` = text description of the world
 - TEXT `hero` = text description of the hero of the story
 - TEXT `plot` = high level description of the plot. without chapters or anything like that. we can use this to later break down into chapters and get smarter about story ark management with a second LLM

adaptors/voice.py

@@ -10,8 +10,8 @@ from pprint import pprint
 import json
 
 url = "https://play.ht/api/v2/tts"
-user_id = os.environ['PLAYHT_USERID']
-api_key = os.environ['PLAYHT_SECRETKEY']
+user_id = os.environ["PLAYHT_USERID"]
+api_key = os.environ["PLAYHT_SECRETKEY"]
 
 headers = {
   "accept": "text/event-stream",
@@ -20,7 +20,33 @@ headers = {
   "X-USER-ID": user_id
 }
 
-def say_it(text, voice):
+voices = ["dylan"]
+
+#TODO pre-generate these with dylan and save locally
+fixed_sayings = {
+    "welcome": [""], #also ask magic word here
+    "wrong_magic_word": [""],
+    "let_me_ask_questions": [""],
+    "ask_world": [""],
+    "ask_hero": [""],
+    "ask_plot": [""],
+    "ask_ending": [""],
+    "ask_style": [""],
+    "its_the_end": [""],
+    "no_more_story": [""]
+}
+
+'''
+return an old file from a dictionary of pre-generated sayings
+'''
+def say_fixed(fixed_msg, voice):
+    #We need to keep a repository of fixed messages generated in our key voices and for each have a file URL
+    return
+
+'''
+generate new saying with play.ht
+'''
+def say_new(text, voice):
   payload = {
     "quality": "medium",
     "output_format": "mp3",
@@ -32,7 +58,7 @@ def say_it(text, voice):
 
   response = requests.post(url, stream=True, headers=headers, json=payload)
 
-  stream_url = response.headers['content-location']
+  stream_url = response.headers["content-location"]
   logger.debug(f"stream_url = {stream_url}")
 
   resp = requests.get(stream_url, stream=True, headers=headers)

helpers/__init__.py

@@ -1,4 +1,32 @@
 import uuid
+import os
+import random
+
+magic_w = os.environ["MAGICWORD"]
 
 def gen_unique_id():
     return str(uuid.uuid4())
+
+def check_magic_word(w):
+    return magic_w in w
+
+def get_fixed_msg(msg_type):
+
+    fixed_sayings = {
+        "welcome": ["Welcome! What's the magic word?"], #also ask magic word here
+        "wrong_magic_word": ["Magic word is wrong. Try again."],
+        "ask_world": ["Let me ask you some questions first. What kind of world should your story unfold in?"],
+        "ask_hero": ["Who should the hero be?"],
+        "ask_plot": ["Can you describe the plot in a few words?"],
+        "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."]
+    }
+
+    if msg_type in fixed_sayings:
+        saying = random.choice(fixed_sayings[msg_type])
+        return saying
+
+    else:
+        raise Exception(f"fixed saying with msg_type {msg_type} not found")

homeros.py

@@ -1,3 +1,6 @@
+def init_story(story):
+    return story
+
 def start_story(story):
     return story
 

main.py

@@ -3,23 +3,93 @@ from pprint import pprint
 import json
 from loguru import logger
 from dotenv import load_dotenv
-import helpers
 load_dotenv()
+import helpers
 
-from homeros import start_story, continue_story, finish_story, define_metadata
+from homeros import init_story, start_story, continue_story, finish_story, define_metadata
 
+'''
+Here we manage the flow and state of the story
+'''
 def do_homeros(user_input, story):
 
     pprint(story)
 
+    # story hasn't started
     if story["status"] == "not_started":
-        story["uuid"] = helpers.gen_unique_id()
-        story["status"] = "generating_metadata"
+        logger.debug("status: initiating a new story")
+        next_message = helpers.get_fixed_msg("welcome")
+        story = init_story(story)
+        story["status"] = "checking_magic_word"
+
+    # we are checking the magic word or it is wrong and we need to ask for it again
+    elif story["status"] == "checking_magic_word" or story["status"] == "wrong_magic_word":
+        logger.debug("status: checking magic word")
+        magic_word_correct = helpers.check_magic_word(user_input)
+        if magic_word_correct:
+            story["status"] = "defining_metadata_world"
+            next_message = helpers.get_fixed_msg("ask_world")
+        else:
+            story["status"] = "wrong_magic_word"
+            next_message = helpers.get_fixed_msg("wrong_magic_word")
+
+    # defining the world
+    elif story["status"] == "defining_metadata_world":
+        logger.debug("status: magic word is wrong")
+        story = define_metadata(user_input, "world", story)
+        story["status"] = "defining_metadata_hero"
+        next_message = helpers.get_fixed_msg("ask_hero")
+
+    # defining the hero
+    elif story["status"] == "defining_metadata_hero":
+        logger.debug("status: defining the hero")
+        story = define_metadata(user_input, "hero", story)
+        story["status"] = "defining_metadata_plot"
+        next_message = helpers.get_fixed_msg("ask_plot")
+
+    # defining the plot
+    elif story["status"] == "defining_metadata_plot":
+        logger.debug("status: defining the plot")
+        story = define_metadata(user_input, "plot", story)
+        story["status"] = "defining_metadata_ending"
+        next_message = helpers.get_fixed_msg("ask_ending")
+
+    # defining the ending
+    elif story["status"] == "defining_metadata_ending":
+        logger.debug("status: defining the ending")
+        story = define_metadata(user_input, "ending", story)
+        story["status"] = "defining_metadata_style"
+        next_message = helpers.get_fixed_msg("ask_style")
+
+    # defining the style and starting the story with the first chunk
+    elif story["status"] == "defining_metadata_style":
+        logger.debug("status: defining the style")
+        story = define_metadata(user_input, "style", story)
+        story["status"] = "ongoing"
+        story = start_story(story)
+        next_message = story["chunks"][-1]
+
+    # 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):
+            story = finish_story(story)
+            story["status"] = "finished"
+        else:
+            story = continue_story(story)
+            story["status"] = "ongoing"
+        next_message = story["chunks"][-1]
+
+    # story has ended, but the user still inputting. tell them it's over
+    elif story["status"] == "ended":
+        next_message = helpers.get_fixed_msg("no_more_story")
+        story["status"] = "finished"
 
     else:
-        story["messages"].append("blah lbah n")
+        raise Exception("strange story status")
+        logger.error(f"we have a story status {story['status']} we didn't catch...")
+        logger.debug(story)
 
-    return json.dumps(story["messages"]), story
+    return next_message, story
 
 demo = gr.Blocks()