import requests
import base64
url = "https://ai.alad.com/v1beta/models/gemini-2.0-flash:generateContent"
params = {"key": "YOUR_API_KEY"}
with open("input.png", "rb") as f:
image_b64 = base64.b64encode(f.read()).decode()
data = {
"contents": [
{
"role": "user",
"parts": [
{"text": "把背景换成海滩场景"},
{"inline_data": {"mime_type": "image/png", "data": image_b64}}
]
}
],
"generationConfig": {
"responseModalities": ["IMAGE"]
}
}
response = requests.post(url, params=params, json=data)
result = response.json()
output = result["candidates"][0]["content"]["parts"][0]["inline_data"]["data"]
with open("output.png", "wb") as f:
f.write(base64.b64decode(output))