Update app.py
Browse files
app.py
CHANGED
|
@@ -6,6 +6,10 @@ from PIL import Image
|
|
| 6 |
|
| 7 |
app = FastAPI()
|
| 8 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 9 |
@app.post("/generate")
|
| 10 |
async def generate_image(request: Request):
|
| 11 |
try:
|
|
@@ -13,11 +17,11 @@ async def generate_image(request: Request):
|
|
| 13 |
base64_image = body.get("image")
|
| 14 |
prompt = body.get("prompt", "")
|
| 15 |
|
| 16 |
-
#
|
| 17 |
image_data = base64.b64decode(base64_image.split(",")[-1])
|
| 18 |
image = Image.open(io.BytesIO(image_data)).convert("RGB")
|
| 19 |
|
| 20 |
-
#
|
| 21 |
buffered = io.BytesIO()
|
| 22 |
image.save(buffered, format="PNG")
|
| 23 |
encoded_img = base64.b64encode(buffered.getvalue()).decode("utf-8")
|
|
@@ -27,8 +31,10 @@ async def generate_image(request: Request):
|
|
| 27 |
"echo_prompt": prompt,
|
| 28 |
"image_base64": f"data:image/png;base64,{encoded_img}"
|
| 29 |
})
|
|
|
|
| 30 |
except Exception as e:
|
| 31 |
-
|
| 32 |
-
|
| 33 |
-
|
| 34 |
-
|
|
|
|
|
|
| 6 |
|
| 7 |
app = FastAPI()
|
| 8 |
|
| 9 |
+
@app.get("/")
|
| 10 |
+
def root():
|
| 11 |
+
return {"status": "ok"}
|
| 12 |
+
|
| 13 |
@app.post("/generate")
|
| 14 |
async def generate_image(request: Request):
|
| 15 |
try:
|
|
|
|
| 17 |
base64_image = body.get("image")
|
| 18 |
prompt = body.get("prompt", "")
|
| 19 |
|
| 20 |
+
# base64 → PIL Image
|
| 21 |
image_data = base64.b64decode(base64_image.split(",")[-1])
|
| 22 |
image = Image.open(io.BytesIO(image_data)).convert("RGB")
|
| 23 |
|
| 24 |
+
# 再エンコード
|
| 25 |
buffered = io.BytesIO()
|
| 26 |
image.save(buffered, format="PNG")
|
| 27 |
encoded_img = base64.b64encode(buffered.getvalue()).decode("utf-8")
|
|
|
|
| 31 |
"echo_prompt": prompt,
|
| 32 |
"image_base64": f"data:image/png;base64,{encoded_img}"
|
| 33 |
})
|
| 34 |
+
|
| 35 |
except Exception as e:
|
| 36 |
+
# ログ出力を追加
|
| 37 |
+
print("=== ERROR OCCURRED ===")
|
| 38 |
+
traceback.print_exc()
|
| 39 |
+
print("======================")
|
| 40 |
+
return JSONResponse(status_code=500, content={"success": False, "error": str(e)})
|