Creating the Story Controller

Welcome to the final lesson of our course on building a short story generation service with Flask. In this lesson, we will focus on creating the Story Generator Controller, a crucial component that manages requests and responses in our application. The controller acts as a bridge between the user input and the story generation logic, ensuring smooth communication within the application. By the end of this lesson, you will understand how to implement and test the controller, completing the core functionality of our service.

Before we dive into the details of the controller, let's briefly recall how the StoryGeneratorService and StoryManager are used in our application. These components are responsible for generating stories and managing them, respectively. The controller will interact with these components to process user requests and return appropriate responses. Remember, the controller is the part of the application that handles incoming HTTP requests, calls the necessary services, and returns the results to the client.

Let's start by understanding the StoryGeneratorController class. This class is responsible for handling story-related requests. Here's how we define it:

from flask import jsonify
from services.story_generator_service import StoryGeneratorService

class StoryGeneratorController:
    def __init__(self):
        self.story_generator_service = StoryGeneratorService()

• We import jsonify from Flask to format our responses as JSON, which is a common data format for web applications.
• We also import the StoryGeneratorService, which we will use to generate stories.
• The __init__ method initializes the controller by creating an instance of StoryGeneratorService. This allows the controller to access the story generation logic.

Next, let's implement the generate_story method, which handles the generation of a new story based on user input.

def generate_story(self, user_input):
    if not user_input:
        return {'error': 'Missing input'}, 400
    try:
        story = self.story_generator_service.generate_story(user_input)
        return {'story': story}
    except Exception as e:
        return {'error': str(e)}, 500

• The method takes user_input as a parameter. This input is essential for generating a story.
• We first check if the user_input is provided. If not, we return an error message with a 400 status code, indicating a bad request.
• If the input is valid, we call the generate_story method from the StoryGeneratorService to generate a story.
• We handle any exceptions that might occur during story generation and return an error message with a 500 status code, indicating a server error.