Getting Started with Gemini API in Google Colab

To use the Gemini models via the google.generativeai library in Colab, we need an API key from Google AI Studio. Visit Google AI Studio – API Keys, sign in with your Google account, and generate a new API key if you haven’t already. Once generated, this key can be securely stored using Colab’s secrets manager, allowing us to authenticate and interact with Gemini models programmatically.

Listing Available Models from Google Generative AI

This script uses the google.generativeai library to list all available models supported by our API key. It retrieves the key from Colab’s userdata securely and prints model names and their supported generation methods.

import google.generativeai as genai
from google.colab import userdata

try:
  GOOGLE_API_KEY = userdata.get('GOOGLE_API_KEY')
  genai.configure(api_key=GOOGLE_API_KEY)

  # List available models
  for m in genai.list_models():
    print(f"Name: {m.name}")
    print(f"Supported generation methods: {m.supported_generation_methods}")
    print("-" * 20)

except Exception as e:
  print(f"An error occurred: {e}")
  print("Check your API key with the name 'GOOGLE_API_KEY'.")

Name: models/embedding-gecko-001
Supported generation methods: ['embedText', 'countTextTokens']
--------------------
Name: models/gemini-1.5-pro-latest
Supported generation methods: ['generateContent', 'countTokens']
--------------------
Name: models/gemini-1.5-pro-002
Supported generation methods: ['generateContent', 'countTokens', 'createCachedContent']
--------------------

Text Generation with Gemini API in Google Colab

This script uses the google.generativeai library to initialize the gemini-1.5-flash-latest model and generate a short poem about the stars. The API key is accessed via userdata.get(), and the generate_content() method handles the text generation task.

import google.generativeai as genai
from google.colab import userdata

try:
  GOOGLE_API_KEY = userdata.get('GOOGLE_API_KEY')
  genai.configure(api_key=GOOGLE_API_KEY)

  # Initialize the model that supports generateContent
  # You can choose another model from the list if you prefer, e.g., 'gemini-1.5-pro-latest'
  model = genai.GenerativeModel('gemini-1.5-flash-latest')

  # Generate content
  response = model.generate_content("Write a short poem about the stars.")
  print(response.text)

except Exception as e:
  print(f"An error occurred: {e}")
  print("Check your API key in the name 'GOOGLE_API_KEY'.")

Distant suns, a scattered gleam,
Across the velvet, midnight stream.
A million points of silver light,
Igniting darkness, pure and bright.
Silent stories, ages old,
In cosmic dust, their tales unfold.

Generating Content Dynamically Based on User Prompt

This script uses the google.generativeai library and allows the user to enter a custom prompt for generating text with the gemini-1.5-flash-latest model. The API key is retrieved via userdata.get(), and the response is printed dynamically based on user input.

import google.generativeai as genai
from google.colab import userdata

try:
  GOOGLE_API_KEY = userdata.get('GOOGLE_API_KEY')
  genai.configure(api_key=GOOGLE_API_KEY)

  # Initialize the model that supports generateContent
  # You can choose another model from the list if you prefer,
  # e.g., 'gemini-1.5-pro-latest'
  model = genai.GenerativeModel('gemini-1.5-flash-latest')

  # Get prompt from user input
  user_prompt = input("Enter your prompt for the model: ")

  # Generate content based on user input
  response = model.generate_content(user_prompt)
  print(response.text)

except Exception as e:
  print(f"An error occurred: {e}")
  print("Please check your API key.")

Enter your prompt for the model: Tell me why flight path are not straight line.

Flight paths aren't straight lines for a variety of reasons, all boiling down to efficiency, safety, and regulatory compliance:

* **Jet Streams:**  These high-altitude air currents can significantly impact flight time and fuel consumption.  Pilots often plan routes to take advantage of tailwinds (flying with the jet stream) and avoid headwinds (flying against it). This results in a curved path, even if the ground distance appears longer.

💻 Local Python Script for VS Code or any other platform.

🔐 .env File (Same Directory)

• Do not add quotes around the key value.
• If you're using Git, add .env to your .gitignore to avoid exposing your API key.

GOOGLE_API_KEY=your_actual_api_key_here 

import os
import google.generativeai as genai
from dotenv import load_dotenv

# Load API key from .env file
load_dotenv()
GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")

try:
    # Configure Gemini API
    genai.configure(api_key=GOOGLE_API_KEY)

    # Load and initialize the model
    model = genai.GenerativeModel('gemini-1.5-flash-latest')

    # Get prompt from user input
    user_prompt = input("Enter your prompt for the model: ")

    # Generate content based on user input
    response = model.generate_content(user_prompt)
    print(response.text)

except Exception as e:
    print(f"An error occurred: {e}")
    print("Please check your API key.")

Chat Mode

import os
import google.generativeai as genai
from dotenv import load_dotenv

# Load API key from .env file
load_dotenv()
GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")

try:
    # Configure Gemini API
    genai.configure(api_key=GOOGLE_API_KEY)

    # Initialize chat model
    model = genai.GenerativeModel('gemini-1.5-flash-latest')
    chat = model.start_chat()

    print("🤖 Gemini AI Chat (type 'exit' to quit)\n")

    while True:
        user_input = input("You: ")

        if user_input.lower() in ['exit', 'quit']:
            print("👋 Exiting chat. Goodbye!")
            break

        response = chat.send_message(user_input)
        print(f"Gemini: {response.text}\n")

except Exception as e:
    print(f"An error occurred: {e}")
    print("Please check your API key.")

Conclusion

Subhendu Mohapatra

Author

🎥 Join me live on YouTube

Passionate about coding and teaching, I publish practical tutorials on PHP, Python, JavaScript, SQL, and web development. My goal is to make learning simple, engaging, and project‑oriented with real examples and source code.

← Subscribe to our YouTube Channel here