Update app.py

app.py

@@ -266,31 +266,62 @@ def generate(
 ##############################################################################
 # 5. Gradio UI
 ##############################################################################
-with gr.Blocks() as demo:
-    gr.Markdown("# InstantID – Beautiful Realistic Asians v7")
-    with gr.Row():
-        with gr.Column():
-            face_in   = gr.Image(label="顔写真", type="numpy")
-            subj_in   = gr.Textbox(label="被写体説明", placeholder="e.g. woman in black suit, smiling")
-            add_in    = gr.Textbox(label="追加プロンプト")
-            addneg_in = gr.Textbox(label="追加ネガティブ")
-            ip_sld    = gr.Slider(0, 1.5, 0.65, step=0.05, label="IP-Adapter scale")
-            cfg_sld   = gr.Slider(1, 15, 6, step=0.5, label="CFG")
-            step_sld  = gr.Slider(10, 50, 20, step=1, label="Steps")
-            w_sld     = gr.Slider(512, 1024, 512, step=64, label="幅")
-            h_sld     = gr.Slider(512, 1024, 768, step=64, label="高さ")
-            up_ck     = gr.Checkbox(label="アップスケール", value=True)
-            up_fac    = gr.Slider(1, 8, 2, step=1, label="倍率")
-            btn       = gr.Button("生成", variant="primary")
-        with gr.Column():
-            out_img = gr.Image(label="結果")
-
-    btn.click(
-        generate,
-        [face_in, subj_in, add_in, addneg_in, cfg_sld, ip_sld, step_sld, w_sld, h_sld, up_ck, up_fac],
-        out_img,
-        api_name="predict",
-    )
-
-print("launching …")
-demo.queue().launch(show_error=True)
+
+# FastAPI app
+app = FastAPI()
+
+@app.post("/generate")
+async def generate_image_api(request: Request):
+    try:
+        body = await request.json()
+
+        image_base64 = body.get("image")
+        subject      = body.get("subject", "beautiful woman")
+        add_prompt   = body.get("add_prompt", "")
+        add_neg      = body.get("add_neg", "")
+        cfg          = float(body.get("cfg", 6.0))
+        ip_scale     = float(body.get("ip_scale", 0.65))
+        steps        = int(body.get("steps", 20))
+        width        = int(body.get("width", 512))
+        height       = int(body.get("height", 768))
+        upscale      = bool(body.get("upscale", False))
+        up_factor    = int(body.get("up_factor", 2))
+
+        # decode image
+        if not image_base64:
+            raise ValueError("Base64画像がありません")
+
+        img_bytes = base64.b64decode(image_base64.split(",")[-1])
+        pil_image = Image.open(BytesIO(img_bytes)).convert("RGB")
+        np_image = np.array(pil_image)
+
+        result_image = generate(
+            face_np=np_image,
+            subject=subject,
+            add_prompt=add_prompt,
+            add_neg=add_neg,
+            cfg=cfg,
+            ip_scale=ip_scale,
+            steps=steps,
+            w=width,
+            h=height,
+            upscale=upscale,
+            up_factor=up_factor,
+        )
+
+        buffer = BytesIO()
+        result_image.save(buffer, format="PNG")
+        result_base64 = base64.b64encode(buffer.getvalue()).decode("utf-8")
+
+        return {"data": ["data:image/png;base64," + result_base64]}
+
+    except Exception as e:
+        return JSONResponse(status_code=500, content={"error": str(e)})
+
+# Gradio UIをマウント
+from app import demo  # もともとの Gradio UI
+app = gr.mount_gradio_app(app, demo, path="/")
+
+if __name__ == "__main__":
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=7860)