File size: 1,045 Bytes
8fa33c0 fcae93c 8fa33c0 fcae93c 8fa33c0 fcae93c 8fa33c0 fcae93c 8fa33c0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 |
from fastapi import FastAPI, Request
from fastapi.responses import JSONResponse
import base64
import io
from PIL import Image
app = FastAPI()
@app.post("/generate")
async def generate_image(request: Request):
try:
body = await request.json()
base64_image = body.get("image")
prompt = body.get("prompt", "")
# Base64文字列 → PIL画像
image_data = base64.b64decode(base64_image.split(",")[-1])
image = Image.open(io.BytesIO(image_data)).convert("RGB")
# 再エンコードしてBase64で返す
buffered = io.BytesIO()
image.save(buffered, format="PNG")
encoded_img = base64.b64encode(buffered.getvalue()).decode("utf-8")
return JSONResponse(content={
"success": True,
"echo_prompt": prompt,
"image_base64": f"data:image/png;base64,{encoded_img}"
})
except Exception as e:
return JSONResponse(status_code=500, content={
"success": False,
"error": str(e)
})
|