Welcome to the final lesson of our course on building an image generation service with Go! In our previous lessons, we've built several key components of our application: the PromptManager for formatting user inputs, the ImageManager for storing and processing images, the ImageGeneratorService for connecting to Google's Gemini API, and most recently, the ImageGeneratorHandler, which handles input validation and response formatting.

Now it's time to bring everything together by creating a Go web server that will expose our functionality through HTTP. This is the final piece of our backend architecture that will allow users to interact with our image generation service through a web interface.

In the previous lesson, we built the ImageGeneratorHandler, which acts as an intermediary between our service layer and the API routes we'll create today. The handler handles the business logic of validating inputs, calling the appropriate service methods, and formatting responses. Now, we'll create the HTTP routes that will receive requests from clients and pass them to our handler.

Our Go API server has two main routes:

1. A route to handle image generation requests -- /generate-image?prompt=
2. A route to retrieve all previously generated images -- /images

By the end of this lesson, you'll have a complete Go web server that integrates with the controller we built previously, providing a clean interface for clients to generate and retrieve images.

Let's start by creating a function to run our Go web server. We'll build upon the InitializeApp function we wrote in the previous lesson and add a new RunServer function:

server/server.go

In this code, we're calling the InitializeApp() function and listening on port 3000. This is the port we want our server to run on to work properly in the CodeSignal environment.

Next, we need a main() function to execute the server. This is very easy:

solution.go

From here, you can run your server with go run solution.go

Now that we have our Go web server set up, let's test it with curl:

You can get the stored images with:

In this lesson, we successfully integrated a Go web server using the Fiber framework to expose our image generation service through HTTP routes. We created two main routes: one for generating images based on a prompt and another for retrieving all previously generated images. By setting up the RunServer function and executing it in the main() function, we were able to run our server and test the API using curl commands.

In the upcoming practice, you'll have the opportunity to reinforce your understanding by implementing additional features or making modifications to the existing server setup. This will help solidify your knowledge of integrating a Go web server with an image generation service.

Now your API is ready for a shiny frontend!