Crafting Effective Prompts for Image Generation in Go

Welcome back! In the previous lesson, you learned how to generate a simple image using Google's Gemini API and its Gemini 3.1 Flash image model. Now, we will learn the art of crafting effective prompts to achieve desired image outputs. Crafting a well-thought-out prompt is crucial because it directly influences the quality and relevance of the generated image. In this lesson, we will explore the key components of a prompt: subject, context, and style. Understanding these components will empower you to create more detailed and specific prompts, leading to more accurate and visually appealing images.

A prompt is essentially a textual description that guides the image generation process. It consists of three main components: subject, context, and style. The subject is the primary focus of the image, such as a cat or a landscape. The context provides additional details about the setting or environment, like a bustling city at night. The style defines the artistic approach, such as digital art or watercolor painting. Each component plays a vital role in shaping the final image.

For example, a simple prompt like "A cat" might generate a generic image of a cat. However, by adding context and style, such as "A black cat sitting on a windowsill overlooking a bustling city at night, in the style of digital art," you can create a more vivid and specific image. This detailed prompt provides the model with more information, resulting in a richer and more accurate output.

Let's walk through the code to see how different prompts affect the generated images. The code initializes our image generation helper function and defines a list of prompts with varying levels of detail. For each prompt, the code generates an image and saves it to the specified directory.

package main

import (
    "log"
    "fmt"
    "os"
    "path/filepath"
)

func main() {
    // Define prompts with varying detail
    prompts := []string {
        "A cat",  // Subject only
        "A cat in the style of watercolor painting", // Subject and style
        "A cat in the style of watercolor painting sitting on a windowsill overlooking a bustling city at night",  // Subject, style, and context
    }
    
    imgCount := 0
    for _, prompt := range prompts {
        imageDataList, err := generateGeminiImages(prompt, nil)
        if err != nil {
            log.Printf("Warning: %v", err)
        }
        
        if len(imageDataList) > 0 {
            fmt.Printf("✅ Generated %d images\n", len(imageDataList))
            
            fileName := fmt.Sprintf("image_%d.jpg", imgCount)
            outputPath := filepath.Join("server/static/images/", fileName)
            if err := os.WriteFile(outputPath, imageDataList[0], 0644); err != nil {
                log.Fatalf("❌ Failed to save image: %v", err)
            }
            
            imgCount++
            fmt.Printf("✅ First image saved as %s\n", outputPath)
        } else {
            log.Fatal("❌ No images were generated")
        }
    }
}

As you can see, the code iterates over each prompt, generating an image for each one using our helper function generateGeminiImages. The level of detail in the prompt directly affects the complexity and specificity of the generated image. By experimenting with different prompts, you can observe how the model interprets and visualizes the descriptions.

The images are saved with sequential numbering (image_0.jpg, image_1.jpg, etc.), which provides a simple way to organize and reference the generated images. This naming convention makes it easy to track which prompt produced which image during your experiments.