Gemini 2.0 Flash Image Generation and Editing

Gemini 2.0 Flash Image Generation and Editing

Nextjs quickstart for to generating and editing images with Google Gemini 2.0 Flash. It allows users to generate images from text prompts or edit existing images through natural language instructions, maintaining conversation context for iterative refinements. Try out the hosted demo at Hugging Face Spaces.

https://github.com/user-attachments/assets/8ffa5ee3-1b06-46a9-8b5e-761edb0e00c3

Get your GEMINI_API_KEY key here and start building.

How It Works:

1. Create Images: Generate images from text prompts using Gemini 2.0 Flash
2. Edit Images: Upload an image and provide instructions to modify it
3. Conversation History: Maintain context through a conversation with the AI for iterative refinements
4. Download Results: Save your generated or edited images

Basic request

For developers who want to call the Gemini API directly, you can use the Google Generative AI JavaScript SDK:

1
const { GoogleGenerativeAI } = require("@google/generative-ai");
2
const fs = require("fs");
3

4
const genAI = new GoogleGenerativeAI(process.env.GEMINI_API_KEY);
5

6
async function generateImage() {
7
  const contents =
8
    "Hi, can you create a 3d rendered image of a pig " +
9
    "with wings and a top hat flying over a happy " +
10
    "futuristic scifi city with lots of greenery?";
11

12
  // Set responseModalities to include "Image" so the model can generate
13
  const model = genAI.getGenerativeModel({
14
    model: "gemini-2.0-flash-exp",
15
    generationConfig: {
16
      responseModalities: ["Text", "Image"]
17
    }
18
  });
19

20
  try {
21
    const response = await model.generateContent(contents);
22
    for (const part of response.response.candidates[0].content.parts) {
23
      // Based on the part type, either show the text or save the image
24
      if (part.text) {
25
        console.log(part.text);
26
      } else if (part.inlineData) {
27
        const imageData = part.inlineData.data;
28
        const buffer = Buffer.from(imageData, "base64");
29
        fs.writeFileSync("gemini-native-image.png", buffer);
30
        console.log("Image saved as gemini-native-image.png");
31
      }
32
    }
33
  } catch (error) {
34
    console.error("Error generating content:", error);
35
  }
36
}

Features

• 🎨 Text-to-image generation with Gemini 2.0 Flash
• 🖌️ Image editing through natural language instructions
• 💬 Conversation history for context-aware image refinements
• 📱 Responsive UI built with Next.js and shadcn/ui
• 🔄 Seamless workflow between creation and editing modes
• ⚡ Uses Gemini 2.0 Flash Javascript SDK

Getting Started

Local Development

First, set up your environment variables:

1
cp .env.example .env

Add your Google AI Studio API key to the .env file:

Get your GEMINI_API_KEY key here.

1
GEMINI_API_KEY=your_google_api_key

Then, install dependencies and run the development server:

1
npm install
2
npm run dev

Open http://localhost:3000 with your browser to see the application.

Deployment

Edgenivo

Docker

1. Build the Docker image:

1
docker build -t nextjs-gemini-image-editing .

1. Run the container with your Google API key:

1
docker run -p 3000:3000 -e GEMINI_API_KEY=your_google_api_key nextjs-gemini-image-editing

Or using an environment file:

1
# Run container with env file
2
docker run -p 3000:3000 --env-file .env nextjs-gemini-image-editing

Open http://localhost:3000 with your browser to see the application.

Technologies Used

• Next.js - React framework for the web application
• Google Gemini 2.0 Flash - AI model for image generation and editing
• shadcn/ui - Re-usable components built using Radix UI and Tailwind CSS

License

This project is licensed under the Apache License 2.0 - see the LICENSE file for details.