|
|
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", "") |
|
|
|
|
|
|
|
|
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: |
|
|
return JSONResponse(status_code=500, content={ |
|
|
"success": False, |
|
|
"error": str(e) |
|
|
}) |
|
|
|
