Update app.py

app.py

@@ -3,13 +3,14 @@ from fastapi.responses import JSONResponse
 import base64
 import io
 from PIL import Image
+import traceback  # ←これを追加
 
 app = FastAPI()
 
 @app.get("/")
 def root():
     return {"status": "ok"}
-    
+
 @app.post("/generate")
 async def generate_image(request: Request):
     try:
@@ -17,11 +18,13 @@ async def generate_image(request: Request):
         base64_image = body.get("image")
         prompt = body.get("prompt", "")
 
-        # base64 → PIL Image
+        if not base64_image:
+            return JSONResponse(status_code=400, content={"success": False, "error": "image is required"})
+
+        # 画像デコードと変換
         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")
@@ -33,8 +36,7 @@ async def generate_image(request: Request):
         })
 
     except Exception as e:
-        # ログ出力を追加
         print("=== ERROR OCCURRED ===")
         traceback.print_exc()
         print("======================")
-        return JSONResponse(status_code=500, content={"success": False, "error": str(e)})
+        return JSONResponse(status_code=500, content={"success": False, "error": str(e)})