Using Gemini 1.5 Pro for Code Generation: A Developer's Practical Guide

import google.generativeai as genai
import os
from pathlib import Path

genai.configure(api_key=os.environ["GEMINI_API_KEY"])
model = genai.GenerativeModel("gemini-1.5-pro-latest")

def upload_repo_files(repo_path: str) -> list:
    """Upload all Python files from a repo as Gemini file parts."""
    files = []
    for py_file in Path(repo_path).rglob("*.py"):
        with open(py_file, "r") as f:
            content = f.read()
        files.append(f"# File: {py_file}\n{content}\n")
    return files

repo_files = upload_repo_files("./my_project")
combined = "\n\n".join(repo_files)

response = model.generate_content(
    f"""Analyze this Python codebase and identify:
1. Security vulnerabilities (SQL injection, hardcoded secrets, etc.)
2. Performance bottlenecks
3. Missing error handling

Codebase:
{combined}""",
    generation_config={"temperature": 0.1, "max_output_tokens": 4096},
)
print(response.text)

model_with_tools = genai.GenerativeModel(
    "gemini-1.5-pro-latest",
    tools="code_execution",
)

response = model_with_tools.generate_content(
    "Write a function to calculate the Fibonacci sequence, then verify it produces correct results for n=10"
)

for part in response.candidates[0].content.parts:
    if hasattr(part, "executable_code"):
        print("Code:", part.executable_code.code)
    elif hasattr(part, "code_execution_result"):
        print("Result:", part.code_execution_result.output)
    else:
        print("Text:", part.text)

import PIL.Image

error_screenshot = PIL.Image.open("error_screenshot.png")
with open("suspect_file.py", "r") as f:
    code = f.read()

response = model.generate_content([
    error_screenshot,
    f"This error appears when running the following code. Identify the bug and provide the fix:\n\n{code}",
])
print(response.text)