Gemini 编辑图片

IMAGE_BASE64=$(base64 < input.png | tr -d '\n')

curl "https://api.duomi.ai/v1beta/models/your-model:generateContent" \
  -H "Authorization: Bearer API_KEY" \
  -H "Content-Type: application/json" \
  --data-binary @- <<JSON
{
  "contents": [{
    "role": "user",
    "parts": [
      {"text": "把背景改成夜晚城市，保留主体"},
      {"inlineData": {"mimeType": "image/png", "data": "$IMAGE_BASE64"}}
    ]
  }],
  "generationConfig": {
    "imageConfig": {"aspectRatio": "9:16"}
  }
}
JSON

import base64
import requests

with open("input.png", "rb") as file:
    image_base64 = base64.b64encode(file.read()).decode("utf-8")

response = requests.post(
    "https://api.duomi.ai/v1beta/models/your-model:generateContent",
    headers={"Authorization": "Bearer API_KEY"},
    json={
        "contents": [{
            "role": "user",
            "parts": [
                {"text": "把背景改成夜晚城市，保留主体"},
                {"inlineData": {"mimeType": "image/png", "data": image_base64}},
            ],
        }],
        "generationConfig": {"imageConfig": {"aspectRatio": "9:16"}},
    },
    timeout=120,
)
response.raise_for_status()
data = response.json()
for candidate in data.get("candidates", []):
    for index, part in enumerate(candidate.get("content", {}).get("parts", [])):
        image = part.get("inlineData")
        if image and image.get("data"):
            with open(f"edited-{index}.png", "wb") as file:
                file.write(base64.b64decode(image["data"]))

import { readFile, writeFile } from "node:fs/promises";

const input = (await readFile("input.png")).toString("base64");
const response = await fetch("https://api.duomi.ai/v1beta/models/your-model:generateContent", {
  method: "POST",
  headers: { Authorization: "Bearer API_KEY", "Content-Type": "application/json" },
  body: JSON.stringify({
    contents: [{ role: "user", parts: [
      { text: "Replace the background with a city at night" },
      { inlineData: { mimeType: "image/png", data: input } }
    ] }],
    generationConfig: { imageConfig: { aspectRatio: "9:16" } }
  })
});
const data = await response.json();
const image = data.candidates?.[0]?.content?.parts?.find((part) => part.inlineData)?.inlineData;
if (image?.data) await writeFile("edited.png", Buffer.from(image.data, "base64"));