File size: 1,182 Bytes
8fa33c0 fcae93c 8fa33c0 fcae93c 22dcdf9 8fa33c0 fcae93c 8fa33c0 22dcdf9 8fa33c0 22dcdf9 8fa33c0 22dcdf9 fcae93c 22dcdf9 |
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 36 37 38 39 40 |
from fastapi import FastAPI, Request
from fastapi.responses import JSONResponse
import base64
import io
from PIL import Image
app = FastAPI()
@app.get("/")
def root():
return {"status": "ok"}
@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
image_data = base64.b64decode(base64_image.split(",")[-1])
image = Image.open(io.BytesIO(image_data)).convert("RGB")
# 再エンコード
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:
# ログ出力を追加
print("=== ERROR OCCURRED ===")
traceback.print_exc()
print("======================")
return JSONResponse(status_code=500, content={"success": False, "error": str(e)}) |