Update app.py

app.py

@@ -1,27 +1,30 @@
-import os
-import io
-import base64
 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 test_image_echo(request: Request):
+async def generate_image(request: Request):
     try:
         body = await request.json()
         base64_image = body.get("image")
-        prompt = body.get("prompt", "test")
+        prompt = body.get("prompt", "")
+
+        # 画像を受け取ってそのまま再エンコードして返す
+        image_data = base64.b64decode(base64_image.split(",")[-1])
+        image = Image.open(io.BytesIO(image_data)).convert("RGB")
 
-        if not base64_image:
-            return JSONResponse(status_code=400, content={"success": False, "error": "No image provided."})
+        buffered = io.BytesIO()
+        image.save(buffered, format="PNG")
+        encoded_img = base64.b64encode(buffered.getvalue()).decode("utf-8")
 
-        # 👇そのまま返す（エコーバック）
         return JSONResponse(content={
             "success": True,
-            "echo_image_base64": base64_image,
-            "prompt_received": prompt
+            "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)})