Update app.py

app.py

@@ -1278,5 +1278,28 @@ with gr.Blocks(css=CUSTOM_CSS, title="OmniParse AI Enterprise") as demo:
     gr_register_btn.click(fn=gr_register_with_tos, inputs=[gr_auth_user, gr_auth_pass], outputs=[gr_auth_response])
     gr_checkout_btn.click(fn=gr_checkout, inputs=[gr_auth_user], outputs=[gr_checkout_out])
 
+# ─── FASTAPI + STRIPE WEBHOOK ────────────────────────────────────────────────
+from fastapi import FastAPI, Request, Response
+import uvicorn
+
+app_api = FastAPI()
+
+@app_api.post("/webhook")
+async def stripe_webhook(request: Request):
+    payload = await request.body()
+    sig_header = request.headers.get("stripe-signature", "")
+    result = handle_stripe_webhook(payload, sig_header)
+    data = json.loads(result)
+    status_code = 200 if data.get("status") == "OK" else 400
+    return Response(content=result, status_code=status_code, media_type="application/json")
+
+@app_api.get("/health")
+async def health():
+    return {"status": "ok", "model_loaded": MODEL_LOADED}
+
+# Mount Gradio into FastAPI
+import gradio as gr
+app_api = gr.mount_gradio_app(app_api, demo, path="/")
+
 if __name__ == "__main__":
-    demo.launch()
+    uvicorn.run(app_api, host="0.0.0.0", port=7860)