Storing and Retrieving Generated Stories

Welcome to the second unit of our course, where we will focus on building the Story Manager for our short story generation service. In the previous lesson, we laid the foundation by creating a structured base prompt using a dedicated class for managing prompts. Now, we will take the next step by developing a system to manage and store the stories generated by our service. The Story Manager will allow us to save, retrieve, and organize stories, making it a crucial component of our application.

Before we dive into building the Story Manager, let's briefly recall some fundamental concepts that will be useful. A class is a way to group related data and functionality together, providing a blueprint for creating objects that can manage and organize information. We will use a collection to store our stories, which allows us to keep track of multiple stories efficiently. This approach will help us manage the stories generated by our service in an organized manner.

Let's start by defining the StoryManager class. This class will be responsible for managing our stories.

class StoryManager:
    def __init__(self):
        self.stories = [
            {
                "id": 0,
                "prompt": "This is a test prompt",
                "content": "This is a test story"
            }
        ]

Here, we define a class named StoryManager. Inside the class, we have a constructor that initializes a collection to store our stories. This collection starts with a single example story. We dont need to keep track of the next available unique identifier for new stories, as its already stored as the length of our array. This setup ensures that each story we add will have a unique ID, as well as having a story present in our tasks without generating one insid each task.

Note that this implementation keeps stories in memory only during the program’s execution. Once the program stops, the stored stories are lost. For long-term persistence, we would later need to store them in a database or file.

Next, we need a method to add stories to our manager. Let's create the add_story method.

    def add_story(self, prompt, story):
        """Save story and return the id"""
        self.stories.append({"id": len(self.stories), "prompt": prompt, "content": story})
        return self.stories[-1]

The add_story method takes two parameters: prompt and content. It creates a new story with a unique identifier based around the number of stories we have, appends it to the collection of stories, and returns the newly added story (the last one in our array). This ensures that each story is stored with its prompt and content and can be referenced by its unique ID.